Aurora
Aurora is a machine learning model that can predict atmospheric variables, such as temperature. It is a foundation model, which means that it was first generally trained on a lot of data, and then can adapted to specialised atmospheric forecasting tasks with relatively little data.
They provide four such specialised versions: one for medium-resolution weather prediction, one for high-resolution weather prediction, one for air pollution prediction, and one for ocean wave prediction.
Check their Model Card here
module load gfms
module load Aurora
With using python, the model could then be used (as example below-from Aurora Document)
>>> from aurora import Aurora
>>> model = Aurora()
>>> from aurora import AuroraSmallPretrained
>>> model = AuroraSmallPretrained()
>>> model.load_checkpoint("microsoft/aurora", "aurora-0.25-small-pretrained.ckpt")