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

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