Generalize emerging optimizer integration - #3325

Closed

skyw wants to merge 5 commits into

skyw:generalize_emerging_optimizer

Commits on Feb 9, 2026

Move muon to its own config class
skyw
committed
781de69
Generalize interface of muon to support more optimizers
skyw
committed
bb43e03
Update muon reference to emerging optimizer
skyw
committed
9b792ee
generalize init_state_fn
skyw
committed
af5c29c

Commits on Feb 10, 2026

Update optimizer name string dispatch condition
skyw
committed
2227479