https://github.com/ai2cm/ace

Ai2 Climate Emulator
https://github.com/ai2cm/ace

Last synced: about 1 month ago
JSON representation

Ai2 Climate Emulator

• Host: GitHub
• URL: https://github.com/ai2cm/ace
• Owner: ai2cm
• License: apache-2.0
• Created: 2023-11-29T23:08:42.000Z (about 1 year ago)
• Default Branch: main
• Last Pushed: 2024-11-18T23:12:57.000Z (about 2 months ago)
• Last Synced: 2024-12-04T10:01:53.668Z (about 1 month ago)
• Language: Python
• Homepage:
• Size: 3.54 MB
• Stars: 56
• Watchers: 5
• Forks: 9
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• open-sustainable-technology - ACE - A 200M-parameter, autoregressive machine learning emulator of an existing comprehensive 100-km resolution global atmospheric model. (Atmosphere / Atmospheric Composition and Dynamics)

README

        
[![Docs](https://readthedocs.org/projects/ai2-climate-emulator/badge/?version=latest)](https://ai2-climate-emulator.readthedocs.io/en/latest/)

[![PyPI](https://img.shields.io/pypi/v/fme.svg)](https://pypi.org/project/fme/)

# ACE: Ai2 Climate Emulator

This repo contains code accompanying "ACE: A fast, skillful learned global atmospheric model for climate prediction" ([arxiv:2310.02074](https://arxiv.org/abs/2310.02074)) and "Application of the Ai2 Climate Emulator to E3SMv2's global atmosphere model, with a focus on precipitation fidelity" ([JGR-ML](https://agupubs.onlinelibrary.wiley.com/doi/full/10.1029/2024JH000136)).

## Documentation

See complete documentation [here](https://ai2-climate-emulator.readthedocs.io/en/latest/).

## Quickstart

A quickstart guide may be found [here](https://ai2-climate-emulator.readthedocs.io/en/latest/quickstart.html).

## Model checkpoints

The trained ACE checkpoint and a 1-year subsample of the validation dataset used in [arxiv:2310.02074](https://arxiv.org/abs/2310.02074) are available in

[this Zenodo repository](https://doi.org/10.5281/zenodo.10791087).

The checkpoint trained on the [E3SMv2](https://agupubs.onlinelibrary.wiley.com/doi/full/10.1029/2024JH000136) model

and the corresponding dataset are available [here](https://portal.nersc.gov/archive/home/projects/e3sm/www/e3smv2-fme-dataset).

## Available datasets

Two versions of the complete dataset described in [arxiv:2310.02074](https://arxiv.org/abs/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.