DeepSpeed
Fix ZeRO-3 Muon all-gather for mixed parameter shapes
#8628

Merged

Fix ZeRO-3 Muon all-gather for mixed parameter shapes #8628

pengdurice merged 6 commits into deepspeedai:master from jinyouzhi:fix_gloo_muon

Fix ZeRO-3 Muon all-gather for mixed parameter shapes

652bd20e

jinyouzhi requested a review from

tjruwase 25 days ago

jinyouzhi requested a review from

tohtana 25 days ago

alanhuangyoo commented on 2026-09-23

Merge origin/master into fix-zero-3-muon-all-gather-for-mixed-par

08e34697

Merge branch 'master' into fix_gloo_muon

234ce4a8

Check full gradient shapes for ZeRO-3 Muon all-gather

caf356b9

Merge remote PR updates into fix_gloo_muon

1aec7b23

alanhuangyoo commented on 2026-09-23

pengdurice added will-review

Merge branch 'master' into fix_gloo_muon

be66d700

pengdurice approved these changes on 2026-09-28

pengdurice enabled auto-merge 19 days ago

PKUWZP requested a review from

PKUWZP 19 days ago

PKUWZP approved these changes on 2026-09-28

pengdurice merged c9e1b4d2 into master 19 days ago

Reviewers

PKUWZP

pengdurice

alanhuangyoo

tjruwase

tohtana

Assignees

No one assigned

Labels

will-review

Milestone

No milestone