.gitattributes
.gitignore
.readthedocs.yaml
CHANGELOG.md
CONTRIBUTORS.md
LICENSE
README.md
pyproject.toml
pytest.ini
docs/Makefile
docs/conf.py
docs/contributing.rst
docs/index.rst
docs/troubleshooting.rst
docs/_static/logo.png
docs/_static/style.css
docs/_templates/.gitkeep
docs/adrs/adr-001.md
docs/adrs/adr-002.md
docs/adrs/template.md
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/performance-guide/mem-snapshot-1-mapper-chunk.png
docs/images/performance-guide/mem-snapshot-4-mapper-chunks.png
docs/images/performance-guide/performance-flowchart.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/introduction/installing.rst
docs/introduction/overview.rst
docs/modules/data.rst
docs/modules/diagnostics.rst
docs/modules/losses.rst
docs/modules/optimizers.rst
docs/modules/schemas.rst
docs/modules/strategy.rst
docs/modules/train.rst
docs/user-guide/basic-set-up.rst
docs/user-guide/benchmarking.rst
docs/user-guide/configuring.rst
docs/user-guide/diffusion-set-up.rst
docs/user-guide/distributed.rst
docs/user-guide/download-era5-o96.rst
docs/user-guide/hydra-intro.rst
docs/user-guide/kcrps-set-up.rst
docs/user-guide/models.rst
docs/user-guide/overview.rst
docs/user-guide/performance-optimisation.rst
docs/user-guide/tracking.rst
docs/user-guide/training.rst
docs/user-guide/yaml/dataloader.yaml
docs/user-guide/yaml/example_crps_config.yaml
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/diffusion.yaml
src/anemoi/training/config/ensemble_crps.yaml
src/anemoi/training/config/hierarchical.yaml
src/anemoi/training/config/interpolator.yaml
src/anemoi/training/config/lam.yaml
src/anemoi/training/config/point_wise.yaml
src/anemoi/training/config/stretched.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/evaluation_ens.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/placeholder.yaml
src/anemoi/training/config/diagnostics/callbacks/rollout_eval.yaml
src/anemoi/training/config/diagnostics/plot/detailed.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/existing.yaml
src/anemoi/training/config/graph/existing_graph.yaml
src/anemoi/training/config/graph/hierarchical_2level.yaml
src/anemoi/training/config/graph/hierarchical_3level.yaml
src/anemoi/training/config/graph/limited_area.yaml
src/anemoi/training/config/graph/multi_scale.yaml
src/anemoi/training/config/graph/point_wise.yaml
src/anemoi/training/config/graph/stretched_grid.yaml
src/anemoi/training/config/model/gnn.yaml
src/anemoi/training/config/model/graphtransformer.yaml
src/anemoi/training/config/model/graphtransformer_diffusion.yaml
src/anemoi/training/config/model/graphtransformer_diffusiontend.yaml
src/anemoi/training/config/model/graphtransformer_ens.yaml
src/anemoi/training/config/model/point_wise.yaml
src/anemoi/training/config/model/transformer.yaml
src/anemoi/training/config/model/transformer_diffusion.yaml
src/anemoi/training/config/model/transformer_diffusiontend.yaml
src/anemoi/training/config/model/transformer_ens.yaml
src/anemoi/training/config/model/transformer_transformermapper.yaml
src/anemoi/training/config/system/example.yaml
src/anemoi/training/config/system/slurm.yaml
src/anemoi/training/config/system/hardware/example.yaml
src/anemoi/training/config/system/hardware/slurm.yaml
src/anemoi/training/config/system/input/example.yaml
src/anemoi/training/config/system/output/example.yaml
src/anemoi/training/config/training/default.yaml
src/anemoi/training/config/training/diffusion.yaml
src/anemoi/training/config/training/ensemble.yaml
src/anemoi/training/config/training/interpolator.yaml
src/anemoi/training/config/training/lam.yaml
src/anemoi/training/config/training/stretched.yaml
src/anemoi/training/config/training/scalers/global.yaml
src/anemoi/training/config/training/scalers/lam.yaml
src/anemoi/training/config/training/scalers/stretched.yaml
src/anemoi/training/data/__init__.py
src/anemoi/training/data/datamodule.py
src/anemoi/training/data/dataset.py
src/anemoi/training/data/grid_indices.py
src/anemoi/training/diagnostics/__init__.py
src/anemoi/training/diagnostics/benchmark_server.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/plot_ens.py
src/anemoi/training/diagnostics/callbacks/profiler.py
src/anemoi/training/diagnostics/callbacks/provenance.py
src/anemoi/training/diagnostics/callbacks/sanity.py
src/anemoi/training/diagnostics/callbacks/stopping.py
src/anemoi/training/diagnostics/mlflow/__init__.py
src/anemoi/training/diagnostics/mlflow/azureml.py
src/anemoi/training/diagnostics/mlflow/logger.py
src/anemoi/training/diagnostics/mlflow/utils.py
src/anemoi/training/diagnostics/mlflow/system_metrics/cpu_monitor.py
src/anemoi/training/diagnostics/mlflow/system_metrics/gpu_monitor.py
src/anemoi/training/distributed/__init__.py
src/anemoi/training/distributed/strategy.py
src/anemoi/training/losses/__init__.py
src/anemoi/training/losses/base.py
src/anemoi/training/losses/combined.py
src/anemoi/training/losses/filtering.py
src/anemoi/training/losses/huber.py
src/anemoi/training/losses/kcrps.py
src/anemoi/training/losses/logcosh.py
src/anemoi/training/losses/loss.py
src/anemoi/training/losses/mae.py
src/anemoi/training/losses/mse.py
src/anemoi/training/losses/multiscale.py
src/anemoi/training/losses/rmse.py
src/anemoi/training/losses/scaler_tensor.py
src/anemoi/training/losses/spatial.py
src/anemoi/training/losses/utils.py
src/anemoi/training/losses/weighted_mse.py
src/anemoi/training/losses/scalers/__init__.py
src/anemoi/training/losses/scalers/base_scaler.py
src/anemoi/training/losses/scalers/loss_weights_mask.py
src/anemoi/training/losses/scalers/node_attributes.py
src/anemoi/training/losses/scalers/scalers.py
src/anemoi/training/losses/scalers/variable.py
src/anemoi/training/losses/scalers/variable_level.py
src/anemoi/training/losses/scalers/variable_masking.py
src/anemoi/training/losses/scalers/variable_tendency.py
src/anemoi/training/optimizers/AdEMAMix.py
src/anemoi/training/schemas/__init__.py
src/anemoi/training/schemas/base_schema.py
src/anemoi/training/schemas/data.py
src/anemoi/training/schemas/dataloader.py
src/anemoi/training/schemas/diagnostics.py
src/anemoi/training/schemas/system.py
src/anemoi/training/schemas/training.py
src/anemoi/training/train/__init__.py
src/anemoi/training/train/profiler.py
src/anemoi/training/train/train.py
src/anemoi/training/train/tasks/__init__.py
src/anemoi/training/train/tasks/base.py
src/anemoi/training/train/tasks/diffusionforecaster.py
src/anemoi/training/train/tasks/ensforecaster.py
src/anemoi/training/train/tasks/forecaster.py
src/anemoi/training/train/tasks/interpolator.py
src/anemoi/training/train/tasks/rollout.py
src/anemoi/training/utils/__init__.py
src/anemoi/training/utils/checkpoint.py
src/anemoi/training/utils/custom_colormaps.py
src/anemoi/training/utils/enums.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/utils/variables_metadata.py
src/anemoi/training/utils/worker_init.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/integration/conftest.py
tests/integration/test_benchmark.py
tests/integration/test_training_cycle.py
tests/integration/aicon/test_cicd_aicon_04_icon-dream_medium.py
tests/integration/aicon/test_cicd_aicon_04_icon-dream_medium.yaml
tests/integration/config/imputer_modifications.yaml
tests/integration/config/test_config.yaml
tests/integration/config/test_diffusion.yaml
tests/integration/config/test_ensemble_crps.yaml
tests/integration/config/test_interpolator.yaml
tests/integration/config/test_lam.yaml
tests/integration/config/test_stretched.yaml
tests/integration/config/testing_modifications.yaml
tests/integration/config/benchmark/base.yaml
tests/integration/config/benchmark/ensemble_crps.yaml
tests/integration/config/benchmark/graphtransformer.yaml
tests/integration/config/benchmark/lam.yaml
tests/integration/config/benchmark/stretched.yaml
tests/unit/conftest.py
tests/unit/requirements.txt
tests/unit/commands/test_config.py
tests/unit/commands/test_mlflow.py
tests/unit/diagnostics/test_callbacks.py
tests/unit/diagnostics/test_checkpoint.py
tests/unit/diagnostics/callbacks/test_timelimit.py
tests/unit/diagnostics/callbacks/test_variable_order.py
tests/unit/diagnostics/mlflow/test_azureml_mlflow_logger.py
tests/unit/diagnostics/mlflow/test_mlflow_logger.py
tests/unit/diagnostics/mlflow/test_mlflow_utils.py
tests/unit/hydra/test_search_path_plugins.py
tests/unit/losses/test_combined_loss.py
tests/unit/losses/test_filtered_loss.py
tests/unit/losses/test_loss_function.py
tests/unit/losses/test_loss_scaling.py
tests/unit/losses/test_multiscale_loss.py
tests/unit/losses/test_scaler.py
tests/unit/schemas/test_expand_paths.py
tests/unit/train/test_optimizer.py
tests/unit/train/test_print_variable_scaling.py
tests/unit/train/test_restarting_run.py
tests/unit/utils/test_masks.py
tests/unit/utils/test_usable_indices.py
tests/unit/utils/test_variable_grouping.py