NVIDIA · Kilometre-scale weather emulator

STRATA (Earth-2 Checkpoints)

STRATA (Storm-resolving Tile-based autoRegressive Atmosphere Transformer Architecture) is a global, kilometre-scale AI atmosphere model from NVIDIA. It emulates the SCREAM global storm-resolving simulation. It predicts the 3-D atmospheric state autoregressively on SCREAM's ne1024pg2 cubed-sphere grid (about 25.2 million columns, ~4.9 km spacing) with a 10-minute time step. The state covers potential temperature, winds, geopotential height, vertical pressure velocity and specific humidity on 24 levels, plus 2-m air temperature, with surface pressure and precipitation fluxes diagnosed at each step. It is a two-stage transformer with 331.7 M parameters, trained as a deterministic single-step predictor with a regression loss (no diffusion). Inference is tile-based, so one GPU can advance part of the globe and multi-GPU domain decomposition covers the full globe. Training used 17 simulated days from three SCREAM runs. The text gives no evaluation figures. The checkpoint is strata_v1, under the OpenMDW-1.1 licence, with training recipes and the inference wrapper in NVlabs/STRATA (Apache-2.0). The model code is in NVIDIA PhysicsNeMo. The Hugging Face release is listed as planned for 2 November 2026.