downgrade jaxlib to 0.4.23 due to issues with jaxlib 0.4.24

PiperOrigin-RevId: 607823141
google · Feb 16, 2024 · 5d7ec50 · 5d7ec50
1 parent d384d6a
commit 5d7ec50
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 @ git+https://github.com/google/jax
     #   orbax-checkpoint
 jax-bitempered-loss==0.0.2
     # via -r praxis-requirements.in
-jaxlib==0.4.24
+jaxlib==0.4.23
     # via
     #   chex
     #   clu