torch-pme:PyTorch 中高效自动微分的长程相互作用计算工具,专为分子动力学与机器学习设计。 • 支持粒子位置、电荷/偶极矩和晶胞参数的自动微分,实现力的自动计算和机器学习任务的无缝集成。 • 提供多种电荷相互作用模型:Particle-Particle Particle-Mesh Ewald (P3M)、Particle Mesh Ewald (PME)、标准Ewald及非周期方法。 • 计算不限于经典1/r库仑势,可处理任意阶1/r^p势,偶极相互作用支持1/r³势的标准Ewald计算。 • 优化 CPU 与 GPU 兼容,完全支持 TorchScript,方便转换为独立运行的 C++ 格式,满足高性能生产需求。 • 额外提供 JAX 的实验性实现 jax-pme,拓展多平台机器学习应用。 • 安装简单,支持 pip 和 conda,且可选集成 metatensor,增强张量计算能力。 • 开源BSD-3-Clause协议,文档详尽,含示例代码,助力科研与工业项目快速上手。 从底层高效实现到灵活多样的物理模型,torch-pme为材料科学和分子模拟带来更精准、可微且易扩展的长程作用力计算方案。 了解详情🔗 lab-cosmo.github.io/torch-pme 代码仓库🔗 github.com/lab-cosmo/torch-pme #分子动力学# #材料科学# #机器学习# #PyTorch# #数值模拟# #高性能计算#