arXiv · Machine Learning· Denghui Lu, Alexander Maeder, Alexandros Nikolaos Ziogas, Mathieu Luisier, Manasa Kaniselvan·· 10 小时前AI 评分44
FlashSO(2)通过低精度模拟加速 SO(2)-等变材料模型
Accelerating SO(2)-Equivariant Materials Models through Low-Precision Emulation
AI 导读
FlashSO(2)采用整数 Ozaki 方案模拟低精度运算，加速 GNN 的 SO(2)-等变层。在 NVIDIA GH200 上，层级吞吐最高达 236（MLH）和 232（MLIP）有效 Tflop/s，分别达到 MALOQ、UMA 参考实现的 40 倍和 7 倍。
来源：arXiv · Machine Learning · arxiv.org