ViperEkura 11073bd1d2 refactor: extract composite optimizer helpers and unify naming ...

- add astrai/optim/composite.py with shared step/zero_grad/state_dict/param_groups helpers and OptimizerFactory
- rename MuonMix to MuonAdamW (matches registered name muon_adamw) and file to muon_adamw.py
- use @OptimizerFactory.register decorator in each optimizer module instead of post-import registration in __init__
- fix closure being invoked once per sub-optimizer in MuonAdamW.step (now exactly once via composite_step)
- NoraNAdamW.step now forwards closure correctly

2026-08-01 08:07:45 +08:00

..

distributed.md

docs: restructure to docs/, add guides and developer docs

2026-07-30 00:49:04 +08:00

evaluation.md

docs: restructure to docs/, add guides and developer docs

2026-07-30 00:49:04 +08:00

inference.md

refactor: unify rotary embedding interface and update docs

2026-07-31 16:52:25 +08:00

params.md

refactor: extract composite optimizer helpers and unify naming

2026-08-01 08:07:45 +08:00

preprocessing.md

docs: restructure to docs/, add guides and developer docs

2026-07-30 00:49:04 +08:00

training.md

refactor: unify rotary embedding interface and update docs

2026-07-31 16:52:25 +08:00