flax
Exposing the experimental _split_transpose JAX scan parameter in Flax.
#3795

Merged

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

copybara-service merged 1 commit into main from test_606543268

copybara-service force pushed 1 year ago

Exposing the experimental _split_transpose JAX scan parameter in Flax.

4671ddd9

copybara-service force pushed to 4671ddd9 1 year ago

copybara-service merged 4671ddd9 into main 1 year ago

copybara-service deleted the test_606543268 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone