flax
8b8b9298
- Merge pull request #4991 from google:use-jnp-stack-in-stack_forest
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
75 days ago
Merge pull request #4991 from google:use-jnp-stack-in-stack_forest PiperOrigin-RevId: 813513992
References
#4991 - Use `jnp.stack` instead of `np.stack` in `flax.training.common_utils.stack_forest`
Author
a-googler
Parents
54fa32d9
c08e60b3
Loading
