ACE: Ai2 Climate Emulator

This repo contains code accompanying "ACE: A fast, skillful learned global atmospheric model for climate prediction" (arxiv:2310.02074).

Documentation

See complete documentation here.

Quickstart

A quickstart guide may be found here.

Model checkpoint

The trained ACE checkpoint and a 1-year subsample of the validation dataset is available in this Zenodo repository.

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.