Skip to content

ai2cm/ace

Repository files navigation

Docs PyPI

ACE: 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).

Documentation

See complete documentation here.

Quickstart

A quickstart guide may be found here.

Model checkpoints

The trained ACE checkpoint and a 1-year subsample of the validation dataset used in arxiv:2310.02074 are available in this Zenodo repository. The checkpoint trained on the E3SMv2 model and the corresponding dataset are available here.

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.