このプロジェクトについて

NVIDIA Earth2Studioは、AI駆動の天気・気候推論ワークフローを構築、研究、探索、展開するためのApache-2.0 Pythonパッケージです。予測モデル、診断モデル、データソース、出力バックエンド、摂動手法、統計操作を囲む統一APIを提供し、コンポーネントを交換したり、より複雑なパイプラインに連結したりできます。 クイックスタート例では、GFSデータとZarr出力を用いたNVIDIA FourCastNet3による決定論的予報、IFSデータを用いたECMWF AIFS、GFSデータを用いたGoogle GraphCast Operationalを示しています。その他の記載された統合には、WeatherNext 2 Cyclones、CAMulator NCAR CAM6気候エミュレータ、季節内から季節への予報向けFuXi-S2S、CorrDiff ERA5-to-HRRR生成ダウンスケーリングが含まれます。 Earth2Studioには、クラウドデータストアへの最適化されたアクセス、統計操作、推論ワークフロー例、APIドキュメント、拡張ガイダンスが含まれます。また、Claude、Codex、OpenCodeなどのコーディングエージェント向けにインストール可能なスキルを提供し、スターターワークフローの推奨、環境設定、データ取得、決定論的予報スクリプトの作成を支援します。 リポジトリにはPython 3.11から3.14のサポートバッジが表示されています。重要なライセンス上の注意点として、Earth2Studioは第三者モデル、チェックポイント、データセットと連携します。これらの資産は各提供者のライセンスの下にあり、ユーザーはそれらをダウンロード、使用、再配布する権利を確認する必要があります。NVIDIA Earth-2オープンモデルのトレーニングレシピはPhysicsNeMoリポジトリに案内されています。