Scream

AI模拟器STRATA实现了解决风暴的全球大气模拟

研究人员开发了STRATA，这是一种用于全球解决风暴的大气模拟的自回归AI模拟器。该模型采用了一种新颖的架构，包括3D块嵌入、局部3D邻域注意力和立体旋转位置嵌入，以4.9公里分辨率处理大气数据。与传统的物理模型相比，STRATA在能效和模拟速度方面取得了显著的改进，在512个H100 GPU上实现了每兆瓦时48个模拟日和每实际时钟日741个模拟日。