arXiv · 2410.22867
Scaling Molecular Dynamics with ab initio Accuracy to 149 Nanoseconds per Day
Abstract
Physical phenomena such as chemical reactions, bond breaking, and phase transition require molecular dynamics (MD) simulation with ab initio accuracy ranging from milliseconds to microseconds. However, previous state-of-the-art neural network based MD packages such as DeePMD-kit can only reach 4.7 nanoseconds per day on the Fugaku supercomputer. In this paper, we present a novel node-based parallelization scheme to reduce communication by 81%, then optimize the computationally intensive kernels with sve-gemm and mixed precision. Finally, we implement intra-node load balance to further improve the scalability. Numerical results on the Fugaku supercomputer show that our work has significantly improved the time-to-solution of the DeePMD-kit by a factor of 31.7x, reaching 149 nanoseconds per day on 12,000 computing nodes. This work has opened the door for millisecond simulation with ab initio accuracy within one week for the first time.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jianxiong Li, Boyang Li, Zhuoqiang Guo, Mingzhen Li, Enji Li, Lijun Liu, Guojun Yuan, Zhan Wang, Guangming Tan, Weile Jia. 2024-10-30. Scaling Molecular Dynamics with ab initio Accuracy to 149 Nanoseconds per Day. https://doi.org/10.1109/sc41406.2024.00036
Cite the original work for its findings. Save a collection to share your selection of sources.