flax
c12ba53e - Avoid depending on JAX internals, which are about to change.

Commit

1 year ago

Avoid depending on JAX internals, which are about to change. PiperOrigin-RevId: 689163398

References

Author

dougalm

Committer

a-googler

Parents