neuralgcm/dinosaur

何を解決するか

Dinosaur は、グローバル大気モデリング用の微分可能な動的コアを提供します。AIを用いた現代の気象・気候モデルが、物理法則（浅水域方程式や原始方程式など）と機械学習を統合する必要に対応し、ハイブリッドモデルの「オンライン学習」を可能にします。

動作方法

JAX で記述されており、シグマ座標上で大気方程式をスペクトル法で解きます。JAX に基づいているため、前向きおよび後向きの自動微分をサポートしており、GPU や TPU での高速化に最適化されており、マルチデバイス並列処理も可能になっています。

対象ユーザー

AI駆動の気象予測、気候モデリング、ハイブリッド物理-AIシステムに取り組む研究者や開発者。

特徴

• 微分可能な動力学: ハイブリッドAI/物理モデルの学習に自動微分をサポート。
• ハードウェア加速: GPU および TPU での効率的な実行を設計。
• スペクトル法: 湿潤および乾燥状態の浅水域方程式および原始方程式を解く。
• 柔軟な輸送: 半ラグランジュ輸送オプションを備えており、時間刻みを6〜12倍に延長可能。