Metadata-Version: 2.4
Name: earth2studio
Version: 0.10.0
Summary: Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
Project-URL: Homepage, https://github.com/NVIDIA/earth2studio
Project-URL: Documentation, https://nvidia.github.io/earth2studio
Project-URL: Issues, https://github.com/NVIDIA/earth2studio/issues
Project-URL: Changelog, https://github.com/NVIDIA/earth2studio/blob/main/CHANGELOG.md
Author: NVIDIA PhysicsNeMo Team
License-Expression: Apache-2.0
License-File: LICENSE
Classifier: Environment :: GPU
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Typing :: Typed
Requires-Python: >=3.11
Requires-Dist: cfgrib>=0.9.10.3
Requires-Dist: cftime
Requires-Dist: fsspec>=2024.2.0
Requires-Dist: gcsfs
Requires-Dist: h5netcdf>=1.0.0
Requires-Dist: h5py>=3.2.0
Requires-Dist: huggingface-hub>=0.27.0
Requires-Dist: loguru
Requires-Dist: nest-asyncio
Requires-Dist: netcdf4<1.7.3,>=1.6.4
Requires-Dist: numcodecs<0.15.0
Requires-Dist: python-dotenv
Requires-Dist: rich
Requires-Dist: s3fs>=2023.5.0
Requires-Dist: torch>=2.5.0
Requires-Dist: tqdm>=4.65.0
Requires-Dist: xarray[parallel]>=2023.1.0
Requires-Dist: zarr>=3.1.0
Provides-Extra: ace2
Requires-Dist: fme; extra == 'ace2'
Requires-Dist: h5py<=3.14.0; extra == 'ace2'
Requires-Dist: pandas>=2.3.2; extra == 'ace2'
Requires-Dist: scipy>=1.15.2; extra == 'ace2'
Provides-Extra: aifs
Requires-Dist: anemoi-inference[huggingface]==0.4.9; extra == 'aifs'
Requires-Dist: anemoi-models==0.3.1; extra == 'aifs'
Requires-Dist: earthkit-regrid==0.4.0; extra == 'aifs'
Requires-Dist: ecmwf-opendata>=0.3.3; extra == 'aifs'
Requires-Dist: flash-attn; extra == 'aifs'
Provides-Extra: all
Requires-Dist: cbottle; (python_full_version >= '3.12') and extra == 'all'
Requires-Dist: cdsapi>=0.7.5; extra == 'all'
Requires-Dist: cucim-cu12>=25.4.0; extra == 'all'
Requires-Dist: dm-haiku>=0.0.14; extra == 'all'
Requires-Dist: earth2grid; extra == 'all'
Requires-Dist: eccodes>=2.38.0; extra == 'all'
Requires-Dist: ecmwf-opendata>=0.3.3; extra == 'all'
Requires-Dist: einops>=0.8.1; extra == 'all'
Requires-Dist: flax>=0.10.6; extra == 'all'
Requires-Dist: fme; extra == 'all'
Requires-Dist: globus-sdk<4.0.0; extra == 'all'
Requires-Dist: graphcast; extra == 'all'
Requires-Dist: h5py<=3.14.0; extra == 'all'
Requires-Dist: hydra-core>=1.3.2; extra == 'all'
Requires-Dist: importlib-metadata; extra == 'all'
Requires-Dist: intake-esgf>=2025.9.26; extra == 'all'
Requires-Dist: jax[cuda12]<0.7.2,>=0.4.26; extra == 'all'
Requires-Dist: makani; extra == 'all'
Requires-Dist: microsoft-aurora; extra == 'all'
Requires-Dist: moviepy; extra == 'all'
Requires-Dist: multi-storage-client[boto3,fsspec,google-cloud-storage]>=0.33.0; extra == 'all'
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'all'
Requires-Dist: nvidia-physicsnemo>=1.2.0; extra == 'all'
Requires-Dist: nvtx>=0.2.11; extra == 'all'
Requires-Dist: omegaconf>=2.3.0; extra == 'all'
Requires-Dist: onnxruntime-gpu>=1.21.0; extra == 'all'
Requires-Dist: pandas>=2.3.2; extra == 'all'
Requires-Dist: pynvml>=12.0.0; extra == 'all'
Requires-Dist: pyproj>=3.7.1; extra == 'all'
Requires-Dist: ruamel-yaml>=0.18.10; extra == 'all'
Requires-Dist: ruamel-yaml>=0.18.14; extra == 'all'
Requires-Dist: scikit-image>=0.25.2; extra == 'all'
Requires-Dist: scipy>=1.10.0; extra == 'all'
Requires-Dist: scipy>=1.15.2; extra == 'all'
Requires-Dist: torch-harmonics>=0.7.4; extra == 'all'
Requires-Dist: torch-harmonics>=0.8.0; extra == 'all'
Provides-Extra: aurora
Requires-Dist: microsoft-aurora>=1.5.0; extra == 'aurora'
Provides-Extra: aurora-fork
Requires-Dist: microsoft-aurora; extra == 'aurora-fork'
Provides-Extra: cbottle
Requires-Dist: cbottle; (python_full_version >= '3.12') and extra == 'cbottle'
Requires-Dist: earth2grid; extra == 'cbottle'
Provides-Extra: climatenet
Provides-Extra: corrdiff
Requires-Dist: einops>=0.8.1; extra == 'corrdiff'
Requires-Dist: nvidia-physicsnemo>=1.2.0; extra == 'corrdiff'
Requires-Dist: nvtx>=0.2.11; extra == 'corrdiff'
Requires-Dist: scipy>=1.15.2; extra == 'corrdiff'
Provides-Extra: cyclone
Requires-Dist: cucim-cu12>=25.4.0; extra == 'cyclone'
Requires-Dist: scikit-image>=0.25.2; extra == 'cyclone'
Provides-Extra: data
Requires-Dist: cdsapi>=0.7.5; extra == 'data'
Requires-Dist: eccodes>=2.38.0; extra == 'data'
Requires-Dist: ecmwf-opendata>=0.3.3; extra == 'data'
Requires-Dist: globus-sdk<4.0.0; extra == 'data'
Requires-Dist: intake-esgf>=2025.9.26; extra == 'data'
Requires-Dist: multi-storage-client[boto3,fsspec,google-cloud-storage]>=0.33.0; extra == 'data'
Requires-Dist: pyproj>=3.7.1; extra == 'data'
Requires-Dist: scipy>=1.15.2; extra == 'data'
Provides-Extra: derived
Provides-Extra: dlesym
Requires-Dist: earth2grid; extra == 'dlesym'
Requires-Dist: hydra-core>=1.3.2; extra == 'dlesym'
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'dlesym'
Requires-Dist: omegaconf>=2.3.0; extra == 'dlesym'
Provides-Extra: dlwp
Requires-Dist: importlib-metadata; extra == 'dlwp'
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'dlwp'
Provides-Extra: fcn
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'fcn'
Provides-Extra: fcn3
Requires-Dist: makani; extra == 'fcn3'
Requires-Dist: moviepy; extra == 'fcn3'
Requires-Dist: ruamel-yaml>=0.18.14; extra == 'fcn3'
Requires-Dist: scipy>=1.10.0; extra == 'fcn3'
Provides-Extra: fengwu
Requires-Dist: onnxruntime-gpu>=1.21.0; extra == 'fengwu'
Provides-Extra: fuxi
Requires-Dist: onnxruntime-gpu>=1.21.0; extra == 'fuxi'
Provides-Extra: graphcast
Requires-Dist: dm-haiku>=0.0.14; extra == 'graphcast'
Requires-Dist: flax>=0.10.6; extra == 'graphcast'
Requires-Dist: graphcast; extra == 'graphcast'
Requires-Dist: jax[cuda12]<0.7.2,>=0.4.26; extra == 'graphcast'
Provides-Extra: interp-modafno
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'interp-modafno'
Provides-Extra: pangu
Requires-Dist: onnxruntime-gpu>=1.21.0; extra == 'pangu'
Provides-Extra: perturbation
Requires-Dist: torch-harmonics>=0.7.4; extra == 'perturbation'
Provides-Extra: precip-afno
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'precip-afno'
Provides-Extra: precip-afno-v2
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'precip-afno-v2'
Provides-Extra: sfno
Requires-Dist: makani; extra == 'sfno'
Requires-Dist: moviepy; extra == 'sfno'
Requires-Dist: pynvml>=12.0.0; extra == 'sfno'
Requires-Dist: ruamel-yaml>=0.18.10; extra == 'sfno'
Requires-Dist: scipy>=1.10.0; extra == 'sfno'
Requires-Dist: torch-harmonics>=0.8.0; extra == 'sfno'
Provides-Extra: solarradiation-afno
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'solarradiation-afno'
Provides-Extra: statistics
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'statistics'
Provides-Extra: stormcast
Requires-Dist: einops>=0.8.1; extra == 'stormcast'
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'stormcast'
Requires-Dist: nvtx>=0.2.11; extra == 'stormcast'
Requires-Dist: omegaconf>=2.3.0; extra == 'stormcast'
Requires-Dist: pyproj>=3.7.1; extra == 'stormcast'
Requires-Dist: scipy>=1.15.2; extra == 'stormcast'
Provides-Extra: windgust-afno
Requires-Dist: nvidia-physicsnemo>=1.0.1; extra == 'windgust-afno'
Description-Content-Type: text/markdown

