flax
6418755c - [JAX] Replace reference to jax._src.typing with public APIs.

Commit

221 days ago

[JAX] Replace reference to jax._src.typing with public APIs. PiperOrigin-RevId: 826052722

Author

hawkinsp

Committer

a-googler

Parents