diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index d09a42d9a..0047cfda2 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -2,7 +2,7 @@ chex>=0.1.3
 dm-env>=1.5
 gymnasium>=1.0
 huggingface-hub
-jax>=0.2.26
+jax>=0.2.26,<0.4.36
 matplotlib~=3.7.4
 numpy>=1.19.5
 Pillow>=9.0.0