DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials

Abstract

In recent years, machine learning potentials (MLPs) have become indispensable tools in physics, chemistry, and materials science, driving the development of software packages for molecular dynamics (MD) simulations and related applications. These packages, typically built on specific machine learning frameworks such as TensorFlow, PyTorch, or JAX, face integration challenges when advanced applications demand communication across different frameworks. The previous TensorFlow-based implementation of DeePMD-kit exemplified these limitations. In this work, we introduce DeePMD-kit version 3, a significant update featuring a multi-backend framework that supports TensorFlow, PyTorch, JAX, and PaddlePaddle backends, and demonstrate the versatility of this architecture through the integration of other MLPs packages and of Differentiable Molecular Force Field. This architecture allows seamless backend switching with minimal modifications, enabling users and developers to integrate DeePMD-kit with other packages using different machine learning frameworks. This innovation facilitates the development of more complex and interoperable workflows, paving the way for broader applications of MLPs in scientific research.

References in corpus (3)

• Array Programming with NumPy
• Machine learning for molecular simulation
• i-PI 3.0: a flexible and efficient framework for advanced atomistic simulations

Cited by in corpus (8)

• Unraveling Antagonistic Collision-Controlled Reactivity in Energetic Molecular Perovskites with Deep Potential Molecular Dynamics
• Machine Learning Enhanced Calculation of Quantum-Classical Binding Free Energies
• Machine Learning Potentials for Alloys: A Detailed Workflow to Predict Phase Diagrams and Benchmark Accuracy
• Comparing the latent features of universal machine-learning interatomic potentials
• Rethinking Thread Scheduling under Oversubscription: A User-Space Framework for Coordinating Multi-runtime and Multi-process Workloads
• Machine learning potential as a guide for eutectic in ultra-refractory multicomponent ceramics
• A review of simulation, measurement techniques, and development in chip thermal design
• The ground state of CuInP${}_2$S${}_6$ thin films: A study of the deep potential method