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" (preprint).
See complete documentation here.
A quickstart guide may be found here.
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.
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.