← heapsort
NEWS↑ trending42

Introducing AutoMuon, a one line drop in for AdamW [P]

Reddit r/MachineLearningΒ·April 26, 2026

AutoMuon, a new Python package, enables the Muon optimizer to be used as a drop-in replacement for AdamW in PyTorch training pipelines. It automatically identifies and applies the appropriate optimizer for each model parameter, using Muon for weight matrices and AdamW for other components.

Read original β†—