diff --git a/requirements-dev.txt b/requirements-dev.txt
index 01184540a..49fa49722 100644
--- 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
index d88a0c02a..e6e498937 100644
--- 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
index 74db0a68a..ebf85b86e 100644
--- 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
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 343ca3a89..68debcdee 100644
--- 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', '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",