Abstract:We present the GPU version of DeePMD-kit, which, upon training a deep neural network model using ab initio data, can drive extremely large-scale molecular dynamics (MD) simulation with ab initio accuracy. Our tests show that for a water system of <math>12,582,912</math> atoms, the GPU version can be 7 times faster than the CPU version under the same power consumption. The code can scale up to the entire Summit supercomputer. For a copper system of <math>113,246,208</math> atoms, the code can perform one nanosecond MD simulation per day, reaching a peak performance of 86 PFLOPS (43% of the peak). Such unprecedented ability to perform MD simulation with ab initio accuracy opens up the possibility of studying many important issues in materials and molecules, such as heterogeneous catalysis, electrochemical cells, irradiation damage, crack propagation, and biochemical reactions.<h3 class="u-h4 u-margin-m-top u-margin-xs-bottom">Program summary</h3>Program Title: DeePMD-kitCPC Library link to program files: <a href="https://dx.doi.org/10.17632/phyn4kgsfx.1">http://dx.doi.org/10.17632/phyn4kgsfx.1</a>Developer's repository link: <a href="https://dx.doi.org/10.5281/zenodo.3961106">http://dx.doi.org/10.5281/zenodo.3961106</a>Licensing provisions: LGPLProgramming language: C++/Python/CUDAJournal reference of previous version: Comput. Phys. Commun. 228 (2018), 178–184.Does the new version supersede the previous version?: Yes.Reasons for the new version: Parallelize and optimize the DeePMD-kit for modern high performance computers.Summary of revisions: The optimized DeePMD-kit is capable of computing 100 million atoms molecular dynamics with ab initio accuracy, achieving 86 PFLOPS in double precision.Nature of problem: Modeling the many-body atomic interactions by deep neural network models. Running molecular dynamics simulations with the models.Solution method: The Deep Potential for Molecular Dynamics (DeePMD) method is implemented based on the deep learning framework TensorFlow. Standard and customized TensorFlow operators are optimized for GPU. Massively parallel molecular dynamics simulations with DeePMD models on high performance computers are supported in the new version.

Scalable hybrid deep neural networks/polarizable potentials biomolecular simulations including long-range effects

Scalable Hybrid Deep Neural Networks/Polarizable Potentials Biomolecular Simulations including long-range effects

Tinker-HP : Accelerating Molecular Dynamics Simulations of Large Complex Systems with Advanced Point Dipole Polarizable Force Fields using GPUs and Multi-GPUs systems

Incorporating Neural Networks into the AMOEBA Polarizable Force Field

FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentials

PhyNEO: A Neural-Network-Enhanced Physics-Driven Force Field Development Workflow for Bulk Organic Molecule and Polymer Simulations.

Deep Potential Molecular Dynamics: A Scalable Model with the Accuracy of Quantum Mechanics

DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics

Routine Molecular Dynamics Simulations Including Nuclear Quantum Effects: From Force Fields to Machine Learning Potentials

Neural Network Potential with Multi-Resolution Approach Enables Accurate Prediction of Reaction Free Energies in Solution

Scalable Training of Neural Network Potentials for Complex Interfaces Through Data Augmentation

Adaptive coupling of a deep neural network potential to a classical force field

A deep potential model with long-range electrostatic interactions

86 PFLOPS Deep Potential Molecular Dynamics simulation of 100 million atoms with ab initio accuracy

Parallel reactive molecular dynamics: Numerical methods and algorithmic techniques

Towards Electronic Structure-Based Ab-Initio Molecular Dynamics Simulations with Hundreds of Millions of Atoms

Raising the Performance of the Tinker-HP Molecular Modeling Package [Article v1.0]

AENET-LAMMPS and AENET-TINKER: Interfaces for Accurate and Efficient Molecular Dynamics Simulations with Machine Learning Potentials

NNP/MM: Accelerating Molecular Dynamics Simulations with Machine Learning Potentials and Molecular Mechanics