flax
afcaf66d - [linen] fold rngs on jit to improve caching

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

a-googler

Parents

Loading