flax
e9bde53c - make flax work with upcoming JAX change to tree_map (being more careful about

Commit
1 year ago
make flax work with upcoming JAX change to tree_map (being more careful about Nones) PiperOrigin-RevId: 641087999
Author
Committer
Parents
Loading