DeepSpeed
Extending Muon Optimizer Support for ZeRO Stage 3
#7919

Merged

Extending Muon Optimizer Support for ZeRO Stage 3 #7919

PKUWZP merged 4 commits into deepspeedai:master from PKUWZP:pr-7798-clean

Extending Muon Optimizer Support for ZeRO Stage 3

56451ea0

PKUWZP requested a review from

tjruwase 22 days ago

PKUWZP requested a review from

tohtana 22 days ago

PKUWZP requested a review from

loadams 22 days ago

chatgpt-codex-connector commented on 2026-03-23

tohtana commented on 2026-03-25

Address review feedback for Muon ZeRO Stage 3 support

da9b7738

Fix yapf formatting in stage3.py and test_muon.py

630f4215

tohtana approved these changes on 2026-03-26

delock approved these changes on 2026-03-26

delock commented on 2026-03-26

Merge branch 'master' into pr-7798-clean

74bce709

PKUWZP merged 956ec6fb into master 20 days ago

Reviewers

delock

tohtana

tjruwase

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone