flax
a147752a - Migrate away from private jax._src.tree_util._registry

Commit
80 days ago
Migrate away from private jax._src.tree_util._registry PiperOrigin-RevId: 918133843
Author
Committer
Parents
Loading