Wing Lian ae66374156 Optimizer refactor and add Muon support (#2367) ...

* add muon optimizer

optimizer_cls_and_kwargs is on trainer_kwargs
only add adamw_kwargs if they're non-null
fix mocks
better handling of override and check the optimizer
unwrap optimizer

* fix import

2025-03-06 11:49:19 -05:00

6.7 KiB

Raw Blame History

View Raw