.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/residual.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/balanced_partition.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/conv.py
src/anemoi/models/layers/diffusion.py
src/anemoi/models/layers/ensemble.py
src/anemoi/models/layers/graph.py
src/anemoi/models/layers/graph_provider.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/residual.py
src/anemoi/models/layers/sparse_projector.py
src/anemoi/models/layers/spectral_helpers.py
src/anemoi/models/layers/spectral_transforms.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/migrations/scripts/1762857427_deprecate_eda.py
src/anemoi/models/migrations/scripts/1762857428_chunking_fix.py
src/anemoi/models/migrations/scripts/1763479917_hardware_schema_update.py
src/anemoi/models/migrations/scripts/1763479918_refactor_mapper.py
src/anemoi/models/migrations/scripts/1767108147_move_to_multiple_datasets.py
src/anemoi/models/migrations/scripts/1773048851_fuse_multiple_perdataset_graphs.py
src/anemoi/models/models/__init__.py
src/anemoi/models/models/autoencoder.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/hierarchical_autoencoder.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/schemas/residual.py
src/anemoi/models/triton/gt.py
src/anemoi/models/triton/utils.py
src/anemoi/models/utils/__init__.py
src/anemoi/models/utils/compile.py
src/anemoi/models/utils/config.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/conftest.py
tests/data_indices/test_collection.py
tests/data_indices/test_data_indices.py
tests/distributed/balanced_partition.py
tests/integration/triton/test_triton_gt.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/test_residual.py
tests/layers/test_sht.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/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_diffusion_sampling_pipeline.py
tests/models/test_diffusion_tendency.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/preprocessing/test_stepwise_processors.py
tests/samplers/test_diffusion_samplers.py
tests/schemas/test_data_processors_schemas.py
tests/utils/test_compile.py