update requirements

brainpy · Sep 21, 2023 · 5aed3bd · 5aed3bd
1 parent 1db3cee
commit 5aed3bd
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -1,8 +1,8 @@
 numpy
 numba
 brainpylib
-jax>=0.4.1
-jaxlib>=0.4.1
+jax>=0.4.1, <0.4.16
+jaxlib>=0.4.1, <0.4.16
 matplotlib>=3.4
 msgpack
 tqdm

diff --git a/requirements-doc.txt b/requirements-doc.txt
@@ -2,8 +2,8 @@ numpy
 tqdm
 msgpack
 numba
-jax>=0.4.1
-jaxlib>=0.4.1
+jax>=0.4.1, <0.4.16
+jaxlib>=0.4.1, <0.4.16
 matplotlib>=3.4
 scipy>=1.1.0
 numba

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-jax>=0.4.1
+jax>=0.4.1, <0.4.16
 tqdm
 msgpack
 numba
diff --git a/setup.py b/setup.py
@@ -51,7 +51,7 @@
   author_email='[email protected]',
   packages=packages,
   python_requires='>=3.8',
-  install_requires=['numpy>=1.15', 'jax>=0.4.1', 'tqdm', 'msgpack', 'numba'],
+  install_requires=['numpy>=1.15', 'jax>=0.4.1, <0.4.16', 'tqdm', 'msgpack', 'numba'],
   url='https://github.com/brainpy/BrainPy',
   project_urls={
     "Bug Tracker": "https://github.com/brainpy/BrainPy/issues",