Breaking the Molecular Dynamics Timescale Barrier Using a Wafer-Scale System

TL;DR

This paper demonstrates that using a wafer-scale system for molecular dynamics simulations significantly accelerates computations, enabling the study of slow processes previously infeasible within practical timeframes.

Contribution

It introduces a novel approach leveraging wafer-scale hardware to massively improve MD simulation speed and energy efficiency, surpassing traditional supercomputers.

Findings

179-fold increase in timesteps per second over GPU-based platforms

Simulation of up to 800,000 atoms at over 270,000 timesteps/sec

Enables reduction of simulation runtimes from years to days

Abstract

Molecular dynamics (MD) simulations have transformed our understanding of the nanoscale, driving breakthroughs in materials science, computational chemistry, and several other fields, including biophysics and drug design. Even on exascale supercomputers, however, runtimes are excessive for systems and timescales of scientific interest. Here, we demonstrate strong scaling of MD simulations on the Cerebras Wafer-Scale Engine. By dedicating a processor core for each simulated atom, we demonstrate a 179-fold improvement in timesteps per second versus the Frontier GPU-based Exascale platform, along with a large improvement in timesteps per unit energy. Reducing every year of runtime to two days unlocks currently inaccessible timescales of slow microstructure transformation processes that are critical for understanding material behavior and function. Our dataflow algorithm runs Embedded Atom Method (EAM) simulations at rates over 270,000 timesteps per second for problems with up to 800k atoms. This demonstrated performance is unprecedented for general-purpose processing cores.