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

CatAstro_Piyush · x · 2026-10-31

日本 Mizukami 团队开源 llmlip-rs,一个用 Rust 编写的机器学习原子间势(MLIP)推理引擎,运行时完全不依赖 PyTorch 或 JAX,提供 Rust 库、CLI 和 C ABI,并附带 LAMMPS 与 OpenMM 插件,Apache-2.0 协议。

性能数据(Apple Silicon 16 线程 CPU,2000 个水分子共 6000 原子):PyTorch MACE FP64 每步 13.09 秒,llmlip-rs FP32 仅 0.74 秒,快 17.7 倍。配图还显示在 Mac 上通过 Metal 跑 6000 原子硅晶体每步约 196 毫秒,比 CPU 快 5.1 倍。对分子模拟/AI for Science 的本地大规模推理很有实用价值。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →