DeepSpeed
Add custom reshaping for universal checkpoint
#5390
Merged

Add custom reshaping for universal checkpoint #5390

tjruwase merged 2 commits into master from tohtana/univ_ckpt_custom_shape
tohtana
add custom reshaping for universal checkpoint
3fde7229
tohtana tohtana requested a review from tjruwase tjruwase 1 year ago
tjruwase Merge branch 'master' into tohtana/univ_ckpt_custom_shape
9d68af27
tjruwase
tjruwase commented on 2024-04-12
tjruwase
tjruwase
tjruwase approved these changes on 2024-04-12
stas00
tjruwase
mosheisland
tjruwase tjruwase merged 2c51aba0 into master 1 year ago
tohtana

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone