flax
support out_sharding mapping in set_metadata
#5312

Merged

support out_sharding mapping in set_metadata #5312

copybara-service merged 1 commit into main from test_878753460

copybara-service force pushed from a7db1d3e to b7031956 3 days ago

support out_sharding mapping in set_metadata

4614ac2b

copybara-service force pushed from b7031956 to 4614ac2b 3 days ago

copybara-service merged 4614ac2b into main 3 days ago

copybara-service deleted the test_878753460 branch 3 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone