Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
Project description
Earth2Studio is a Python-based package designed to get users up and running with AI weather and climate models fast. Our mission is to enable everyone to build, research and explore AI driven meteorology.
- Earth2Studio Documentation -
Install | User-Guide | Examples | API
Quick start
Install Earth2Studio:
pip install earth2studio
Run a deterministic weather prediction in just a few lines of code:
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)
Features
Earth2Studio provides access to pre-trained AI weather models and inference features through an easy to use and extendable Python interface. This package focuses on supplying users the tools to build their own workflows, 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 including Modulus
For a more complete list of feature set, be sure to view the documentation. Don't see what you need? Great news, extension and customization are at the heart of our design.
Contributors
Check out the Contributing 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 LICENSE file for full license text.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file earth2studio-0.3.0.tar.gz
.
File metadata
- Download URL: earth2studio-0.3.0.tar.gz
- Upload date:
- Size: 96.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.10.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6716e138aa68efb3d64056f698d1f40dc7a8c210ef3d02e6bab50c3825e10032 |
|
MD5 | 8d24f489dbc6f1cd865560f35edc55b1 |
|
BLAKE2b-256 | f34aee45c78a173a34d14ca75f831a7bc119f48a2adc03cfa681ca48a0a2186e |
File details
Details for the file earth2studio-0.3.0-py3-none-any.whl
.
File metadata
- Download URL: earth2studio-0.3.0-py3-none-any.whl
- Upload date:
- Size: 177.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.0.0 CPython/3.10.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 88e80ba6df6879674fdade433a2c53ec50c665595f6370288ab1cedef7b0c020 |
|
MD5 | 53a5f532815cd8312c26dad048acf9a9 |
|
BLAKE2b-256 | 1cfdca19e1572750df36ae73bd73aeb0ec04d592c4c265efd9b84061032c8d91 |