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.

SPONGE : A GPU‐Accelerated Molecular Dynamics Package with Enhanced Sampling and AI‐Driven Algorithms

Xponge: A Python Package to Perform Pre- and Post-Processing of Molecular Simulations.

Implementation of the moving particle semi-implicit method for free-surface flows on GPU clusters.

PySAGES: flexible, advanced sampling methods accelerated with GPUs

Molecular dynamics simulation of macromolecules using graphics processing unit

GPGPU Supported Cooperative Acceleration in Molecular Dynamics

Real-time High-Quality Surface Rendering for Large Scale Particle-Based Fluids

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

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

Graphics Processing Unit Acceleration and Parallelization of GENESIS for Large-Scale Molecular Dynamics Simulations

A GPU-accelerated package for simulation of flow in nanoporous source rocks with many-body dissipative particle dynamics

Robust and effective ab initio molecular dynamics simulations on the GPU cloud infrastructure using the Schrödinger Materials Science Suite

Employing multi-GPU power for molecular dynamics simulation: an extension of GALAMOST

GALAMOST: GPU-accelerated large-scale molecular simulation toolkit

Routine Microsecond Molecular Dynamics Simulations with AMBER on GPUs. 1. Generalized Born

ESPResSo++ 2.0: Advanced methods for multiscale molecular simulation

A GPU accelerated mixed-precision Smoothed Particle Hydrodynamics framework with cell-based relative coordinates

A massive MPI parallel framework of smoothed particle hydrodynamics with optimized memory management for extreme mechanics problems

Enhanced molecular dynamics performance with a programmable graphics processor

Exact analytical algorithm for solvent accessible surface area and derivatives in implicit solvent molecular simulations on GPUs