Examples#

This is a collection of examples in Earth2Studio that demonstrate various functionality and commonly used workflows.

Running Examples

Earth2Studio examples can be downloaded as a notebook or runnable Python script. Each requires installation of different optional dependency groups or additional packages for the specific models used or post-processing steps. Use uv to auto install dependencies on execution:

uv run <example_script>.py

If you are using a container or other type of environment, then pip installing will likely be needed. Look for the uv inline metadata blocks of the form:

# /// script
# dependencies = [
#   "earth2studio @ git+https://github.com/NVIDIA/earth2studio.git",
#   "cartopy",
# ]
# ///

Pip install these packages then execute the example with:

python <example_script>.py

Image

Running Deterministic Inference

Running Deterministic Inference
Image

Running Diagnostic Inference

Running Diagnostic Inference
Image

Running Ensemble Inference

Running Ensemble Inference
Image

Generative Downscaling

Generative Downscaling
Image

Single Variable Perturbation Method

Single Variable Perturbation Method
Image

Model Hook Injection: Perturbation

Model Hook Injection: Perturbation
Image

Statistical Inference

Statistical Inference
Image

Distributed Manager Inference

Distributed Manager Inference
Image

Running StormCast Inference

Running StormCast Inference
Image

Running StormCast Ensemble Inference

Running StormCast Ensemble Inference
Image

Huge Ensembles (HENS) Checkpoints

Huge Ensembles (HENS) Checkpoints
Image

Temporal Interpolation

Temporal Interpolation
Image

Tropical Cyclone Tracking

Tropical Cyclone Tracking
Image

Running DLESyM Inference

Running DLESyM Inference
Image

CBottle Data Generation and Infilling

CBottle Data Generation and Infilling
Image

CBottle Super Resolution

CBottle Super Resolution
Image

IO Backend Performance

IO Backend Performance
Image

Ensemble Forecasting with Downscaling

Ensemble Forecasting with Downscaling
Image

Creating a Local Data Source

Creating a Local Data Source

Extending Earth2Studio#

Examples focused on demonstrating how to extend Earth2Studio with user-defined components.

Image

Extending Prognostic Models

Extending Prognostic Models
Image

Extending Diagnostic Models

Extending Diagnostic Models
Image

Extending Data Sources

Extending Data Sources

Gallery generated by Sphinx-Gallery