Test update to jax-rl 2

rainx0r · Jul 3, 2023 · be664a6 · be664a6
1 parent c364ee0
commit be664a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax-rl/requirements.txt b/jax-rl/requirements.txt
@@ -1,6 +1,6 @@
 # Jax
 --find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-jax[cuda12_pip]==0.4.12
+jax[cuda12_pip]==0.4.11
 
 # DeepMind
 dm-haiku==0.0.9