Unit tests + Refactoring + Small inference optimization#28

Merged

PetarStam merged 13 commits into

Apr 7, 2026

Commits on Mar 19, 2026

update unet and tests
Petar Stamenkovic
committed

Commits on Mar 20, 2026

add preconditioning model tests
Petar Stamenkovic
committed

Commits on Mar 23, 2026

add Song UNet tests
Petar Stamenkovic
committed
fix errors in tests
Petar Stamenkovic
committed

Commits on Mar 24, 2026

change var calculation in group norm eval to unbiased to match pytorch functional
Petar Stamenkovic
committed
add tests for nn layers
Petar Stamenkovic
committed

Commits on Mar 31, 2026

refactor training script to bemore readable and extendable
Petar Stamenkovic
committed

Commits on Apr 2, 2026

tests for training and small fixes
Petar Stamenkovic
committed
switch skip_scale fromnumpy to float
Petar Stamenkovic
committed
add torch inference mode context
Petar Stamenkovic
committed
change CI config to santis
Petar Stamenkovic
committed
add anemoi-datasets to container
Petar Stamenkovic
committed
use torch compile by default in inference
Petar Stamenkovic
committed