Repository navigation

fix(training): align Muon distributed setup and checkpointing - #6443

Open

cuichenx wants to merge 5 commits into

NVIDIA-NeMo:mainfrom

cuichenx:chcui/tess/muon-distributed-workflows

Open

fix(training): align Muon distributed setup and checkpointing#6443
cuichenx wants to merge 5 commits into
NVIDIA-NeMo:mainfrom
cuichenx:chcui/tess/muon-distributed-workflows

Commits on Oct 13, 2026
- fix(training): align Muon distributed setup and checkpointing
  cuichenx
  committed
  b728726
Commits on Oct 14, 2026
- test(training): align optimizer fixtures with Muon setup
  cuichenx
  committed
  bd6e3cf
Commits on Oct 15, 2026
- fix(ckpt): handle nested optimizer chains during checkpoint restore
  
  rohitrango
  authored and
  cuichenx
  committed
  ee73562
Commits on Oct 16, 2026
- fix(training): preserve compact LayerWise workflows
  
  cuichenx
  and
  yaoyu-33
  committed
  1e5fbe5
- fix(recipes): preserve Kimi Muon layout settings
  cuichenx
  committed
  095afe9