.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CHANGELOG.md
LICENSE
README.md
pyproject.toml
.github/ci-config.yml
.github/workflows/changelog-pr-update.yml
.github/workflows/changelog-release-update.yml
.github/workflows/ci.yml
.github/workflows/label-public-pr.yml
.github/workflows/python-publish.yml
.github/workflows/readthedocs-pr-update.yml
docs/Makefile
docs/conf.py
docs/index.rst
docs/installing.rst
docs/overview.rst
docs/_static/logo.png
docs/_static/style.css
docs/_templates/.gitkeep
docs/modules/data_indices.rst
docs/modules/distributed.rst
docs/modules/interface.rst
docs/modules/layers.rst
docs/modules/models.rst
docs/modules/preprocessing.rst
docs/schemas/anemoi-models_schematic.drawio
docs/schemas/anemoi-models_schematic.png
docs/schemas/data_indices.drawio
docs/schemas/data_indices.png
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/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/attention.py
src/anemoi/models/layers/block.py
src/anemoi/models/layers/chunk.py
src/anemoi/models/layers/conv.py
src/anemoi/models/layers/graph.py
src/anemoi/models/layers/mapper.py
src/anemoi/models/layers/mlp.py
src/anemoi/models/layers/processor.py
src/anemoi/models/layers/utils.py
src/anemoi/models/models/__init__.py
src/anemoi/models/models/encoder_processor_decoder.py
src/anemoi/models/preprocessing/__init__.py
src/anemoi/models/preprocessing/imputer.py
src/anemoi/models/preprocessing/normalizer.py
src/anemoi/models/preprocessing/remapper.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_attention.py
tests/layers/test_graph.py
tests/layers/test_mlp.py
tests/layers/block/test_block_graphconv.py
tests/layers/block/test_block_graphtransformer.py
tests/layers/block/test_block_transformer.py
tests/layers/chunk/test_chunk_gnn.py
tests/layers/chunk/test_chunk_graphtransformer.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/processor/test_base_processor.py
tests/layers/processor/test_graphconv_processor.py
tests/layers/processor/test_graphtransformer_processor.py
tests/layers/processor/test_transformer_processor.py
tests/models/test_models.py
tests/preprocessing/test_preprocessor_imputer.py
tests/preprocessing/test_preprocessor_normalizer.py
tests/preprocessing/test_preprocessor_remapper.py