Amuse optimizer - #1830

Draft

polycrit wants to merge 16 commits into

amuse-optimizer

Commits on Aug 3, 2026

Commits on Aug 5, 2026

fix collect_class_init_kwargs to also collect keyword-only args
polycrit
committed
224b0f8
add MultiOptimizer composing sub-optimizers over disjoint param subsets
polycrit
committed
55bc603
resolve returnn custom optimizers by short name (multi, lion, amuse)
polycrit
committed
c9068f4
create multi optimizer from config with params_filter partition and per-sub weight decay split
polycrit
committed
3f006ba
call schedule-free optimizer train()/eval() at train epoch boundaries
polycrit
committed
ead1878
restructure AMUSE to one update type per instance, composed via multi
polycrit
committed
0f331d0
Merge remote-tracking branch 'origin/master' into amuse-optimizer
polycrit
committed
eee94f4
save valid optimizer checkpoint metadata also for callable optimizer configs
polycrit
committed
0f3a28e
cache the param to sub-optimizer mapping in the multi optimizer state view
polycrit
committed
bc7541e
guard AMUSE beta1 computation against a degenerate averaging weight
polycrit
committed
ce15546
guard AMUSE beta1 ramp against zero lr at the warmup boundary
polycrit
committed
c128e7e
validate AMUSE constructor arguments
polycrit
committed
7182473