downgrade jaxlib to 0.4.23 due to issues with jaxlib 0.4.24

PiperOrigin-RevId: 607823141
google · Feb 16, 2024 · e24ab47 · e24ab47
1 parent 25f189b
commit e24ab47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt
@@ -184,7 +184,7 @@ jax==0.2.24
     #   orbax-checkpoint
 jax-bitempered-loss==0.0.2
     # via -r praxis-requirements.in
-jaxlib==0.4.24
+jaxlib==0.4.23
     # via
     #   chex
     #   clu