deepmodeling/deepmd-kit

解决的问题

DeePMD-kit 将量子力学参考数据转换为快速、可扩展的原子间势函数。这使得分子和材料科学领域的研究人员能够在无需承担完整量子力学计算的计算成本的情况下，对有限分子、共价体系、周期性固体和金属进行大规模模拟。

工作原理

该工具包提供了一套工作流程，支持从头训练模型，或从预训练的 Deep Potential (DPA4) 模型开始并针对特定体系进行微调。它支持多种后端，包括 TensorFlow、PyTorch、JAX 和 Paddle。训练完成后，模型可被压缩以实现更快的推理，并可导出用于各种分子动力学引擎。

适用人群

专为需要高性能原子间势函数进行分子动力学模拟的分子和材料科学领域的科学家和工程师设计。

主要亮点

• 预训练优先工作流：支持下载并微调 DPA4 模型。
• 灵活的模型组合：提供 DPA4（高精度）和 DPA4C（高吞吐量）两种模型。
• 广泛的物理目标：可建模能量、力、virials、Hessians、自旋、磁力、偶极矩和电子态密度。
• 后端无关性：兼容 PyTorch、TensorFlow、JAX 和 Paddle。
• 丰富的集成能力：可与 LAMMPS、i-PI、ASE、GROMACS 和 CP2K 等模拟引擎连接。