RecodeX 重构消息,英伟达在技术博客中公布,在由17280个原子构成的水分子系统上,B300(Blackwell)GPU采用BF16x9矩阵乘法方案运行PET分子动力学计算,速度最高达到FP32方式的1.7倍。该数据以分子动力学完整单步计算为基准,而非仅统计矩阵乘法。

另一方案TF32x3则在A100 GPU上测试。PET(Point Edge Transformer)是一种机器学习原子间势,可根据原子结构估算能量与原子间作用力。上述性能数据来自英伟达针对特定模型与设备进行的自测。