flax
4671ddd9 - Exposing the experimental _split_transpose JAX scan parameter in Flax.
Commit
View On GitHub
Commit
1 year ago
Exposing the experimental _split_transpose JAX scan parameter in Flax. PiperOrigin-RevId: 620068356
Author
a-googler a-googler
Committer
a-googler a-googler
Parents
fe995740
FAQ Terms Privacy Refunds Impressum
Loading