Merge branch 'main' into feat/add-sudoku-environment

instadeepai · Jun 1, 2023 · 83e087f · 83e087f
2 parents 873cb08 + d549cec
commit 83e087f
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