On MUON optimization: From non-convergence to an error analysis with Polar Express and the Newton-Schulz polynomial from implementations
A generalized variant of the MUON optimizer involving an arbitrary number of generalized NS steps with polynomials of possibly arbitrary high degree is proposed and studied and an error analysis for MUON with the generalized NS steps is established that provides convergence rates in terms of the number of gradient step...