flax
Plumb spmd_axis_name through transforms.vmap through to JAX vmap
#2398

Merged

Plumb spmd_axis_name through transforms.vmap through to JAX vmap #2398

copybara-service merged 0 commits into main from test_467287573

copybara-service force pushed from fcf4b96a 3 years ago

copybara-service merged 07e513f6 into main 3 years ago

copybara-service force pushed to 07e513f6 3 years ago

copybara-service deleted the test_467287573 branch 3 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone