Train progress bar#84

Open

maja-jablonska wants to merge 17 commits into

andycasey:masterfrom

maja-jablonska:train-progress-bar

Commits on Jun 5, 2026

Port The Cannon's numerical core to JAX

maja-jablonska
and
claude
committed

Commits on Jun 8, 2026

Add progress bar to train() and keep trained arrays as JAX

maja-jablonska
and
claude
committed
Skip the optimizer at regularization=0 via closed-form solve

maja-jablonska
and
claude
committed
Add cross-validated Cannon sweep with offline W&B and PBS jobs

maja-jablonska
and
claude
committed
Add end-to-end train/validate script (from start.ipynb)

maja-jablonska
and
claude
committed
Add run_sweep driver and point the PBS job at it

maja-jablonska
and
claude
committed
Build label sets from base + age variants + abundances

maja-jablonska
and
claude
committed

Commits on Jun 9, 2026