earth2studio 0.3.0

快速开始

安装Earth2Studio

pip install earth2studio

仅用几行代码运行确定性天气预测

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)

功能

Earth2Studio通过易于使用和可扩展的Python接口提供对预训练AI气象模型和推理功能的访问。此软件包侧重于为用户提供构建自己的工作流程、管道、API、软件包等工具，包括以下模块化组件：

• 预训练天气/气候预测模型集合
• 预训练诊断天气模型集合
• 多种在线和本地数据源，用于初始化、评分、分析等
• IO实用工具，用于将预测数据导出为用户友好的格式
• 构建集合预测的扰动方法套件
• 常见任务/用例的示例工作流程和示例
• 无缝集成到其他NVIDIA软件包，包括Modulus

要查看更完整的特性列表，请务必查看文档。没有找到您需要的功能？好消息，扩展和定制是我们设计的核心。

贡献者

有关技术要求和高级哲学、结构和设计的用户指南，请查看贡献文档。

许可证

Earth2Studio是在Apache License 2.0下提供的，请参阅LICENSE文件以获取完整的许可文本。