diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 797667b55..99f1e6e75 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -1,8 +1,8 @@
 chex>=0.1.3,<0.1.6
 dm-env>=1.5
 gym>=0.22.0
-jax>=0.2.26
-jaxlib>=0.1.74
+jax>=0.2.26,<=0.4.1
+jaxlib>=0.1.74,<=0.4.1
 matplotlib>=3.3.4
 numpy>=1.19.5
 Pillow>=9.0.0