DOE PAGES title logo U.S. Department of Energy
Office of Scientific and Technical Information

Title: Breaking the mold: Overcoming the time constraints of molecular dynamics on general-purpose hardware

Journal Article · · Journal of Chemical Physics
DOI: https://doi.org/10.1063/5.0249193 · OSTI ID:2520186
ORCiD logo [1]; ORCiD logo [2]; ORCiD logo [2]; ORCiD logo [3]; ORCiD logo [4]; ORCiD logo [4]; ORCiD logo [4]; ORCiD logo [4]; ORCiD logo [4]; ORCiD logo [1]; ORCiD logo [5]; ORCiD logo [2]; ORCiD logo [4]; ORCiD logo [2]
  1. Los Alamos National Laboratory (LANL), Los Alamos, NM (United States)
  2. Sandia National Lab. (SNL-NM), Albuquerque, NM (United States)
  3. Lawrence Livermore National Laboratory (LLNL), Livermore, CA (United States); Cerebras Systems Inc., Sunnyvale, CA (United States)
  4. Cerebras Systems Inc., Sunnyvale, CA (United States)
  5. Lawrence Livermore National Laboratory (LLNL), Livermore, CA (United States)

The evolution of molecular dynamics (MD) simulations has been intimately linked to that of computing hardware. For decades following the creation of MD, simulations have improved with computing power along the three principal dimensions of accuracy, atom count (spatial scale), and duration (temporal scale). Since the mid-2000s, computer platforms have, however, failed to provide strong scaling for MD, as scale-out central processing unit (CPU) and graphics processing unit (GPU) platforms that provide substantial increases to spatial scale do not lead to proportional increases in temporal scale. Important scientific problems therefore remained inaccessible to direct simulation, prompting the development of increasingly sophisticated algorithms that present significant complexity, accuracy, and efficiency challenges. While bespoke MD-only hardware solutions have provided a path to longer timescales for specific physical systems, their impact on the broader community has been mitigated by their limited adaptability to new methods and potentials. In this work, we show that a novel computing architecture, the Cerebras wafer scale engine, completely alters the scaling path by delivering unprecedentedly high simulation rates up to 1.144 M steps/s for 200 000 atoms whose interactions are described by an embedded atom method potential. This enables direct simulations of the evolution of materials using general-purpose programmable hardware over millisecond timescales, dramatically increasing the space of direct MD simulations that can be carried out. In this paper, we provide an overview of advances in MD over the last 60 years and present our recent result in the context of historical MD performance trends.

Research Organization:
Lawrence Livermore National Laboratory (LLNL), Livermore, CA (United States); Los Alamos National Laboratory (LANL), Los Alamos, NM (United States)
Sponsoring Organization:
USDOE; USDOE Laboratory Directed Research and Development (LDRD) Program; USDOE National Nuclear Security Administration (NNSA); USDOE Office of Science (SC)
Grant/Contract Number:
89233218CNA000001; AC05-00OR22725; AC52-07NA27344; NA0003525
OSTI ID:
2520186
Report Number(s):
LA-UR--24-32162; LLNL--JRNL-2000971
Journal Information:
Journal of Chemical Physics, Journal Name: Journal of Chemical Physics Journal Issue: 7 Vol. 162; ISSN 0021-9606
Publisher:
American Institute of Physics (AIP)Copyright Statement
Country of Publication:
United States
Language:
English

References (38)

Machine Learning Interatomic Potentials as Emerging Tools for Materials Science journal September 2019
Umbrella sampling: Umbrella sampling journal May 2011
Fast Parallel Algorithms for Short-Range Molecular Dynamics journal March 1995
Accelerated Molecular Dynamics Methods in a Massively Parallel World book January 2020
The embedded-atom method: a review of theory and applications journal March 1993
Accelerated Molecular Dynamics Methods: Introduction and Recent Developments book January 2009
Enhanced sampling techniques in molecular dynamics simulations of biological systems journal May 2015
The parallel replica dynamics method – Coming of age journal April 2015
LAMMPS - a flexible simulation tool for particle-based materials modeling at the atomic, meso, and continuum scales journal February 2022
Kokkos: Enabling manycore performance portability through polymorphic memory access patterns journal December 2014
Long-timescale simulation methods journal April 2005
Hardware accelerator for molecular dynamics: MDGRAPE-2 journal October 2003
Atomic scale structure of sputtered metal multilayers journal November 2001
GROMACS 4:  Algorithms for Highly Efficient, Load-Balanced, and Scalable Molecular Simulation journal February 2008
Parallel tempering: Theory, applications, and new perspectives journal January 2005
Studies in Molecular Dynamics. I. General Method journal August 1959
The midpoint method for parallelization of particle simulations journal May 2006
Extension and evaluation of the multilevel summation method for fast long-range electrostatics calculations journal June 2014
Erratum: “Perspective: Machine learning potentials for atomistic simulations” [J. Chem. Phys. 145, 170901 (2016)] journal December 2016
Enhanced sampling in molecular dynamics journal August 2019
Heterogeneous parallelization and acceleration of molecular dynamics simulations in GROMACS journal October 2020
Simulating materials failure by using up to one billion atoms and the world's fastest computer: Work-hardening journal April 2002
The importance of fluctuations in fluid mixing journal April 2007
Computer simulation of the dynamics of hydrated protein crystals and its comparison with x-ray data. journal July 1983
Forward flux sampling for rare event simulations journal October 2009
Correlations in the Motion of Atoms in Liquid Argon journal October 1964
Computer "Experiments" on Classical Fluids. I. Thermodynamical Properties of Lennard-Jones Molecules journal July 1967
Embedded-atom-method functions for the fcc metals Cu, Ag, Au, Ni, Pd, Pt, and their alloys journal June 1986
Embedded-atom-method tantalum potential developed by the force-matching method journal March 2003
Cerebras Architecture Deep Dive: First Look Inside the Hardware/Software Co-Design for Deep Learning journal May 2023
Kokkos 3: Programming Model Extensions for the Exascale Era journal January 2021
A Molecular Dynamics Computer Simulation of an Eight‐Base‐Pair DNA Fragment in Aqueous Solution: Comparison with Experimental Two‐Dimensional NMR Dataa journal December 1986
How Fast-Folding Proteins Fold journal October 2011
Trillion-Atom Molecular Dynamics Becomes a Reality journal September 2008
Anton, a special-purpose machine for molecular dynamics simulation journal July 2008
The hardware lottery journal December 2021
Parallel Molecular Dynamics With the Embedded Atom Method journal January 1992
RUMD: A general purpose molecular dynamics package optimized to utilize GPU hardware down to a few thousand particles journal January 2017