flax
33911e60 - [Flax] Update JAX version check for dataclass registration.

Commit

1 year ago

[Flax] Update JAX version check for dataclass registration. PiperOrigin-RevId: 627513259

Author

epiqueras

Committer

a-googler

Parents