earth2studio.models.px: Prognostics#

Prognostic models are a class of models that perform time-integration. Thus are typically used to generate forecast predictions.

Warning

Pre-trained prognostic models provided in Earth2Studio may be provided under different licenses. We encourage users to familiarize themselves with each prior to use.

Region
Class
Product
Release
Rec VRAM

ACE2ERA5

ACE2-ERA5 prognostic model wrapper.

AIFS

Artificial Intelligence Forecasting System (AIFS), a data driven forecast model developed by the European Centre for Medium-Range Weather Forecasts (ECMWF).

AIFSENS

Artificial Intelligence Forecasting System Ensemble (AIFS ENS v1.0), a probabilistic, ensemble-based forecast model from the European Centre for Medium-Range Weather Forecasts (ECMWF).

Atlas

Atlas prognostic model for ERA5 variables on a 0.25° global lat-lon grid.

Aurora

Aurora 0.25 degree global forecast model.

CBottleVideo

Climate in a bottle video prognostic Climate in a Bottle (cBottle) is an AI model for emulating global km-scale climate simulations and reanalysis on the equal-area HEALPix grid.

DiagnosticWrapper

Wraps a prognostic model and one or more diagnostic models into a single prognostic model.

DLESyM

DLESyM-V1-ERA5 prognostic model.

DLESyMLatLon

DLESyM prognostic model supporting lat/lon input and output coordinates.

DLWP

Deep learning weather prediction (DLWP) prognostic model.

FCN

FourCastNet global prognostic model.

FCN3

FourCastNet 3 advances global weather modeling by implementing a scalable, geometric machine learning (ML) approach to probabilistic ensemble forecasting.

FengWu

FengWu (operational) weather model consists of single auto-regressive model with a time-step size of 6 hours.

FuXi

FuXi weather model consists of three auto-regressive U-net transfomer models with a time-step size of 6 hours.

GraphCastOperational

GraphCast operational model

GraphCastSmall

GraphCast Small 1.0 degree model

InterpModAFNO

ModAFNO interpolation for global prognostic models.

Pangu24

Pangu Weather 24 hour model.

Pangu6

Pangu Weather 6 hour model.

Pangu3

Pangu Weather 3 hour model.

Persistence

Persistence model that generates a forecast by applying the identity operator on the initial condition and indexing the lead time by 6 hours.

SFNO

Spherical Fourier Operator Network global prognostic model.

StormCast

StormCast generative convection-allowing model for regional forecasts consists of two core models: a regression and diffusion model.

StormScopeGOES

StormScope model forecasting GOES data on the HRRR grid.

StormScopeMRMS

StormScope model forecasting MRMS data on the HRRR grid.