proposition. Full-matrix Muon and Shampoo preserve the orthogonal gauge [ftip-00BA]
AGENTDRAFTED
Fix one factor, write \(n\in \{d_1,d_2\}\) for its row dimension, and let
\(G_t,M_t\in \mathbb R^{n\times k}\) be its gradient and Muon momentum buffer.
For \(0\leq \beta <1\), \(M_{-1}=0\), and the source's linear momentum update
\(M_t=\beta M_{t-1}+G_t\), Muon's direction is
\(\Delta _t=\operatorname {msign}(M_t)\). Under real arithmetic this stateful
update and the source's damped Shampoo update are right-equivariant. The Muon
claim includes the stated finite Newton--Schulz approximations; the Shampoo
claim uses its complete left and right accumulators.
Proposition 4.2(3)--(4) and its proof are in
Appendix B.3 of The loss does not see the basis, but Adam does[singh2026lossbasis]. Variable splitting, coordinatewise
clipping, mixed update rules, finite precision, and different damping
conventions lie outside this statement. The Muon part is stateful; it does not
replace the momentum buffer by the current gradient or invoke the memoryless
classification of Definition [ftip-00B7].