Skip to content

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

Exposing the experimental _split_transpose JAX scan parameter in Flax.

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

Annotations

1 warning

Test import standalone (3.10)

succeeded Mar 28, 2024 in 31s