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

TL;DR

This paper demonstrates that the Cerebras Wafer Scale Engine enables molecular dynamics simulations at unprecedented speeds on general-purpose hardware, overcoming traditional hardware limitations and allowing millisecond-scale simulations of materials.

Contribution

The study introduces a novel computing architecture that significantly accelerates MD simulations, enabling direct millisecond-scale simulations on general-purpose hardware.

Findings

Achieved up to 1.144 million steps per second for 200,000 atoms

Enabled direct simulation of material evolution over milliseconds

Showed that new hardware architecture surpasses traditional scaling limits

Abstract

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 CPU and 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.144M steps/second 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.