earth2studio

Quick start

Running AI weather prediction can be done with just a few lines of code.

• For detailed installation steps, including model-specific installations, see the [install guide][e2studio_install_url].
• See the [examples][e2studio_examples_url] gallery providing different inference workflow samples.
• Swap out [data sources][e2studio_data_api] or [models][e2studio_px_api] depending on your use case!

NVIDIA FourCastNet3

from earth2studio.models.px import FCN3
from earth2studio.data import GFS
from earth2studio.io import ZarrBackend
from earth2studio.run import deterministic as run

model = FCN3.load_model(FCN3.load_default_package())
data = GFS()
io = ZarrBackend("outputs/fcn3_forecast.zarr")
run(["2025-01-01T00:00:00"], 10, model, data, io)

ECMWF AIFS

from earth2studio.models.px import AIFS
from earth2studio.data import IFS
from earth2studio.io import ZarrBackend
from earth2studio.run import deterministic as run

model = AIFS.load_model(AIFS.load_default_package())
data = IFS()
io = ZarrBackend("outputs/aifs_forecast.zarr")
run(["2025-01-01T00:00:00"], 10, model, data, io)

Google Graphcast

from earth2studio.models.px import GraphCastOperational
from earth2studio.data import GFS
from earth2studio.io import ZarrBackend
from earth2studio.run import deterministic as run

package = GraphCastOperational.load_default_package()
model = GraphCastOperational.load_model(package)
data = GFS()
io = ZarrBackend("outputs/graphcast_operational_forecast.zarr")
run(["2025-01-01T00:00:00"], 4, model, data, io)

[!IMPORTANT] Earth2Studio is an interface to third‑party models, checkpoints, and datasets. Licenses for these assets are owned by their providers. Ensure you have the rights to download, use, and (if applicable) redistribute each model and dataset. Links to the original license and source are often provided in the API docs for each model/data source.

Latest News

[!NOTE] As of version 0.14.0, Earth2Studio TOML default installs now target CUDA 13.

• Himawari AHI, Himawari-8/9 AHI ISatSS L2 Full Disk satellite data source is now available for geostationary satellite observations over the Western Pacific.
• GHCN-Daily, Global Historical Climatology Network daily station observation data frame source providing access to global surface observations.
• Orbit-2, precipitation downscaling diagnostic model added for high-resolution precipitation estimates.
• Conventional Observation Sources, new data sources for in-situ observations including NNJA conventional obs and real-time GDAS conventional obs.

For a complete list of latest features and improvements see the changelog.

Overview

Earth2Studio is an AI inference pipeline toolkit focused on weather and climate applications that is designed to ride on top of different AI frameworks, model architectures, data sources and SciML tooling while providing a unified API.