arXiv Machine Learning By Ziyuan Tang, Tianshi Xu, Yousef Saad, Yuanzhe Xi

Hierarchical Muon: Tiled Newton-Schulz Updates for Efficient Muon Optimization

Read the original on arXiv Machine Learning →

arXiv:2606. 27216v1 Announce Type: cross Abstract: Muon-type optimizers construct update directions for dense neural-network weights by applying a finite Newton-Schulz map to momentum-gradient matrices.

Summary generated by The Flow from the publisher's feed. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Jun 29

Aurora: A Leverage-Aware Spectral Optimizer

arXiv:2606. 27715v1 Announce Type: new Abstract: We show that for tall matrix parameters, like projection matrices in the MLP layers, the Muon update can have row norms that are arbitrarily non-uniform.

By Alec Dewulf, Dhruv Pai, Li Yang, Ashley Zhang, Ben Keigwin