Randomly sampled coupled `LossContributions.n_steps`#869

Open

jpdunc23 wants to merge 45 commits intomainfrom

feature/coupled-loss-contrib-random-n_steps

Commits on Jan 21, 2026

Allow stochastic atmos training in CoupledStepper
jpdunc23
committed
Merge branch 'main' of github.com:ai2cm/ace into coupled-ft-with-stochastic-atmos
jpdunc23
committed

Commits on Feb 21, 2026

Commits on Feb 22, 2026

Add `optimize_last_step_only` to `LossContributionsConfig`
jpdunc23
committed

Commits on Feb 23, 2026

Randomly sample coupled `LossContributions.n_steps`
jpdunc23
committed

Commits on Mar 25, 2026

Merge branch 'main' of github.com:ai2cm/ace into coupled-ft-with-stochastic-atmos
jpdunc23
committed

Commits on Apr 2, 2026

Commits on Apr 6, 2026

Commits on Apr 7, 2026

Add `CoupledTrainStepper._accumulate_loss()`
jpdunc23
committed

Commits on Apr 8, 2026

Commits on Apr 9, 2026

Commits on Apr 13, 2026

Commits on Apr 15, 2026

Commits on Apr 16, 2026

Merge branch 'main' of github.com:ai2cm/ace into feature/coupled-loss-contrib-random-n_steps
jpdunc23
committed