fix: put upper bounds on versions of jax and jaxlib (#150)

instadeepai · Jun 1, 2023 · d549cec · d549cec
1 parent b16cf5d
commit d549cec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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