Rust 版 MLIP 推理引擎 llmlip.rs:比 PyTorch MACE 快 17.7 倍

CatAstro_Piyush · x · 2026-10-31

日本 mizukami-group 开源了 llmlip.rs,一个用 Rust 编写的机器学习原子间势(MLIP)推理引擎,运行时不依赖 PyTorch 或 JAX,提供 Rust 库、CLI 和 C ABI,支持能量、保守力预测及部分架构的应力/virial 计算,并附带 LAMMPS 与 OpenMM 插件,采用 Apache-2.0 等许可。

作者给出的基准颇具冲击力:在 Apple Silicon CPU(16 线程)上模拟 2000 个水分子(6000 原子)的周期性体系,MACE-Osaka24 Medium 每步耗时 llmlip.rs FP32 仅 0.74s、FP64 1.57s,而 PyTorch MACE FP64 为 13.09s、FP32 4.04s,最高提速 17.7 倍。另一组测试中,6000 原子硅晶体在 M3 Ultra 上经 Metal GPU 加速达到每步约 196ms(约 0.44 ns/day@1fs),比 CPU 路径快 5.1 倍。对做分子动力学模拟的研究者,Mac 本地就能跑大规模 MLIP 模拟。

所属事件:日本团队开源 Rust 版 MLIP 推理引擎 llmlip-rs(2 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →