lars c09ebd2410 Merge pull request 'Add class-balanced secondary particle-type loss (gitea #44)' (#70) from fix/issue-44 into master ...

Reviewed-on: #70

2026-08-18 10:41:42 +02:00

..

analysis

Offset event_id across multi-shard reference reads in giant analyze (gitea #22)

2026-08-17 15:53:02 +02:00

data

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

model

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

tools

Let dwarf warm-cache take --config so it can't under-warm a config's cache keys (gitea #59)

2026-08-17 08:54:14 +02:00

training

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

__init__.py

Initial commit: giant surrogate model with two-phase roadmap in README

2026-06-17 09:29:19 +02:00

_migration.py

Unify the two v0.2->v0.3 migration surfaces (issues.md Issue 6)

2026-08-13 09:56:22 +02:00

checkpoint_io.py

Decouple secondary-species vocabulary from conditioning.particle.emb_dim (gitea #29)

2026-08-13 16:11:14 +02:00

cli.py

Add bf16 autocast to the training loop (gitea #47)

2026-08-17 15:36:12 +02:00

cond_layout.py

Give the cond_cat/cond_cont column layout one owner (gitea #37)

2026-08-14 14:16:20 +02:00

config.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

constants.py

Condition on material/particle physical properties instead of learned embeddings

2026-07-17 15:12:54 +02:00

geometry.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

materials.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

particles.py

Decouple secondary-species vocabulary from conditioning.particle.emb_dim (gitea #29)

2026-08-13 16:11:14 +02:00

pipeline.py

Add per-stage init_from/freeze (gitea #42)

2026-08-17 14:23:20 +02:00

rollout.py

Implement n_sec.mode = "stop_token" for the AR secondary decoder (gitea #40)

2026-08-17 10:44:14 +02:00

sample.py

Implement n_sec.mode = "stop_token" for the AR secondary decoder (gitea #40)

2026-08-17 10:44:14 +02:00

validate.py

Implement n_sec.mode = "stop_token" for the AR secondary decoder (gitea #40)

2026-08-17 10:44:14 +02:00