.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CHANGELOG.md
CONTRIBUTORS.md
LICENSE
README.md
pyproject.toml
pytest.ini
.github/CODEOWNERS
.github/ci-config.yml
.github/ci-hpc-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/push-to-private.yml
.github/workflows/python-publish.yml
.github/workflows/python-pull-request.yml
.github/workflows/readtthedocs-pr-update.yml
docs/Makefile
docs/conf.py
docs/index.rst
docs/installing.rst
docs/overview.rst
docs/requirements.txt
docs/_static/logo.png
docs/_static/style.css
docs/_templates/.gitkeep
docs/dev/code-structure.rst
docs/dev/contributing.rst
docs/dev/hydra.rst
docs/dev/testing.rst
docs/images/global-sliding-window-attention.png
docs/images/gnn-encoder-decoder-multimesh.jpg
docs/images/model_sharding.png
docs/images/transformer-block.png
docs/images/mlflow/mlflow_compare.png
docs/images/mlflow/mlflow_constant.png
docs/images/mlflow/mlflow_resumed_run.png
docs/images/mlflow/mlflow_run.png
docs/images/mlflow/mlflow_server.png
docs/images/profiler/anemoi_profiler_architecture.png
docs/images/profiler/anemoi_profiler_benchmark_profiler.png
docs/images/profiler/anemoi_profiler_config.png
docs/images/profiler/anemoi_profiler_high_level.png
docs/images/profiler/anemoi_profiler_mlflow_integration.png
docs/images/profiler/anemoi_profiler_mlflow_integration_2.png
docs/images/profiler/anemoi_profiler_mlflow_integration_3.png
docs/images/profiler/anemoi_profiler_speed_report.png
docs/images/profiler/anemoi_profiler_speedreport_diagram.png
docs/images/profiler/anemoi_profiler_training_rates.png
docs/images/profiler/anemoi_profiler_validation_rates.png
docs/images/profiler/example_memory_report.png
docs/images/profiler/example_memory_timeline.png
docs/images/profiler/example_model_summary.png
docs/images/profiler/example_model_summary_2.png
docs/images/profiler/example_system_report.png
docs/images/profiler/example_time_report.png
docs/images/profiler/idle_time_breakdown.png
docs/images/profiler/kernel_breakdown_dfs.png
docs/images/profiler/kernel_breakdown_plots.png
docs/images/profiler/memory_snapshot_diagram.png
docs/images/profiler/memory_snapshot_output.png
docs/images/profiler/temporal_breakdown.png
docs/modules/data.rst
docs/modules/diagnostics.rst
docs/modules/losses.rst
docs/modules/strategy.rst
docs/modules/train.rst
docs/start/hydra-intro.rst
docs/start/installing.rst
docs/start/training-your-first-model.rst
docs/user-guide/benchmarking.rst
docs/user-guide/configuring.rst
docs/user-guide/debugging.rst
docs/user-guide/distributed.rst
docs/user-guide/introduction.rst
docs/user-guide/models.rst
docs/user-guide/tracking.rst
docs/user-guide/training.rst
src/anemoi/training/__init__.py
src/anemoi/training/__main__.py
src/anemoi/training/_version.py
src/anemoi/training/commands/__init__.py
src/anemoi/training/commands/checkpoint.py
src/anemoi/training/commands/config.py
src/anemoi/training/commands/mlflow.py
src/anemoi/training/commands/profiler.py
src/anemoi/training/commands/train.py
src/anemoi/training/config/__init__.py
src/anemoi/training/config/config.yaml
src/anemoi/training/config/debug.yaml
src/anemoi/training/config/data/zarr.yaml
src/anemoi/training/config/dataloader/native_grid.yaml
src/anemoi/training/config/diagnostics/evaluation.yaml
src/anemoi/training/config/diagnostics/benchmark_profiler/detailed.yaml
src/anemoi/training/config/diagnostics/benchmark_profiler/simple.yaml
src/anemoi/training/config/diagnostics/callbacks/pretraining.yaml
src/anemoi/training/config/diagnostics/callbacks/rollout_eval.yaml
src/anemoi/training/config/diagnostics/plot/detailed.yaml
src/anemoi/training/config/diagnostics/plot/none.yaml
src/anemoi/training/config/diagnostics/plot/rollout_eval.yaml
src/anemoi/training/config/diagnostics/plot/simple.yaml
src/anemoi/training/config/graph/encoder_decoder_only.yaml
src/anemoi/training/config/graph/limited_area.yaml
src/anemoi/training/config/graph/multi_scale.yaml
src/anemoi/training/config/graph/stretched_grid.yaml
src/anemoi/training/config/hardware/example.yaml
src/anemoi/training/config/hardware/slurm.yaml
src/anemoi/training/config/hardware/files/example.yaml
src/anemoi/training/config/hardware/paths/example.yaml
src/anemoi/training/config/model/gnn.yaml
src/anemoi/training/config/model/graphtransformer.yaml
src/anemoi/training/config/model/transformer.yaml
src/anemoi/training/config/training/default.yaml
src/anemoi/training/data/__init__.py
src/anemoi/training/data/datamodule.py
src/anemoi/training/data/dataset.py
src/anemoi/training/data/scaling.py
src/anemoi/training/diagnostics/__init__.py
src/anemoi/training/diagnostics/continents.json
src/anemoi/training/diagnostics/logger.py
src/anemoi/training/diagnostics/maps.py
src/anemoi/training/diagnostics/plots.py
src/anemoi/training/diagnostics/profilers.py
src/anemoi/training/diagnostics/callbacks/__init__.py
src/anemoi/training/diagnostics/callbacks/checkpoint.py
src/anemoi/training/diagnostics/callbacks/evaluation.py
src/anemoi/training/diagnostics/callbacks/optimiser.py
src/anemoi/training/diagnostics/callbacks/plot.py
src/anemoi/training/diagnostics/callbacks/profiler.py
src/anemoi/training/diagnostics/callbacks/provenance.py
src/anemoi/training/diagnostics/mlflow/__init__.py
src/anemoi/training/diagnostics/mlflow/auth.py
src/anemoi/training/diagnostics/mlflow/client.py
src/anemoi/training/diagnostics/mlflow/logger.py
src/anemoi/training/diagnostics/mlflow/utils.py
src/anemoi/training/distributed/__init__.py
src/anemoi/training/distributed/strategy.py
src/anemoi/training/losses/__init__.py
src/anemoi/training/losses/combined.py
src/anemoi/training/losses/huber.py
src/anemoi/training/losses/logcosh.py
src/anemoi/training/losses/mae.py
src/anemoi/training/losses/mse.py
src/anemoi/training/losses/rmse.py
src/anemoi/training/losses/utils.py
src/anemoi/training/losses/weightedloss.py
src/anemoi/training/train/__init__.py
src/anemoi/training/train/forecaster.py
src/anemoi/training/train/profiler.py
src/anemoi/training/train/train.py
src/anemoi/training/utils/__init__.py
src/anemoi/training/utils/checkpoint.py
src/anemoi/training/utils/jsonify.py
src/anemoi/training/utils/masks.py
src/anemoi/training/utils/mlflow_sync.py
src/anemoi/training/utils/seeding.py
src/anemoi/training/utils/usable_indices.py
src/anemoi_training.egg-info/PKG-INFO
src/anemoi_training.egg-info/SOURCES.txt
src/anemoi_training.egg-info/dependency_links.txt
src/anemoi_training.egg-info/entry_points.txt
src/anemoi_training.egg-info/requires.txt
src/anemoi_training.egg-info/top_level.txt
src/hydra_plugins/anemoi_searchpath/__init__.py
src/hydra_plugins/anemoi_searchpath/anemoi_searchpath_plugin.py
tests/conftest.py
tests/requirements.txt
tests/diagnostics/test_callbacks.py
tests/diagnostics/test_checkpoint.py
tests/diagnostics/mlflow/test_auth.py
tests/diagnostics/mlflow/test_client.py
tests/diagnostics/mlflow/test_expansion.py
tests/hydra/test_search_path_plugins.py
tests/train/test_loss_function.py
tests/train/test_loss_scaling.py
tests/train/test_scalar.py
tests/utils/test_usable_indices.py