TorchMD-Net 2.0: Fast Neural Network Potentials for Molecular Simulations

Abstract

Achieving a balance between computational speed, prediction accuracy, and universal applicability in molecular simulations has been a persistent challenge. This paper presents substantial advancements in the TorchMD-Net software, a pivotal step forward in the shift from conventional force fields to neural network-based potentials. The evolution of TorchMD-Net into a more comprehensive and versatile framework is highlighted, incorporating cutting-edge architectures such as TensorNet. This transformation is achieved through a modular design approach, encouraging customized applications within the scientific community. The most notable enhancement is a significant improvement in computational efficiency, achieving a very remarkable acceleration in the computation of energy and forces for TensorNet models, with performance gains ranging from 2-fold to 10-fold over previous iterations. Other enhancements include highly optimized neighbor search algorithms that support periodic boundary conditions and the smooth integration with existing molecular dynamics frameworks. Additionally, the updated version introduces the capability to integrate physical priors, further enriching its application spectrum and utility in research. The software is available at https://github.com/torchmd/torchmd-net.

Version accepted in Journal of Chemical Theory and Computation

References in corpus (17)

• SchNet - a deep learning architecture for molecules and materials
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
• Less is more: sampling chemical space with active learning
• DeePMD-kit v2: A software package for Deep Potential models
• A Fourth-Generation High-Dimensional Neural Network Potential with Accurate Electrostatics Including Non-local Charge Transfer
• Ab initio thermodynamics of liquid and solid water
• SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effects
• TorchMD: A deep learning framework for molecular simulations
• Machine Learning Coarse-Grained Potentials of Protein Thermodynamics
• SchNetPack 2.0: A neural network toolbox for atomistic machine learning
• Optimizing CUDA Code By Kernel Fusion---Application on BLAS
• Accelerating Dissipative Particle Dynamics Simulations on GPUs: Algorithms, Numerics and Applications
• Data efficiency and extrapolation trends in neural network interatomic potentials

Cited by in corpus (9)

• mdCATH: A Large-Scale MD Dataset for Data-Driven Computational Biophysics
• Neural Network Potential with Multi-Resolution Approach Enables Accurate Prediction of Reaction Free Energies in Solution
• Machine Learning Enhanced Calculation of Quantum-Classical Binding Free Energies
• Two-dimensional electronic spectroscopy in the condensed phase using equivariant transformer accelerated molecular dynamics simulations
• AMARO: All Heavy-Atom Transferable Neural Network Potentials of Protein Thermodynamics
• Scaling Graph Neural Networks to Large Proteins
• Two-dimensional electronic spectra from trajectory-based dynamics: pure-state Ehrenfest, spin-mapping, and mean classical path approaches
• Artificial Intelligence for Direct Prediction of Molecular Dynamics Across Chemical Space
• Knowledge Distillation of a Protein Language Model Yields a Foundational Implicit Solvent Model