flax
Make fully replicated array before saving checkpoints for examples that use pmap.
#4707

Merged

Make fully replicated array before saving checkpoints for examples that use pmap. #4707

copybara-service merged 1 commit into main from test_748033335

copybara-service force pushed from a2223ad2 to bcf8b716 1 year ago

copybara-service force pushed from bcf8b716 to 2e18ca55 1 year ago

Make fully replicated array before saving checkpoints for examples th…

aa897790

copybara-service force pushed from 2e18ca55 to aa897790 1 year ago

copybara-service merged aa897790 into main 1 year ago

copybara-service deleted the test_748033335 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone