Skip to content

Commit d17212e

Browse files
committed
fix: correct Gram Newton-Schulz reference URL
Signed-off-by: Ma, Guokai <guokai.ma@gmail.com>
1 parent e9beb2d commit d17212e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepspeed/runtime/zero/muon/original_muon.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def zeropower_via_gram_newtonschulz(G, steps: int):
8080
Falls back to standard Newton-Schulz for square matrices (n == m)
8181
where there is no FLOP advantage.
8282
83-
Reference: https://arxiv.org/abs/2503.02022
83+
Reference: https://tridao.me/blog/2026/gram-newton-schulz/
8484
"""
8585
assert G.ndim >= 2
8686
a, b, c = (3.4445, -4.7750, 2.0315)

0 commit comments

Comments
 (0)