Repository navigation

Internal ensembling in Aurora models - #199

Merged

Wessel (wesselb) merged 24 commits into

microsoft:mainfrom

Slowika:agaslowik/ensemble-members-internal-to-aurora

Oct 14, 2026

Merged

Internal ensembling in Aurora models#199
Wessel (wesselb) merged 24 commits into
microsoft:mainfrom
Slowika:agaslowik/ensemble-members-internal-to-aurora

Commits on Aug 13, 2026
Commits on Aug 27, 2026
- Address wesselb feedback.
  Agnieszka Słowik (Slowika)
  committed
  aa60f79
Commits on Sep 6, 2026
- Update `rollout_ensemble` docstring.
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  225ef3a
- Remove stale `try/except` block.
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  ecce6df
- Add input validation to `rollout_ensemble`.
  Agnieszka Słowik (Slowika)
  committed
  ff7db21
- Update `models.md` doc.
  Agnieszka Słowik (Slowika)
  committed
  4312687
- Update test to demonstrate an exception is raised in `rollout_ensemble`.
  Agnieszka Słowik (Slowika)
  committed
  f7a5783
Commits on Sep 12, 2026
- Address CI failures in test_rollout_ensemble_num_ensemble_members_one_raises.
  Agnieszka Słowik (Slowika)
  committed
  4d449f0
Commits on Sep 15, 2026
- Docs and API.
  Agnieszka Słowik (Slowika)
  committed
  6297886
- Simplify tests.
  Agnieszka Słowik (Slowika)
  committed
  a584479
Commits on Oct 8, 2026
- Add divisibility check and docstring update.
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  59b755a
- Clean up docstring (reference to `split_batch`)
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  91102b3
- Update docs/models.md with an anchor for the Internal Ensembling section
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  e45927e
- Update docs/models.md
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  a549825
- Simplify test_ensemble.py imports
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  6a46de1
- Add divisibility check test
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  cfd3a98
- Update comments in test_ensemble
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  9d62e6c
- Simplify rollout_ensemble test of raising
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  bd7f3bb
- Merge branch 'main' into agaslowik/ensemble-members-internal-to-aurora
  Agnieszka Słowik (Slowika)
  authored
  a23e908
Commits on Oct 14, 2026
- Update aurora/rollout.py
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  64d2e57
- Update docs/models.md
  
  Agnieszka Słowik (Slowika)
  and
  Wessel (wesselb)
  authored
  f5cb3ec