# NVIDIA Earth2Studio

[![python version][e2studio_python_img]][e2studio_python_url]
[![license][e2studio_license_img]][e2studio_license_url]
[![coverage][e2studio_cov_img]][e2studio_cov_url]
[![mypy][e2studio_mypy_img]][e2studio_mypy_url]
[![format][e2studio_format_img]][e2studio_format_url]
[![ruff][e2studio_ruff_img]][e2studio_ruff_url]
[![uv][e2studio_uv_img]][e2studio_uv_url]

Earth2Studio is a Python-based package designed to get users up and running
with AI Earth system models *fast*.
Our mission is to enable everyone to build, research and explore AI driven weather and
climate science.

**- Earth2Studio Documentation -**

[Install][e2studio_install_url] | [User-Guide][e2studio_userguide_url] |
[Examples][e2studio_examples_url] | [API][e2studio_api_url]

![Earth2Studio Banner](https://huggingface.co/datasets/NickGeneva/Earth2StudioAssets/raw/main/0.2.0/earth2studio_feature_banner.png?id=1)

## Quick start

Install Earth2Studio:

```bash
pip install earth2studio[dlwp]
```

Run a deterministic AI weather prediction in just a few lines of code:

```python
from earth2studio.models.px import DLWP
from earth2studio.data import GFS
from earth2studio.io import NetCDF4Backend
from earth2studio.run import deterministic as run

model = DLWP.load_model(DLWP.load_default_package())
ds = GFS()
io = NetCDF4Backend("output.nc")

run(["2024-01-01"], 10, model, ds, io)
```

Swap out for a different AI model by just [installing](https://nvidia.github.io/earth2studio/userguide/about/install.html#prognostics)
and replacing `DLWP` references with another [forecast model][e2studio_px_api].

## Latest News

- The latest [**Climate in a Bottle**](https://blogs.nvidia.com/blog/earth2-generative-ai-foundation-model-global-climate-kilometer-scale-resolution/)
    generative AI model from NVIDIA research has been
    added via several APIs including a [data source](https://nvidia.github.io/earth2studio/modules/generated/data/earth2studio.data.CBottle3D.html),
    [infilling](https://nvidia.github.io/earth2studio/modules/generated/models/dx/earth2studio.models.dx.CBottleInfill.html)
    and [super-resolution](https://nvidia.github.io/earth2studio/modules/generated/models/dx/earth2studio.models.dx.CBottleSR.html)
    APIs. See the [cBottle examples](https://nvidia.github.io/earth2studio/examples/index.html)
    for more.
- The long awaited **GraphCast 1 degree** [prognostic model](https://nvidia.github.io/earth2studio/modules/generated/models/px/earth2studio.models.px.GraphCastSmall.html)
    and **GraphCast Operational** [prognostic model](https://nvidia.github.io/earth2studio/modules/generated/models/px/earth2studio.models.px.GraphCastOperational.html)
    are now added.
- Advanced **Subseasonal-to-Seasonal (S2S) forecasting** [recipe](./recipes/s2s/)
    added demonstrating new inference pipelines for subseasonal weather forecasts (from
    2 weeks to 3 months).

For a complete list of latest features and improvements see the [changelog](./CHANGELOG.md).

## Overview

Earth2Studio is an *AI inference pipeline toolkit* focused on weather and climate
applications that is designed to ride on top of different AI frameworks, model
architectures, data sources and SciML tooling while providing a unified API.

![Earth2Studio Overview 1](https://huggingface.co/datasets/NickGeneva/Earth2StudioAssets/resolve/main/0.9.0/earth2studio-readme-overview-1.png?id=1)

The composability of the different core components in Earth2Studio easily allows the
development and deployment of increasingly complex pipelines that may chain multiple
data sources, AI models and other modules together.

![Earth2Studio Overview 1](https://huggingface.co/datasets/NickGeneva/Earth2StudioAssets/resolve/main/0.9.0/earth2studio-readme-overview-2.png?id=1)

The unified ecosystem of Earth2Studio provides users the opportunity to rapidly
swap out components for alternatives.
In addition to the largest model zoo of weather/climate AI models, Earth2Studio is
packed with useful functionality such as optimized data access to cloud data stores,
statistical operations and more to accelerate your pipelines.

![Earth2Studio Overview 1](https://huggingface.co/datasets/NickGeneva/Earth2StudioAssets/resolve/main/0.9.0/earth2studio-readme-overview-3.webp?id=1)

Earth2Studio can be used for seamless deployment of Earth-2 models trained in
[PhysicsNeMo][physicsnemo_repo_url].

## Features

Earth2Studio package focuses on supplying users the tools to build their own
workflows, pipelines, APIs, packages, etc. via modular components including:

Prognostic Models

[Prognostic models][e2studio_px_url]
    in Earth2Studio perform time integration, taking atmospheric fields at a specific
    time and auto-regressively predicting the same fields into the future (typically 6
    hours per step), enabling both single time-step predictions and extended time-series
    forecasting.

Earth2Studio maintains the largest collection of pre-trained state-of-the-art AI
    weather/climate models ranging from global forecast models to regional specialized
    models, covering various resolutions, architectures, and forecasting capabilities to
    suit different computational and accuracy requirements.

Available models include but are not limited to:

| Model | Resolution | Architecture | Time Step | Coverage |
|-------|------------|--------------|-----------|----------|
| GraphCast Small | 1.0° | Graph Neural Network | 6h | Global |
| GraphCast Operational | 0.25° | Graph Neural Network | 6h | Global |
| Pangu 3hr | 0.25° | Transformer | 3h | Global |
| Pangu 6hr | 0.25° | Transformer | 6h | Global |
| Pangu 24hr | 0.25° | Transformer | 24h | Global |
| Aurora | 0.25° | Transformer | 6h | Global |
| FuXi | 0.25° | Transformer | 6h | Global |
| AIFS | 0.25° | Transformer | 6h | Global |
| StormCast | 3km | Diffusion + Regression | 1h | Regional (US) |
| SFNO | 0.25° | Neural Operator | 6h | Global |
| DLESyM | 0.25° | Convolutional | 6h | Global |

For a complete list, see the [prognostic model API docs][e2studio_px_api].

Diagnostic Models

[Diagnostic models][e2studio_dx_url] in Earth2Studio perform time-independent
    transformations, typically taking geospatial fields at a specific time and
    predicting new derived quantities without performing time integration enabling users
    to build pipelines to predict specific quantities of interest that may not be
    provided by forecasting models.

Earth2Studio contains a growing collection of specialized diagnostic models for
    various phenomena including precipitation prediction, tropical cyclone tracking,
    solar radiation estimation, wind gust forecasting, and more.

Available diagnostics include but are not limited to:

| Model | Resolution | Architecture | Coverage | Output |
|-------|------------|--------------|----------|--------|
| PrecipitationAFNO | 0.25° | Neural Operator  | Global | Total precipitation |
| SolarRadiationAFNO1H | 0.25° | Neural Operator  | Global | Surface solar radiation |
| WindgustAFNO | 0.25° | AFNO | Global | Maximum wind gust |
| TCTrackerVitart | 0.25° | Algorithmic | Global | TC tracks & properties |
| CBottleInfill | 100km | Diffusion | Global | Global climate sample |
| CBottleSR | 5km | Diffusion | Regional / Global | High-res climate |
| CorrDiff | Variable | Diffusion | Regional | Fine-scale weather |
| CorrDiffTaiwan | 2km | Diffusion | Regional (Taiwan) | Taiwan fine-scale weather |

For a complete list, see the [diagnostic model API docs][e2studio_dx_api].

Datasources

[Data sources][e2studio_data_url]
    in Earth2Studio provide a standardized API for accessing weather and climate
    datasets from various providers (numerical models, data assimilation results, and
    AI-generated data), enabling seamless integration of initial conditions for model
    inference and validation data for scoring across different data formats and storage
    systems.

Earth2Studio includes data sources ranging from operational weather models (GFS, HRRR,
    IFS) and reanalysis datasets (ERA5 via ARCO, CDS) to AI-generated climate data
    (cBottle) and local file systems. Fetching data is just plain easy, Earth2Studio
    handles the complicated parts giving the users an easy to use Xarray data array of
    requested data under a shared package wide [vocabulary][e2studio_lex_url] and
    coordinate system.

Available data sources include but are not limited to:

| Data Source | Type | Resolution | Coverage | Data Format |
|-------------|------|------------|----------|-------------|
| GFS | Operational | 0.25° | Global | GRIB2 |
| GFS_FX | Forecast | 0.25° | Global | GRIB2 |
| HRRR | Operational | 3km | Regional (US) | GRIB2 |
| HRRR_FX | Forecast | 3km | Regional (US) | GRIB2 |
| ARCO ERA5 | Reanalysis | 0.25° | Global | Zarr |
| CDS | Reanalysis | 0.25° | Global | NetCDF |
| IFS | Operational | 0.25° | Global | GRIB2 |
| NCAR_ERA5 | Reanalysis | 0.25° | Global | NetCDF |
| WeatherBench2 | Reanalysis | 0.25° | Global | Zarr |
| GEFS_FX | Ensemble Forecast | 0.25° | Global | GRIB2 |
| IMERG | Precipitation | 0.1° | Global | NetCDF |
| CBottle3D | AI Generated | 100km | Global | HEALPix |

For a complete list, see the [data source API docs][e2studio_data_api].

IO Backends

[IO backends][e2studio_io_url] in
    Earth2Studio provides a standardized interface for writing and storing
    pipeline outputs across different file formats and storage systems enabling users
    to store inference outputs for later processing.

Earth2Studio includes IO backends ranging from traditional scientific formats (NetCDF)
    and modern cloud-optimized formats (Zarr) to in-memory storage backends.

Available IO backends include:

| IO Backend | Format | Features | Location |
|------------|--------|----------|----------|
| ZarrBackend | Zarr | Compression, Chunking | In-Memory/Local |
| AsyncZarrBackend | Zarr | Async writes, Parallel I/O | In-Memory/Local/Remote |
| NetCDF4Backend | NetCDF4 | CF-compliant, Metadata | In-Memory/Local |
| XarrayBackend | Xarray Dataset | Rich metadata, Analysis-ready | In-Memory |
| KVBackend | Key-Value| Fast Temporary Access | In-Memory |

For a complete list, see the [IO API docs][e2studio_io_api].

Perturbation Methods

[Perturbation methods][e2studio_pb_url]
    in Earth2Studio provide a standardized interface for adding noise
    to data arrays, typically enabling the creation of ensembling forecast pipelines
    that capture uncertainty in weather and climate predictions.

Available perturbations include but are not limited to:

| Perturbation Method | Type | Spatial Correlation | Temporal Correlation |
|---------------------|------|-------------------|---------------------|
| Gaussian | Noise | None | None |
| Correlated SphericalGaussian | Noise | Spherical | AR(1) process |
| Spherical Gaussian | Noise | Spherical (Matern) | None |
| Brown | Noise | 2D Fourier | None |
| Bred Vector | Dynamical | Model-dependent | Model-dependent |
| Hemispheric Centred Bred Vector | Dynamical | Hemispheric | Model-dependent |

For a complete list, see the [perturbations API docs][e2studio_pb_url].

Statistics / Metrics

[Statistics and metrics][e2studio_stat_url]
    in Earth2Studio provide operations typically useful for in-pipeline evaluation of
    forecast performance across different dimensions (spatial, temporal, ensemble)
    through various statistical measures including error metrics, correlation
    coefficients, and ensemble verification statistics.

Available operations include but are not limited to:

| Statistic | Type | Application |
|-----------|------|-------------|
| RMSE | Error Metric | Forecast accuracy |
| ACC | Correlation | Pattern correlation |
| CRPS | Ensemble Metric | Probabilistic skill |
| Rank Histogram | Ensemble Metric | Ensemble reliability |
| Standard Deviation | Moment | Spread measure |
| Spread-Skill Ratio | Ensemble Metric | Ensemble calibration |

For a complete list, see the [statistics API docs][e2studio_stat_api].

For a more complete list of features, be sure to view the [documentation][e2studio_docs_url].
Don't see what you need?
Great news, extension and customization are at the heart of our [design][e2studio_customization_url].

## Contributors

Check out the [contributing](CONTRIBUTING.md) document for details about the technical
requirements and the userguide for higher level philosophy, structure, and design.

## License

Earth2Studio is provided under the Apache License 2.0, please see the
[LICENSE file][e2studio_license_url] for full license text.

[e2studio_python_img]: https://img.shields.io/badge/Python-3.11%20|%203.12%20|%203.13-blue?style=flat-square&logo=python
[e2studio_license_img]: https://img.shields.io/badge/License-Apache%202.0-green?style=flat-square
[e2studio_format_img]: https://img.shields.io/badge/Code%20Style-Black-black?style=flat-square
[e2studio_mypy_img]: https://img.shields.io/badge/mypy-Checked-blue?style=flat-square&labelColor=grey
[e2studio_cov_img]: https://img.shields.io/codecov/c/github/nickgeneva/earth2studio?style=flat-square&logo=codecov
[e2studio_ruff_img]: https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json&style=flat-square
[e2studio_uv_img]: https://img.shields.io/endpoint?url=https%3A%2F%2Fraw.githubusercontent.com%2Fastral-sh%2Fuv%2Fmain%2Fassets%2Fbadge%2Fv0.json&style=flat-square

[e2studio_python_url]: https://www.python.org/downloads/
[e2studio_license_url]: ./LICENSE
[e2studio_format_url]: https://github.com/psf/black
[e2studio_cov_url]: ./test/
[e2studio_mypy_url]: https://mypy-lang.org/
[e2studio_ruff_url]: https://github.com/astral-sh/ruff
[e2studio_uv_url]: https://github.com/astral-sh/uv

[e2studio_docs_url]: https://nvidia.github.io/earth2studio/
[e2studio_install_url]: https://nvidia.github.io/earth2studio/userguide/about/install.html
[e2studio_userguide_url]: https://nvidia.github.io/earth2studio/userguide/
[e2studio_examples_url]: https://nvidia.github.io/earth2studio/examples/
[e2studio_api_url]: https://nvidia.github.io/earth2studio/modules/
[e2studio_customization_url]: https://nvidia.github.io/earth2studio/examples/extend/index.html
[e2studio_px_url]: https://nvidia.github.io/earth2studio/userguide/components/prognostic.html
[e2studio_px_api]: https://nvidia.github.io/earth2studio/modules/models.html#earth2studio-models-px-prognostic
[e2studio_dx_url]: https://nvidia.github.io/earth2studio/userguide/components/diagnostic.html
[e2studio_dx_api]: https://nvidia.github.io/earth2studio/modules/models.html#earth2studio-models-dx-diagnostic
[e2studio_data_url]: https://nvidia.github.io/earth2studio/userguide/components/datasources.html
[e2studio_data_api]: https://nvidia.github.io/earth2studio/modules/datasources.html
[e2studio_io_url]: https://nvidia.github.io/earth2studio/userguide/components/io.html
[e2studio_io_api]: https://nvidia.github.io/earth2studio/modules/io.html
[e2studio_pb_url]: https://nvidia.github.io/earth2studio/userguide/components/perturbation.html
[e2studio_pb_api]: https://nvidia.github.io/earth2studio/modules/perturbation.html
[e2studio_stat_url]: https://nvidia.github.io/earth2studio/userguide/components/statistics.html
[e2studio_stat_api]: https://nvidia.github.io/earth2studio/modules/statistics.html
[e2studio_lex_url]: https://nvidia.github.io/earth2studio/userguide/advanced/lexicon.html

[physicsnemo_repo_url]: https://github.com/NVIDIA/physicsnemo
