モデル・研究
NVIDIA STRATAのウェイトを公開
原題: Earth-2 Checkpoints: STRATA
全球のstorm-resolving大気をtile単位で予測
NVIDIAは2026年11月2日、全球大気model STRATAのcheckpointをHugging Faceで公開しました。
STRATAは331.7M parameterのTransformerで、全球storm-resolving simulationをemulateします。
約4.9km格子の全球大気を10分刻みで自己回帰的に予測します。
気温、風、geopotential height、鉛直気圧速度、比湿などを24のmodel levelで扱います。
全球stateを重なりのあるtileへ分割し、単一GPUでは一部領域、複数GPUでは全球を進められます。
checkpoint、normalization statistics、channel定義、static forcing fieldを配布しています。
training recipeとinference wrapperはNVlabsの公式GitHubでApache 2.0により公開されています。
model weightsはOpenMDW License 1.1で、商用・非商用利用に対応するとmodel cardへ記載されています。
Ampere、Hopper、BlackwellのNVIDIA GPUとLinuxが対応環境です。
評価値はNVIDIAのmodel cardと論文に基づく提供元評価であり、独立した実運用評価とは区別が必要です。