Skip to main content
Ctrl+K
Earth2Studio - Home Earth2Studio - Home

Earth2Studio

  • User Guide
  • API
  • Examples
  • Recipes
  • Changelog
  • Github
Earth2Studio - Home Earth2Studio - Home

Earth2Studio

  • Github

Table of Contents

  • User Guide
    • Getting Started
      • Install
      • Introduction
      • Data Movement
    • Core Components
      • Prognostic Models
      • Diagnostic Models
      • Data and Forecast Sources
      • Perturbations
      • Output Handling
      • Statistics and Metrics
    • Advanced Usage
      • Batch Dimension
      • AutoModels
      • Lexicon
    • Developer Guide
      • Overview
      • Dependency Management
      • Style Guide
      • Documentation
      • Testing Guide
      • Build Process
      • Recipe Guide
    • Support
      • Troubleshooting
      • Frequently Asked Questions
  • API
    • earth2studio.models.px: Prognostics
      • ACE2ERA5
      • AIFS
      • AIFSENS
      • Atlas
      • Aurora
      • CBottleVideo
      • DiagnosticWrapper
      • DLESyM
      • DLESyMLatLon
      • DLWP
      • FCN
      • FCN3
      • FengWu
      • FuXi
      • GenCastMini
      • GraphCastOperational
      • GraphCastSmall
      • InterpModAFNO
      • Pangu24
      • Pangu6
      • Pangu3
      • Persistence
      • SFNO
      • StormCast
      • StormScopeGOES
      • StormScopeMRMS
    • earth2studio.models.dx: Diagnostics
      • CBottleInfill
      • CBottleSR
      • CBottleTCGuidance
      • CorrDiffCMIP6
      • CorrDiffTaiwan
      • ClimateNet
      • DerivedRH
      • DerivedRHDewpoint
      • DerivedSurfacePressure
      • DerivedTCWV
      • DerivedVPD
      • DerivedWS
      • OrbitGlobalPrecip
      • PrecipitationAFNO
      • PrecipitationAFNOv2
      • SolarRadiationAFNO1H
      • SolarRadiationAFNO6H
      • TCTrackerWuDuan
      • TCTrackerVitart
      • WindgustAFNO
      • Identity
    • earth2studio.models.da: Data Assimilation
      • HealDA
      • InterpEquirectangular
      • StormCastSDA
    • earth2studio.data: Data Sources
      • ARCO
      • CDS
      • CMIP6
      • CMIP6MultiRealm
      • GFS
      • GOES
      • HimawariAHI
      • HRRR
      • IFS
      • IFS_ENS
      • JPSS
      • MRMS
      • MeteosatFCI
      • NClimGridDaily
      • NCAR_ERA5
      • PlanetaryComputerECMWFOpenDataIFS
      • PlanetaryComputerGOES
      • PlanetaryComputerMODISFire
      • PlanetaryComputerOISST
      • PlanetaryComputerSentinel3AOD
      • Random
      • WB2ERA5
      • WB2ERA5_121x240
      • WB2ERA5_32x64
      • WB2Climatology
      • DataArrayFile
      • DataSetFile
      • DataArrayPathList
      • CBottle3D
    • earth2studio.data: Forecast Sources
      • AIFS_FX
      • CAMS_FX
      • AIFS_ENS_FX
      • GFS_FX
      • GEFS_FX
      • GEFS_FX_721x1440
      • HRRR_FX
      • IFS_FX
      • IFS_ENS_FX
    • earth2studio.data: DataFrame Sources
      • GHCNDaily
      • GOESGLM
      • ISD
      • JPSS_ATMS
      • JPSS_CRIS
      • MetOpAMSUA
      • MetOpAVHRR
      • MetOpIASI
      • MetOpMHS
      • NNJAObsConv
      • NomadsGDASObsConv
      • RandomDataFrame
      • UFSObsConv
      • UFSObsSat
    • earth2studio.io: IO Backends
      • AsyncZarrBackend
      • KVBackend
      • NetCDF4Backend
      • XarrayBackend
      • ZarrBackend
    • earth2studio.perturbation: Perturbations
      • Brown
      • BredVector
      • CorrelatedSphericalGaussian
      • Gaussian
      • HemisphericCentredBredVector
      • LaggedEnsemble
      • SphericalGaussian
      • Zero
    • earth2studio.statistics: Statistics
      • acc
      • brier
      • crps
      • energy_score
      • fss
      • lsd
      • mae
      • mean
      • rank_histogram
      • rmse
      • spread_skill_ratio
      • std
      • variance
      • lat_weight
    • earth2studio.run: Workflows
      • deterministic
      • diagnostic
      • ensemble
    • earth2studio: Utilities
      • handshake_dim
      • handshake_coords
      • handshake_size
      • map_coords
      • split_coords
      • LatLonInterpolation
      • timearray_to_datetime
      • to_time_array
      • datasource_to_file
      • fetch_data
      • prep_data_array
      • Package
      • batch_func
  • Examples
    • Getting Started
      • Running Deterministic Inference
      • Running Diagnostic Inference
      • Running Ensemble Inference
    • Medium Range
      • Single Variable Perturbation Method
      • Model Hook Injection: Perturbation
      • Huge Ensembles (HENS) Checkpoints
      • Temporal Interpolation
      • Tropical Cyclone Tracking
    • Downscaling
      • Generative Downscaling
      • CBottle Super Resolution
      • Ensemble Forecasting with Downscaling
    • Nowcasting
      • Running StormCast Inference
      • Running StormCast Ensemble Inference
      • StormScope Satellite and Radar Nowcasting
    • Data Assimilation
      • StormCast Score-Based Data Assimilation
      • HealDA Global Data Assimilation
    • Subseasonal to Seasonal
      • Statistical Inference
      • Running DLESyM Inference
    • Miscellaneous
      • Distributed Manager Inference
      • CBottle Data Generation and Infilling
      • IO Backend Performance
      • Creating a Local Data Source
      • CBottle Tropical Cyclone Guidance
    • Extending Earth2Studio
      • Extending Prognostic Models
      • Extending Diagnostic Models
      • Extending Data Sources
  • Examples
  • Extending Earth2Studio

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

previous

CBottle Tropical Cyclone Guidance

next

Extending Prognostic Models

NVIDIA NVIDIA
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA.