update requirements.txt for jax and numpy

- See usage for latest jax here: jax-ml/jax#18403
deepmodeling · Oct 10, 2024 · d03e08f · d03e08f
1 parent 079b9b6
commit d03e08f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,12 +1,12 @@
-numpy>=1.18
+numpy>=1.18, <2
 mkdocs>=1.3.0
 mkdocs-autorefs>=0.4.1
 mkdocs-gen-files>=0.3.4
 mkdocs-literate-nav>=0.4.1
 mkdocstrings>=0.19.0
 mkdocstrings-python>=0.7.0
 pygments>=2.12
-jax>=0.4.1
+jax>=0.4.1, <=0.4.20
 jaxlib>=0.4.1
 pymbar>=4.0.0
 tqdm