fix: Require jax below 0.4.36 (#269)

instadeepai · Dec 9, 2024 · 40f4a33 · 40f4a33
1 parent 1ec8808
commit 40f4a33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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