Closed bwohlberg closed 5 months ago
Bump maximum jaxlib / jax version and some other minor changes.
jaxlib
jax
Bump maximum
jaxlib
/jax
version and some other minor changes.