├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yaml ├── LICENSE ├── PrithviWxC ├── __init__.py ├── config_files │ ├── config.yaml │ └── small.yaml ├── configs.py ├── dataloaders │ ├── __init__.py │ ├── merra2.py │ └── merra2_rollout.py ├── definitions.py ├── download.py ├── model.py └── rollout.py ├── README.md ├── data └── validation │ ├── validation_data.json │ └── validation_rmse.nc ├── docs ├── Makefile ├── arch_main.png ├── make.bat ├── requirements.txt └── source │ ├── PrithviWxC.dataloaders.rst │ ├── PrithviWxC.rst │ ├── _static │ └── css │ │ └── custom.css │ ├── conf.py │ ├── index.rst │ └── installation.rst ├── examples ├── Prithvi-WxC_input_data.ipynb ├── PrithviWxC_inference.ipynb └── PrithviWxC_rollout.ipynb ├── pyproject.toml └── validation ├── __init__.py ├── config.py ├── config.yaml ├── get_assets.py ├── loss.py ├── reproducibility.py └── validate_prithvi_wxc.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/LICENSE -------------------------------------------------------------------------------- /PrithviWxC/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/__init__.py -------------------------------------------------------------------------------- /PrithviWxC/config_files/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/config_files/config.yaml -------------------------------------------------------------------------------- /PrithviWxC/config_files/small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/config_files/small.yaml -------------------------------------------------------------------------------- /PrithviWxC/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/configs.py -------------------------------------------------------------------------------- /PrithviWxC/dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PrithviWxC/dataloaders/merra2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/dataloaders/merra2.py -------------------------------------------------------------------------------- /PrithviWxC/dataloaders/merra2_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/dataloaders/merra2_rollout.py -------------------------------------------------------------------------------- /PrithviWxC/definitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/definitions.py -------------------------------------------------------------------------------- /PrithviWxC/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/download.py -------------------------------------------------------------------------------- /PrithviWxC/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/model.py -------------------------------------------------------------------------------- /PrithviWxC/rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/PrithviWxC/rollout.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/README.md -------------------------------------------------------------------------------- /data/validation/validation_data.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/data/validation/validation_data.json -------------------------------------------------------------------------------- /data/validation/validation_rmse.nc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/data/validation/validation_rmse.nc -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/arch_main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/arch_main.png -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/PrithviWxC.dataloaders.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/PrithviWxC.dataloaders.rst -------------------------------------------------------------------------------- /docs/source/PrithviWxC.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/PrithviWxC.rst -------------------------------------------------------------------------------- /docs/source/_static/css/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/_static/css/custom.css -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/index.rst -------------------------------------------------------------------------------- /docs/source/installation.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/docs/source/installation.rst -------------------------------------------------------------------------------- /examples/Prithvi-WxC_input_data.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/examples/Prithvi-WxC_input_data.ipynb -------------------------------------------------------------------------------- /examples/PrithviWxC_inference.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/examples/PrithviWxC_inference.ipynb -------------------------------------------------------------------------------- /examples/PrithviWxC_rollout.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/examples/PrithviWxC_rollout.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/pyproject.toml -------------------------------------------------------------------------------- /validation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /validation/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/config.py -------------------------------------------------------------------------------- /validation/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/config.yaml -------------------------------------------------------------------------------- /validation/get_assets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/get_assets.py -------------------------------------------------------------------------------- /validation/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/loss.py -------------------------------------------------------------------------------- /validation/reproducibility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/reproducibility.py -------------------------------------------------------------------------------- /validation/validate_prithvi_wxc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NASA-IMPACT/Prithvi-WxC/HEAD/validation/validate_prithvi_wxc.py --------------------------------------------------------------------------------