lars f3fec8bcb3 Add ComposedRouter for multi-axis MoE gating ...

Route on several independent axes at once (e.g. energy x pdg), each with
its own expert count and hyperparameters. The joint gate is the outer
product of per-axis softmax gates, so it stays a partition of unity and
top1/balance_loss factor per-axis. Config uses flat axis{i}_{field} keys
in model.router (TOML/CLI friendly), also settable via repeatable
--router-axis flags.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

2026-07-15 13:53:21 +02:00

..

test_analysis.py

Fix rollout edep mismatch and add truth overlay to Tier 4 observables

2026-07-13 12:38:44 +02:00

test_bump_dataset_version.py

Fix ruff, ty, and pytest failures; apply ruff format

2026-07-02 16:59:22 +02:00

test_cli_predict.py

Add --comment option to predict, recorded in YAML sidecar

2026-07-06 09:37:48 +02:00

test_config.py

Apply ruff format and document lint/type tooling in CLAUDE.md

2026-06-18 17:40:27 +02:00

test_create_root_files.py

Support --energy-gev in dwarf make-root for the new minicalosim energy arg

2026-07-13 13:59:09 +02:00

test_dataset.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_dwarf.py

Unify dataset/tooling scripts into a single dwarf Typer CLI

2026-07-02 09:52:37 +02:00

test_flow.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_geometry.py

Add fast slab lookup for the GeometryOracle, replacing knn as the default

2026-07-08 11:42:12 +02:00

test_loader.py

Apply ruff format after merging phase2-secondary-prediction

2026-07-15 10:00:36 +02:00

test_network.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_phase2.py

Rescale secondary energies to exactly consume the e_sec budget

2026-07-13 13:45:13 +02:00

test_rollout.py

Stream giant rollout output instead of buffering the whole run

2026-07-15 11:10:04 +02:00

test_router.py

Add ComposedRouter for multi-axis MoE gating

2026-07-15 13:53:21 +02:00

test_steps_to_parquet_parallel.py

Fix ruff, ty, and pytest failures; apply ruff format

2026-07-02 16:59:22 +02:00

test_steps_to_parquet.py

Apply ruff format after merging phase2-secondary-prediction

2026-07-15 10:00:36 +02:00

test_transforms.py

Apply ruff format after merging phase2-secondary-prediction

2026-07-15 10:00:36 +02:00