Ademamix#24

Open

davidtweedle wants to merge 33 commits intomlcommons:mainfrom

davidtweedle:ademamix

Commits on Aug 21, 2025

Commits on Sep 13, 2025

updated beta and alpha schedulers to use stephint
davidtweedle
committed

Commits on Sep 17, 2025

testing ademamix
davidtweedle
committed

Commits on Sep 25, 2025

Commits on Oct 16, 2025

Matched submission to nadamw style
davidtweedle
committed

Commits on Oct 23, 2025

Commits on Feb 12, 2026

removed sharding to test on a100s
davidtweedle
committed

Commits on Feb 24, 2026

updated ademamix to add learning rate scheduler, move jax.jit outside update params?
davidtweedle
committed

Commits on Mar 5, 2026

typo
davidtweedle
committed

Commits on Apr 10, 2026