Skip to content

Commit ba5196b

Browse files
Fix polar express coeffs (#74)
* changed polar express coeffs Signed-off-by: mikail <[email protected]>
1 parent daacec6 commit ba5196b

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

emerging_optimizers/orthogonalized_optimizers/muon_utils.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -38,14 +38,14 @@
3838
],
3939
"polar_express": [
4040
# Polar Express iteration from: https://arxiv.org/abs/2505.16932
41-
(7.2086, -15.5131, 9.0178),
42-
(3.9623, -2.5813, 0.4542),
43-
(3.9466, -2.5765, 0.4544),
44-
(3.8991, -2.5671, 0.4566),
45-
(3.7186, -2.5308, 0.4653),
46-
(3.1390, -2.3073, 0.4733),
47-
(2.1715, -1.5246, 0.3885),
48-
(1.8648, -1.2224, 0.3577),
41+
(8.2051, -22.9019, 16.4607),
42+
(4.0664, -2.8612, 0.5184),
43+
(3.9096, -2.8234, 0.5250),
44+
(3.2856, -2.4153, 0.4853),
45+
(2.2779, -1.6198, 0.3985),
46+
(1.8726, -1.2307, 0.3585),
47+
(1.8564, -1.2132, 0.3568),
48+
(1.8750, -1.2500, 0.3750),
4949
],
5050
"aol": [
5151
# from https://github.com/thib-s/flash-newton-schulz/blob/main/newton_schulz_triton.py#L511

0 commit comments

Comments
 (0)