1. Stream-AI-MD
- Author
-
Heng Ma, Hyunseung Yoo, Vishal Subbiah, Thomas D. Uram, Alexander Brace, Austin Clyde, Corey Adams, Jessica Liu, Venkatram Vishwanath, Andew Hock, Michael A. Salim, Murali Emani, Anda Trifa, and Arvind Ramanathan
- Subjects
Computer science ,business.industry ,Deep learning ,Severe acute respiratory syndrome coronavirus 2 (SARS-CoV-2) ,Molecular biophysics ,Symmetric multiprocessor system ,Parallel computing ,Folding (DSP implementation) ,Supercomputer ,ComputingMethodologies_PATTERNRECOGNITION ,Workflow ,Leverage (statistics) ,Artificial intelligence ,business - Abstract
Emerging hardware tailored for artificial intelligence (AI) and machine learning (ML) methods provide novel means to couple them with traditional high performance computing (HPC) workflows involving molecular dynamics (MD) simulations. We propose Stream-AI-MD, a novel instance of applying deep learning methods to drive adaptive MD simulation campaigns in a streaming manner. We leverage the ability to run ensemble MD simulations on GPU clusters, while the data from atomistic MD simulations are streamed continuously to AI/ML approaches to guide the conformational search in a biophysically meaningful manner on a wafer-scale AI accelerator. We demonstrate the efficacy of Stream-AI-MD simulations for two scientific use-cases: (1) folding a small prototypical protein, namely ββα-fold (BBA) FSD-EY and (2) understanding protein-protein interaction (PPI) within the SARS-CoV-2 proteome between two proteins, nsp16 and nsp10. We show that Stream-AI-MD simulations can improve time-to-solution by ~50X for BBA protein folding. Further, we also discuss performance trade-offs involved in implementing AI-coupled HPC workflows on heterogeneous computing architectures.
- Published
- 2021