Repository files navigation

Install Earth2Studio:

pip install earth2studio

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

fromearth2studio.models.pximportDLWPfromearth2studio.dataimportGFSfromearth2studio.ioimportNetCDF4Backendfromearth2studio.runimportdeterministicasrunmodel=DLWP.load_model(DLWP.load_default_package())ds=GFS()io=NetCDF4Backend("output.nc")run(["2024-01-01"],10,model,ds,io)

Earth2Studio provides access to pre-trained AI weather models and inferencefeatures through an easy to use and extendable Python interface.This package focuses on supplying users the tools to build their ownworkflows, pipelines, APIs, packages, etc. via modular components including:

• Collection of pre-trained weather/climate prediction models
• Collection of pre-trained diagnostic weather models
• Variety of online and on-prem data sources for initialization, scoring, analysis, etc.
• IO utilities for exporting predicted data to user friendly formats
• Suite of perturbation methods for building ensemble predictions
• Sample workflows and examples for common tasks / use cases
• Seamless integration into other Nvidia packages includingPhysicsNeMo

For a more complete list of feature set, be sure to view thedocumentation.Don't see what you need?Great news, extension and customization are at the heart of ourdesign.

Check out theContributing document for details about the technicalrequirements and the userguide for higher level philosophy, structure, and design.

Earth2Studio is provided under the Apache License 2.0, please seeLICENSE file for full license text.