Awesome

Ai2 Climate Emulator

This repo contains code accompanying "ACE: A fast, skillful learned global atmospheric model for climate prediction" (arxiv:2310.02074) and "Application of the Ai2 Climate Emulator to E3SMv2's global atmosphere model, with a focus on precipitation fidelity" (JGR-ML).

Installation

pip install fme

Documentation

See complete documentation here and a quickstart guide here.

Model checkpoints

Pretrained model checkpoints are available in the ACE Hugging Face collection.

Available datasets

Two versions of the complete dataset described in arxiv:2310.02074 are available on a requester pays Google Cloud Storage bucket:

gs://ai2cm-public-requester-pays/2023-11-29-ai2-climate-emulator-v1/data/repeating-climSST-1deg-zarrs
gs://ai2cm-public-requester-pays/2023-11-29-ai2-climate-emulator-v1/data/repeating-climSST-1deg-netCDFs

The zarr format is convenient for ad-hoc analysis. The netCDF version contains our train/validation split which was used for training and inference.

The datasets used in the forthcoming ACE2 paper are available at:

gs://ai2cm-public-requester-pays/2024-11-13-ai2-climate-emulator-v2-amip/data/c96-1deg-shield/
gs://ai2cm-public-requester-pays/2024-11-13-ai2-climate-emulator-v2-amip/data/era5-1deg-1940-2022.zarr/