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).
pip install fme
See complete documentation here and a quickstart guide here.
Pretrained model checkpoints are available in the ACE Hugging Face collection.
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/