Add per variable loss to Stepper and Log using TrainAggs#981

Open

Arcomano1234 wants to merge 23 commits intomainfrom

feature/per-channel-loss-train-agg

Commits on Mar 16, 2026

claude first attempt to add per variable loss to TrainAgg
Arcomano1234
committed

Commits on Mar 20, 2026

Commits on Mar 23, 2026

claude losses return 1d vector over channel dim
Arcomano1234
committed

Commits on Mar 24, 2026

claude refactor to no reduce losses
Arcomano1234
committed

Commits on Mar 26, 2026

Merge branch 'main' into feature/losses-return-1d-loss-vector
Arcomano1234
committed