update requirements (#146)

update requirements

Author: chaoming0625

requirements-dev.txt

@@ -2,11 +2,11 @@
 
 numba
 matplotlib>=3.4
-jaxlib>=0.1.64
+jaxlib>=0.3.0
 sympy>=1.6
 scipy>=1.1.0
 networkx
-brainpylib>=0.0.3
+brainpylib>=0.0.4
 
 # test requirements
 pytest

requirements-doc.txt

@@ -1,10 +1,10 @@
 -r requirements.txt
 
 matplotlib>=3.4
-jaxlib>=0.1.64
+jaxlib>=0.3.0
 sympy>=1.6
 scipy>=1.1.0
-brainpylib
+brainpylib>=0.0.4
 numba
 
 # document requirements

requirements-win.txt

@@ -5,7 +5,8 @@ matplotlib>=3.4
 sympy>=1.6
 scipy>=1.1.0
 networkx
-brainpylib>=0.0.3
+brainpylib>=0.0.4
+jaxlib>=0.3.0
 
 # test requirements
 pytest

requirements.txt

@@ -1,3 +1,3 @@
 numpy>=1.15
-jax>=0.2.10
+jax>=0.3.0
 tqdm

setup.py

@@ -30,12 +30,12 @@
   python_requires='>=3.7',
   install_requires=[
     'numpy>=1.15',
-    'jax>=0.2.10',
+    'jax>=0.3.0',
     'tqdm',
   ],
   extras_require={
-    'cpu': ['jaxlib>=0.1.64', 'brainpylib>=0.03'],
-    'cuda': ['jaxlib>=0.1.64', 'brainpylib>=0.03'],
+    'cpu': ['jaxlib>=0.3.0', 'brainpylib>=0.0.4'],
+    'cuda': ['jaxlib>=0.3.0', 'brainpylib>=0.0.4'],
   },
   url='https://github.com/PKU-NIP-Lab/BrainPy',
   keywords='computational neuroscience, brain-inspired computation, '