.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
LICENSE
README.md
RELEASE.md
mkdocs.yaml
poetry.lock
pyproject.toml
uv.lock
.github/CODEOWNERS
.github/workflows/functional_tests.yml
.github/workflows/lint.yml
.github/workflows/package.yml
.github/workflows/publish-final-release.yml
.github/workflows/publish-release-candidate.yml
.github/workflows/unit_tests.yml
docs/index.md
docs/archesweather/evaluate.md
docs/archesweather/index.md
docs/archesweather/reproduce.md
docs/archesweather/run.ipynb
docs/archesweather/setup.md
docs/assets/extra.css
docs/assets/logo.png
docs/contributing/bug.md
docs/contributing/contribute.md
docs/contributing/index.md
docs/contributing/testing.md
docs/getting_started/installation.md
docs/getting_started/using_geoarches.md
docs/user_guide/api.md
docs/user_guide/custom_models.md
docs/user_guide/evaluate.md
docs/user_guide/index.md
docs/user_guide/train.md
geoarches/__init__.py
geoarches/main_hydra.py
geoarches/submit.py
geoarches.egg-info/PKG-INFO
geoarches.egg-info/SOURCES.txt
geoarches.egg-info/dependency_links.txt
geoarches.egg-info/entry_points.txt
geoarches.egg-info/requires.txt
geoarches.egg-info/scm_file_list.json
geoarches.egg-info/scm_version.json
geoarches.egg-info/top_level.txt
geoarches/backbones/archesweather.py
geoarches/backbones/archesweather_layers.py
geoarches/backbones/dit.py
geoarches/backbones/weatherlearn_utils/README.md
geoarches/backbones/weatherlearn_utils/crop.py
geoarches/backbones/weatherlearn_utils/earth_position_index.py
geoarches/backbones/weatherlearn_utils/pad.py
geoarches/backbones/weatherlearn_utils/patch_embed.py
geoarches/backbones/weatherlearn_utils/patch_recovery.py
geoarches/backbones/weatherlearn_utils/shift_window_mask.py
geoarches/configs/__init__.py
geoarches/configs/config.yaml
geoarches/configs/cluster/example-slurm.yaml
geoarches/configs/cluster/local.yaml
geoarches/configs/dataloader/dcpp.yaml
geoarches/configs/dataloader/era5.yaml
geoarches/configs/dataloader/era5pred.yaml
geoarches/configs/module/archesweather-diffusion.yaml
geoarches/configs/module/archesweather.yaml
geoarches/configs/module/archesweathergen.yaml
geoarches/configs/module/metrics/era5_brier.yaml
geoarches/configs/module/metrics/era5_deterministic.yaml
geoarches/configs/module/metrics/era5_ensemble.yaml
geoarches/dataloaders/dcpp.py
geoarches/dataloaders/era5.py
geoarches/dataloaders/era5pred.py
geoarches/dataloaders/netcdf.py
geoarches/dataloaders/zarr.py
geoarches/download/dl_aw_models.sh
geoarches/download/dl_era.py
geoarches/download/dl_forecast.py
geoarches/download/dl_hres.py
geoarches/evaluation/compute_quantiles.py
geoarches/evaluation/eval_multistep.py
geoarches/evaluation/metric_registry.py
geoarches/evaluation/plot.py
geoarches/evaluation/submit_eval.sh
geoarches/inference/encode_dataset.py
geoarches/lightning_modules/__init__.py
geoarches/lightning_modules/base_module.py
geoarches/lightning_modules/diffusion.py
geoarches/lightning_modules/diffusion_ensemble.py
geoarches/lightning_modules/forecast.py
geoarches/metrics/brier_skill_score.py
geoarches/metrics/deterministic_metrics.py
geoarches/metrics/deterministic_metrics_legacy.py
geoarches/metrics/ensemble_metrics.py
geoarches/metrics/label_wrapper.py
geoarches/metrics/metric_base.py
geoarches/metrics/rank_histogram.py
geoarches/metrics/spherical_power_spectrum.py
geoarches/stats/__init__.py
geoarches/stats/archesweather_constant_masks.pt
geoarches/stats/climatology_metrics.pt
geoarches/stats/dcpp_spatial_norm_stats.pt
geoarches/stats/delta24_stats.pt
geoarches/stats/delta24_stats_with_w.pt
geoarches/stats/deltapred24_aws_denorm.pt
geoarches/stats/pangu_norm_stats2.pt
geoarches/stats/pangu_norm_stats2_with_w.pt
geoarches/utils/tensordict_utils.py
geoarches/utils/visualization.py
paper/configs/archesweather-m-seed0.yaml
paper/configs/archesweather-m-seed1.yaml
paper/configs/archesweather-m-skip-seed0.yaml
paper/configs/archesweather-m-skip-seed1.yaml
paper/configs/archesweathergen.yaml
paper/results/README.md
paper/results/gencast_0p25deg/test-multistep=10-era5_brier_skill_score.nc
paper/results/gencast_0p25deg/test-multistep=10-era5_ensemble_metrics.nc
paper/results/gencast_0p25deg/test-multistep=10-era5_rank_histogram_50_members.nc
paper/results/gencast_1deg/test-multistep=10-era5_brier_skill_score.nc
paper/results/gencast_1deg/test-multistep=10-era5_ensemble_metrics.nc
paper/results/gencast_1deg/test-multistep=10-era5_rank_histogram_50_members.nc
tests/test_stats.py
tests/dataloaders/test_dcpp.py
tests/dataloaders/test_era5.py
tests/evaluation/test_metric_registry.py
tests/functional/target_prediction.pt
tests/functional/test_archesweathergen_real_data.py
tests/functional/test_train_test_workflow.py
tests/metrics/test_brier_skill_score.py
tests/metrics/test_ensemble_metrics.py
tests/metrics/test_label_wrapper.py
tests/metrics/test_metric_base.py
tests/metrics/test_rank_histogram.py
tests/metrics/test_spherical_power_spectrum.py