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

Running Deterministic Inference

Running Deterministic Inference

Running Diagnostic Inference

Running Diagnostic Inference

Running Ensemble Inference

Running Ensemble Inference

Generative Downscaling

Generative Downscaling

Single Variable Perturbation Method

Single Variable Perturbation Method

Model Hook Injection: Perturbation

Model Hook Injection: Perturbation

Statistical Inference

Statistical Inference

Distributed Manager Inference

Distributed Manager Inference

Running StormCast Inference

Running StormCast Inference

Running StormCast Ensemble Inference

Running StormCast Ensemble Inference

Huge Ensembles (HENS) Checkpoints

Huge Ensembles (HENS) Checkpoints

Temporal Interpolation

Temporal Interpolation

Tropical Cyclone Tracking

Tropical Cyclone Tracking

Running DLESyM Inference

Running DLESyM Inference

CBottle Data Generation and Infilling

CBottle Data Generation and Infilling

CBottle Super Resolution

CBottle Super Resolution

IO Backend Performance

IO Backend Performance

Extending Earth2Studio#

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

Extending Prognostic Models

Extending Prognostic Models

Extending Diagnostic Models

Extending Diagnostic Models

Extending Data Sources

Extending Data Sources

Gallery generated by Sphinx-Gallery