update requirements

chaoming0625 · Sep 22, 2023 · 0c46f00 · 0c46f00
1 parent 572312f
commit 0c46f00
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, <0.4.16
-jaxlib>=0.4.1, <0.4.16
+jax
+jaxlib
 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, <0.4.16
-jaxlib>=0.4.1, <0.4.16
+jax
+jaxlib
 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, <0.4.16
+jax
 tqdm
 msgpack
 numba
diff --git a/setup.py b/setup.py
@@ -51,7 +51,7 @@
   author_email='chao.brain@qq.com',
   packages=packages,
   python_requires='>=3.8',
-  install_requires=['numpy>=1.15', 'jax>=0.4.1, <0.4.16', 'tqdm', 'msgpack', 'numba'],
+  install_requires=['numpy>=1.15', 'jax', 'tqdm', 'msgpack', 'numba'],
   url='https://github.com/brainpy/BrainPy',
   project_urls={
     "Bug Tracker": "https://github.com/brainpy/BrainPy/issues",