flax
05a2add6
- Avoid passing python collections to jax reductions
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Avoid passing python collections to jax reductions PiperOrigin-RevId: 329010801
References
linen-rtd
#439 - Avoid passing python collections to jax reductions
Author
a-googler
Committer
a-googler
Parents
aee7fbe9
Loading
