ClimaX is a foundation model for weather and climate science. It can apply to various downstream tasks ranging from weather forecasting to climate projections as described in the following paper https://arxiv.org/pdf/2301.10343.pdf. It supports efficient scalable distributed training, powered by PyTorch Lightning. For more details, please visit https://microsoft.github.io/ClimaX

The ClimaX module is available under the project dk92.  

You can access it via the following commands

module use /g/data/dk92/apps/Modules/modulefiles/
module load climax/0.2.3

NCI ClimaX Examples

Below are some introductory tutorial examples for running ClimaX at NCI.

• ClimaX Training on Weatherbench data
• ClimaX: Prediction on Weatherbench data