Post-Grokking Collapse at the Representation-Readout Interface in Muon-Trained Transformers

Ali Janati, Kaoutar El Maghraoui, Andrei Kanavalau +1

Under the standard split, Muon gets hidden matrices and AdamW embeddings/output head. Muon groks modular addition faster, but its solutions do not hold. All nine configurations on…