.gitattributes
.gitignore
.readthedocs.yaml
CHANGELOG.md
CONTRIBUTORS.md
LICENSE
README.md
pyproject.toml
pytest.ini
docs/Makefile
docs/conf.py
docs/contributing.rst
docs/create-migrations.rst
docs/index.rst
docs/_static/anemoi-models_schematic.drawio
docs/_static/anemoi-models_schematic.png
docs/_static/data_indices.drawio
docs/_static/data_indices.png
docs/_static/logo.png
docs/_static/style.css
docs/_templates/.gitkeep
docs/cli/migration.rst
docs/introduction/installing.rst
docs/introduction/overview.rst
docs/modules/activations.rst
docs/modules/data_indices.rst
docs/modules/distributed.rst
docs/modules/interface.rst
docs/modules/layers.rst
docs/modules/migrations.rst
docs/modules/models.rst
docs/modules/normalization.rst
docs/modules/preprocessing.rst
docs/modules/schemas.rst
docs/usage/create_model.rst
src/anemoi/models/__init__.py
src/anemoi/models/__main__.py
src/anemoi/models/_version.py
src/anemoi/models/commands/__init__.py
src/anemoi/models/commands/hello.py
src/anemoi/models/commands/migration.py
src/anemoi/models/data_indices/__init__.py
src/anemoi/models/data_indices/collection.py
src/anemoi/models/data_indices/index.py
src/anemoi/models/data_indices/tensor.py
src/anemoi/models/distributed/__init__.py
src/anemoi/models/distributed/graph.py
src/anemoi/models/distributed/khop_edges.py
src/anemoi/models/distributed/primitives.py
src/anemoi/models/distributed/shapes.py
src/anemoi/models/distributed/transformer.py
src/anemoi/models/distributed/utils.py
src/anemoi/models/interface/__init__.py
src/anemoi/models/layers/__init__.py
src/anemoi/models/layers/activations.py
src/anemoi/models/layers/attention.py
src/anemoi/models/layers/block.py
src/anemoi/models/layers/bounding.py
src/anemoi/models/layers/chunk.py
src/anemoi/models/layers/conv.py
src/anemoi/models/layers/diffusion.py
src/anemoi/models/layers/ensemble.py
src/anemoi/models/layers/graph.py
src/anemoi/models/layers/mapper.py
src/anemoi/models/layers/mlp.py
src/anemoi/models/layers/normalization.py
src/anemoi/models/layers/processor.py
src/anemoi/models/layers/truncation.py
src/anemoi/models/layers/utils.py
src/anemoi/models/migrations/__init__.py
src/anemoi/models/migrations/migrator.py
src/anemoi/models/migrations/setup_context.py
src/anemoi/models/migrations/scripts/1755530253_initial.py
src/anemoi/models/models/__init__.py
src/anemoi/models/models/base.py
src/anemoi/models/models/diffusion_encoder_processor_decoder.py
src/anemoi/models/models/encoder_processor_decoder.py
src/anemoi/models/models/ens_encoder_processor_decoder.py
src/anemoi/models/models/hierarchical.py
src/anemoi/models/models/interpolator.py
src/anemoi/models/preprocessing/__init__.py
src/anemoi/models/preprocessing/imputer.py
src/anemoi/models/preprocessing/mappings.py
src/anemoi/models/preprocessing/normalizer.py
src/anemoi/models/preprocessing/postprocessor.py
src/anemoi/models/preprocessing/remapper.py
src/anemoi/models/samplers/__init__.py
src/anemoi/models/samplers/diffusion_samplers.py
src/anemoi/models/schemas/__init__.py
src/anemoi/models/schemas/common_components.py
src/anemoi/models/schemas/data_processor.py
src/anemoi/models/schemas/decoder.py
src/anemoi/models/schemas/encoder.py
src/anemoi/models/schemas/models.py
src/anemoi/models/schemas/processor.py
src/anemoi_models.egg-info/PKG-INFO
src/anemoi_models.egg-info/SOURCES.txt
src/anemoi_models.egg-info/dependency_links.txt
src/anemoi_models.egg-info/entry_points.txt
src/anemoi_models.egg-info/requires.txt
src/anemoi_models.egg-info/top_level.txt
tests/data_indices/test_collection.py
tests/data_indices/test_data_indices.py
tests/layers/test_activations.py
tests/layers/test_attention.py
tests/layers/test_bounding.py
tests/layers/test_graph.py
tests/layers/test_layer_utils.py
tests/layers/test_mlp.py
tests/layers/test_noise_embeddings.py
tests/layers/block/test_block_graphconv.py
tests/layers/block/test_block_graphtransformer.py
tests/layers/block/test_block_pointwise.py
tests/layers/block/test_block_transformer.py
tests/layers/block/test_block_transformermapper.py
tests/layers/chunk/test_chunk_gnn.py
tests/layers/chunk/test_chunk_graphtransformer.py
tests/layers/chunk/test_chunk_pointwise.py
tests/layers/chunk/test_chunk_transformer.py
tests/layers/mapper/test_base_mapper.py
tests/layers/mapper/test_graphconv_mapper.py
tests/layers/mapper/test_graphtransformer_mapper.py
tests/layers/mapper/test_transformer_mapper.py
tests/layers/processor/test_base_processor.py
tests/layers/processor/test_graphconv_processor.py
tests/layers/processor/test_graphtransformer_processor.py
tests/layers/processor/test_pointwise_processor.py
tests/layers/processor/test_transformer_processor.py
tests/migrations/conftest.py
tests/migrations/test_migration_order.py
tests/migrations/test_migrations.py
tests/migrations/migrations/1750840837_add_foo.py
tests/migrations/migrations/1750841219_add_bar.py
tests/migrations/migrations/1750859824_add_baz.py
tests/migrations/migrations/1750859905_rename_baz.py
tests/migrations/migrations/1751895180_final.py
tests/migrations/migrations/1751895203_recent.py
tests/models/test_models.py
tests/preprocessing/test_postprocessor.py
tests/preprocessing/test_preprocessor_imputer.py
tests/preprocessing/test_preprocessor_normalizer.py
tests/preprocessing/test_preprocessor_remapper.py
tests/samplers/test_diffusion_samplers.py
tests/schemas/test_data_processors_schemas.py