TorchLean API

NN.MLTheory.Optimization.Muon

Muon #

Muon updates a momentum buffer and passes that buffer through an explicit matrix orthogonalizer before applying the parameter step. The modules collected here separate the reusable contracts, Newton-Schulz backend, step certificates, and exact QR backend.