[2370][model] Make the EncoderModule self-contained#2372

Draft

Tewson1 wants to merge 13 commits into

ecmwf:developfrom

MeteoSwiss:tsivam/dev/iss2370_encoder

Commits on May 18, 2026

Commits on May 19, 2026

clean up code formatting in encoder, engines, and model modules
Tharmeekan
committed
Merge branch 'develop' into tsivam/dev/iss2370_encoder
Tharmeekan
committed
Bug fixes
Tharmeekan
committed
Fix FSDP2 sharding bug by converting non-trainable EncoderModule tensors to buffers
Tharmeekan
committed

Commits on May 20, 2026

Add pipeline configuration for testing; update Trainer to handle plain vectors/buffers like pe_embed
Tharmeekan
committed

Commits on May 21, 2026

Fix load_model when plain tensor
Tharmeekan
committed

Commits on May 26, 2026

Fix load_model to transfer full_tensor
Tharmeekan
committed

Commits on May 31, 2026

Merge remote-tracking branch 'origin/develop' into tsivam/dev/iss2370_encoder
Tharmeekan
committed

Commits on Jun 3, 2026

Update dependencies/ bug fixes
Tharmeekan
committed
Delete config/pipeline.yml
Tewson1
authored