flax
e9bde53c
- make flax work with upcoming JAX change to tree_map (being more careful about
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
make flax work with upcoming JAX change to tree_map (being more careful about Nones) PiperOrigin-RevId: 641087999
References
#3976 - make flax work with upcoming JAX change to tree_map (being more careful about
Author
mattjj
Committer
a-googler
Parents
860bef59
Loading