flax
0797d053
- Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`. PiperOrigin-RevId: 610521805
References
#3714 - Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`.
Author
IvyZX
Committer
a-googler
Parents
83e74665
Loading