fix(muon): avoid EP/FSDP AGRS OOM - #2069

Open

jayhenry wants to merge 2 commits into

InternLM:mainfrom

jayhenry:fix_muon_oom

Commits on Sep 4, 2026

fix(muon): expose padded all-to-all remainder strategy
jayhenry
committed
adb5cc0
fix(muon): use FSDP-local shard dimension for EP
jayhenry
committed
dd24c6c