Molecular Dynamics Simulations on Cloud Computing and Machine Learning Platforms

Scientific computing applications have benefited greatly from high performance computing infrastructure such as supercomputers. However, we are seeing a paradigm shift in the computational structure, design, and requirements of these applications. Increasingly, data-driven and machine learning approaches are being used to support, speed-up, and enhance scientific computing applications, especially molecular dynamics simulations. Concurrently, cloud computing platforms are increasingly appealing for scientific computing, providing “infi-nite” computing powers, easier programming and deployment models, and access to computing accelerators such as TPUs (Tensor Processing Units). This confluence of machine learning (ML) and cloud computing represents exciting opportunities for cloud and systems researchers. ML-assisted molecular dynamics simulations are a new class of workload, and exhibit unique computational patterns. These simulations present new challenges for low-cost and high-performance execution. We argue that transient cloud resources, such as low-cost preemptible cloud VMs, can be a viable platform for this new workload. Finally, we present some low-hanging fruits and long-term challenges in cloud resource management, and the integration of molecular dynamics simulations into ML platforms (such as TensorFlow).

[1]  Geoffrey C. Fox,et al.  Machine learning surrogates for molecular dynamics simulations of soft materials , 2020, J. Comput. Sci..

[2]  JCS Kadupitige,et al.  Modeling The Temporally Constrained Preemptions of Transient Cloud VMs , 2020, HPDC.

[3]  Andrew L. Ferguson,et al.  Machine learning and data science in soft materials engineering , 2018, Journal of physics. Condensed matter : an Institute of Physics journal.

[4]  F. Solis,et al.  Ionic structure and decay length in highly concentrated confined electrolytes , 2020, AIP Advances.

[5]  Sharon C. Glotzer,et al.  Rational design of nanomaterials from assembly and reconfigurability of polymer-tethered nanoparticles , 2015 .

[6]  Lei Huang,et al.  AI-Driven Multiscale Simulations Illuminate Mechanisms of SARS-CoV-2 Spike Dynamics , 2020, bioRxiv.

[7]  D. Heyes,et al.  Advances in nonequilibrium molecular dynamics simulations of lubricants and additives , 2018 .

[8]  Geoffrey C. Fox,et al.  Machine Learning for Performance Enhancement of Molecular Dynamics Simulations , 2019, ICCS.

[9]  M. Hagan,et al.  Recent advances in coarse-grained modeling of virus assembly. , 2016, Current opinion in virology.

[10]  Andrew D. White,et al.  HOOMD-TF: GPU-Accelerated, Online Machine Learning in the HOOMD-blue Molecular Dynamics Engine , 2020, J. Open Source Softw..

[11]  Samuel S. Schoenholz,et al.  JAX MD: A Framework for Differentiable Physics , 2020, NeurIPS.

[12]  Toni Giorgino,et al.  TorchMD: A Deep Learning Framework for Molecular Simulations , 2020, Journal of Chemical Theory and Computation.

[13]  Frank Noé,et al.  Machine Learning of Coarse-Grained Molecular Dynamics Force Fields , 2018, ACS central science.

[14]  Renato Luiz de Freitas Cunha,et al.  HPC Cloud for Scientific and Business Applications: Taxonomy, Vision, and Research Challenges , 2018, ACM Comput. Surv..

[15]  Xiang Gao,et al.  TorchANI: A Free and Open Source PyTorch-Based Deep Learning Implementation of the ANI Neural Network Potentials , 2020, J. Chem. Inf. Model..

[16]  Geoffrey C. Fox,et al.  Machine learning for parameter auto-tuning in molecular dynamics simulations: Efficient dynamics of ions near polarizable nanoparticles , 2019, Int. J. High Perform. Comput. Appl..

[17]  David W Toth,et al.  The TensorMol-0.1 model chemistry: a neural network augmented with long-range physics , 2017, Chemical science.

[18]  N. Aluru,et al.  Molecular Dynamics Properties without the Full Trajectory: A Denoising Autoencoder Network for Properties of Simple Liquids. , 2019, The journal of physical chemistry letters.

[19]  Steve Plimpton,et al.  Fast parallel algorithms for short-range molecular dynamics , 1993 .

[20]  Sharon C. Glotzer,et al.  HOOMD-blue: A Python package for high-performance molecular dynamics and hard particle Monte Carlo simulations , 2013 .

[21]  J. Kadupitiya,et al.  Probing the Rheological Properties of Liquids Under Conditions of Elastohydrodynamic Lubrication Using Simulations and Machine Learning , 2021, Tribology Letters.