Scaling the leading accuracy of deep equivariant models to biomolecular simulations of realistic size
This work brings the leading accuracy, sample efficiency, and robustness of deep equivariant neural networks to the extreme computational scale. This is achieved through a combination of innovative model architecture, massive parallelization, and models and implementations optimized for efficient GPU utilization. The resulting Allegro architecture bridges the accuracy-speed tradeoff of atomistic simulations and enables description of dynamics in structures of unprecedented complexity at quantum fidelity. To illustrate the scalability of Allegro, we perform nanoseconds-long stable simulations of protein dynamics and scale up to a 44-million atom structure of a complete, all-atom, explicitly solvated HIV capsid on the Perlmutter supercomputer. We demonstrate excellent strong scaling up to 100 million atoms and 70% weak scaling to 5120 A100 GPUs.
Code (2)
Tasks
GPUSimilar Papers 제목 키워드 기반
A Scalable and Quantum-Accurate Foundation Model for Biomolecular Force Field via Linearly Tensorized Quadrangle Attention
Accurate atomistic biomolecular simulations are vital for disease mechanism understanding, drug discovery, and biomaterial design, but existing simulation methods exhibit significant limitations. Classical force fields a…
Drug DiscoveryPerformance of universal machine-learned potentials with explicit long-range interactions in biomolecular simulations
Universal machine-learned potentials promise transferable accuracy across compositional and vibrational degrees of freedom, yet their application to biomolecular simulations remains underexplored. This work systematicall…
Period-Amplitude Co-variation in Biomolecular Oscillators
The period and amplitude of biomolecular oscillators are functionally important properties in multiple contexts. For a biomolecular oscillator, the overall constraints in how tuning of amplitude affects period, and vice …
Grappa -- A Machine Learned Molecular Mechanics Force Field
Simulating large molecular systems over long timescales requires force fields that are both accurate and efficient. In recent years, E(3) equivariant neural networks have lifted the tension between computational efficien…
Computational EfficiencyEfficient Learning of Long-Range and Equivariant Quantum Systems
In this work, we consider a fundamental task in quantum many-body physics - finding and learning ground states of quantum Hamiltonians and their properties. Recent works have studied the task of predicting the ground sta…