flax
f234fc1b - Account for JAX flag name change in https://github.com/google/jax/pull/20370

Commit
2 years ago
Account for JAX flag name change in https://github.com/google/jax/pull/20370 PiperOrigin-RevId: 617977877
Author
Jake VanderPlas
Committer
Parents
Loading