flax
323d6b85 - Remove fallbacks for old versions of JAX from Flax.

Commit

3 years ago

Remove fallbacks for old versions of JAX from Flax. Flax's setup.py requires JAX >= 0.4.2, so these code paths are dead. PiperOrigin-RevId: 509309356

Author

hawkinsp

Committer

a-googler

Parents