xla
Implement SPMDSavePlanner to take distributed checkpoints
#5170

Merged

Implement SPMDSavePlanner to take distributed checkpoints #5170

jonb377 merged 4 commits into master from jonbolin-save-planner

jonb377 added distributed

jonb377 requested a review from

yeounoh 3 years ago

jonb377 requested a review from

alanwaketan 3 years ago

jonb377 requested a review from

JackCaoG 3 years ago

jonb377 force pushed from 5f4fa0fb to 70b8f71c 3 years ago

alanwaketan commented on 2023-06-14

alanwaketan approved these changes on 2023-06-14

jonb377 force pushed from e2f6f665 to 3d67c833 3 years ago

Implement SPMDSavePlanner to take distributed checkpoints

f5dabf56

Update comment

09369210

jonb377 force pushed from 6d13ed8f to da527176 3 years ago

jonb377 force pushed from da527176 to 6bd0af5a 3 years ago

Duplicate unstable upstream APIs

094e5a3c

jonb377 force pushed from 6bd0af5a to 094e5a3c 3 years ago

Duplicate private API narrow_tensor_by_index

da0d604f

jonb377 merged 4aa4dcaf into master 3 years ago

jonb377 deleted the jonbolin-save-planner branch 3 years ago

Reviewers

alanwaketan

yeounoh

JackCaoG

Assignees

No one assigned

Labels

distributed

Milestone

No milestone