Skip to main content

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

Project description

Earth2Studio Banner

python version license format coverage

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

Earth2Studio Banner

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

earth2studio-0.3.0.tar.gz (96.1 kB view details)

Uploaded Source

Built Distribution

earth2studio-0.3.0-py3-none-any.whl (177.4 kB view details)

Uploaded Python 3

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

Hashes for earth2studio-0.3.0.tar.gz
Algorithm Hash digest
SHA256 6716e138aa68efb3d64056f698d1f40dc7a8c210ef3d02e6bab50c3825e10032
MD5 8d24f489dbc6f1cd865560f35edc55b1
BLAKE2b-256 f34aee45c78a173a34d14ca75f831a7bc119f48a2adc03cfa681ca48a0a2186e

See more details on using hashes here.

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

Hashes for earth2studio-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 88e80ba6df6879674fdade433a2c53ec50c665595f6370288ab1cedef7b0c020
MD5 53a5f532815cd8312c26dad048acf9a9
BLAKE2b-256 1cfdca19e1572750df36ae73bd73aeb0ec04d592c4c265efd9b84061032c8d91

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page