AIAny
AI Infra2024
Icon for item

NVIDIA Earth2Studio

Chains pre-trained AI weather and climate models like GraphCast, Pangu, and FourCastNet into composable inference pipelines. Swap prognostic or diagnostic components, plug in reanalysis sources, and add ensemble perturbations or in-loop metrics.

Introduction

The race for AI weather forecasting has produced a dozen rival models — GraphCast from DeepMind, Pangu from Huawei, Aurora from Microsoft, FourCastNet from NVIDIA — and each ships with its own checkpoints, data formats, and assumptions. The real contribution here is not another model but the connective tissue that lets you run, swap, and chain all of them through one consistent interface, turning a fragmented research landscape into something you can actually experiment across.

What Sets It Apart
  • One of the largest curated zoos of pre-trained weather/climate models — prognostic (GraphCast, Pangu, Aurora, FuXi, AIFS, StormCast, SFNO, FourCastNet3) plus diagnostic ones (precipitation, solar radiation, wind gust, tropical-cyclone tracking, super-resolution) — behind a single API, so comparing architectures no longer means rewriting glue code each time.
  • Every stage is an interchangeable part: prognostic and diagnostic models, data sources (ERA5 reanalysis, operational feeds), perturbation methods, IO backends (Zarr / NetCDF / Xarray), and in-pipeline statistics. An ensemble or downscaling run becomes a configuration, not a bespoke script.
  • Ensembles and uncertainty are first-class — native perturbation methods and metrics live inside the pipeline rather than being bolted on afterward.
Who Should Use It

Great fit if you are a researcher or engineer who wants to benchmark several AI weather models head-to-head, assemble ensemble or super-resolution workflows, or move from prototype to deployment without stitching together incompatible codebases. Look elsewhere if you need a classic physics-based numerical weather model (this orchestrates data-driven ML instead), a turnkey forecasting product rather than a Python framework you assemble yourself, or you only ever touch a single model — then the abstraction overhead won't earn its keep.

Information

  • Websitegithub.com
  • OrganizationsNVIDIA
  • AuthorsEarth2Studio Contributors, NVIDIA
  • Published date2024/04/22

Categories

More Items

Enables RL post-training with million-token prompts under a fixed GPU budget by evaluating shared prompt state without autograd, retaining only minimal model state, and replaying short response branches; instantiated as GRPO and demonstrated on Qwen3.6-27B and GLM-5.2 up to multi-million token execution.

GitHub
AI Infra2026

Defines OpenTelemetry semantic conventions for generative AI telemetry — spans, metrics, and events for GenAI clients, the Model Context Protocol (MCP), and provider-specific integrations. Includes YAML models, human-readable docs, and reference implementations to standardize observability across GenAI deployments.

GitHub
AI Infra2024

Provides a lightweight build platform for HIP and ROCm that supports building ROCm, PyTorch, and JAX from source, multi-architecture nightly releases, and integrated CI/CD and developer tooling for Linux and Windows.