The Polar Express: Optimal Matrix Sign Methods and their Application to the Muon Algorithm
ICLROral2026
TL;DR
We introduce a GPU-friendly algorithm for computing the polar decomposition of a matrix to low accuracy that is optimal in its class. This improves Muon.
Opening excerpt from the authors’ abstract. source