fix: match jax and jaxlib versions

instadeepai · Nov 3, 2023 · a083450 · a083450
1 parent 3248fae
commit a083450
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,7 +1,8 @@
 chex>=0.1.3
 dm-env>=1.5
 gym>=0.22.0
-jax>=0.2.26,<0.4.0
+jax>=0.2.26,<=0.4.15
+jaxlib>=0.2.26,<=0.4.15
 matplotlib>=3.3.4,<3.8
 numpy>=1.19.5
 Pillow>=9.0.0