flax
afcaf66d
- [linen] fold rngs on jit to improve caching
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[linen] fold rngs on jit to improve caching PiperOrigin-RevId: 654699932
References
#4064 - [linen] fold rngs on jit to improve caching
Author
Cristian Garcia
Committer
a-googler
Parents
4e83e091
Loading