earth2studio.models.da: Data Assimilation#

Data assimilation models are a class of models that integrate observational data into model states or grids. These models can ingest both sparse observations (via DataFrames) and dense fields (via DataArrays) to produce output suitable for downstream tasks such as driving a prognostic model or generating a guided forecast. Data assimilation models support both stateless and stateful operation, allowing them to process observations independently or maintain internal state across time steps.

Warning

Data Assimilation models are a new addition to Earth2Studio and APIs might be subject to change without warning while the implementation is hardened.

Region
Class
Product
Release
Rec VRAM

HealDA

HealDA data assimilation model for global weather analysis from sparse observations on a HEALPix grid.

InterpEquirectangular

Interpolation assimilation model that interpolates sparse observations to a lat-lon grid.

StormCastSDA

StormCast with score-based data assimilation (SDA) using diffusion posterior sampling for convection-allowing regional forecasts.