Repository navigation

[Refactor] Unify Muon matrix layout for Newton-Schulz and lr_ratio - #2161

Merged

jayhenry merged 8 commits into

mainfrom

fix/muon-matrix-shape-contract

Oct 26, 2026

Merged

[Refactor] Unify Muon matrix layout for Newton-Schulz and lr_ratio#2161
jayhenry merged 8 commits into
mainfrom
fix/muon-matrix-shape-contract

Commits on Oct 23, 2026
- [Fix] Unify Muon matrix layout for NS and lr_ratio
  
  YanhuiDua
  and
  cursoragent
  committed
  e30a4aa
- [Fix] Harden Muon layout contract for NS callbacks and tests
  
  YanhuiDua
  and
  cursoragent
  committed
  570b0b6
- [Docs] Add Muon precision A/B loss and grad_norm plots for PR #2161
  
  YanhuiDua
  and
  cursoragent
  committed
  86a9b3e
- [Docs] Refresh GLM-5.2 precision curves from PR/main repeat runs
  
  YanhuiDua
  and
  cursoragent
  committed
  f1c38c7
- [Docs] Stop tracking precision plot PNGs in the PR branch
  
  YanhuiDua
  and
  cursoragent
  committed
  5c0bfaf
- [Fix] Route only 2D params to Muon in MuonConfig
  YanhuiDua
  committed
  e157ac8
- [Fix] Keep 2D Muon inputs on the batched Newton-Schulz kernel path
  YanhuiDua
  committed
  86a1e6f
- [Test] Refactor Muon layout contract tests for clarity and speed
  YanhuiDua
  committed
  da6efbe