├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── LICENSE_FourCastNetv2 ├── README.md ├── ai_models_fourcastnetv2 ├── __init__.py ├── fourcastnetv2 │ ├── __init__.py │ ├── activations.py │ ├── contractions.py │ ├── layers.py │ └── sfnonet.py └── model.py ├── setup.py ├── tests ├── requirements.txt └── test_code.py └── tox.ini /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE_FourCastNetv2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/LICENSE_FourCastNetv2 -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/README.md -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/__init__.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/fourcastnetv2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/fourcastnetv2/__init__.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/fourcastnetv2/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/fourcastnetv2/activations.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/fourcastnetv2/contractions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/fourcastnetv2/contractions.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/fourcastnetv2/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/fourcastnetv2/layers.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/fourcastnetv2/sfnonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/fourcastnetv2/sfnonet.py -------------------------------------------------------------------------------- /ai_models_fourcastnetv2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/ai_models_fourcastnetv2/model.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/setup.py -------------------------------------------------------------------------------- /tests/requirements.txt: -------------------------------------------------------------------------------- 1 | # Empty for now 2 | -------------------------------------------------------------------------------- /tests/test_code.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/tests/test_code.py -------------------------------------------------------------------------------- /tox.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ecmwf-lab/ai-models-fourcastnetv2/HEAD/tox.ini --------------------------------------------------------------------------------