Skip to content

Exposing the experimental _split_transpose JAX scan parameter in Flax. #12192

Exposing the experimental _split_transpose JAX scan parameter in Flax.

Exposing the experimental _split_transpose JAX scan parameter in Flax. #12192

Annotations

1 warning

Test import standalone (3.10)

succeeded Mar 28, 2024 in 42s