TorchMD: A deep learning framework for molecular simulations

12/22/2020
by   Stefan Doerr, et al.
0

Molecular dynamics simulations provide a mechanistic description of molecules by relying on empirical potentials. The quality and transferability of such potentials can be improved leveraging data-driven models derived with machine learning approaches. Here, we present TorchMD, a framework for molecular simulations with mixed classical and machine learning potentials. All of force computations including bond, angle, dihedral, Lennard-Jones and Coulomb interactions are expressed as PyTorch arrays and operations. Moreover, TorchMD enables learning and simulating neural network potentials. We validate it using standard Amber all-atom simulations, learning an ab-initio potential, performing an end-to-end training and finally learning and simulating a coarse-grained model for protein folding. We believe that TorchMD provides a useful tool-set to support molecular simulations of machine learning potentials. Code and data are freely available at <github.com/torchmd>.

READ FULL TEXT
research
09/21/2022

SPICE, A Dataset of Drug-like Molecules and Peptides for Training Machine Learning Potentials

Machine learning potentials are an important tool for molecular simulati...
research
12/18/2018

Molecular Dynamics with Neural-Network Potentials

Molecular dynamics simulations are an important tool for describing the ...
research
12/14/2022

Machine Learning Coarse-Grained Potentials of Protein Thermodynamics

A generalized understanding of protein dynamics is an unsolved scientifi...
research
12/20/2022

Combining machine-learned and empirical force fields with the parareal algorithm: application to the diffusion of atomistic defects

We numerically investigate an adaptive version of the parareal algorithm...
research
12/09/2019

JAX, M.D.: End-to-End Differentiable, Hardware Accelerated, Molecular Dynamics in Pure Python

A large fraction of computational science involves simulating the dynami...
research
10/09/2022

Hyperactive Learning (HAL) for Data-Driven Interatomic Potentials

Data-driven interatomic potentials have emerged as a powerful class of s...
research
01/20/2022

NNP/MM: Fast molecular dynamics simulations with machine learning potentials and molecular mechanics

Parametric and non-parametric machine learning potentials have emerged r...

Please sign up or login with your details

Forgot password? Click here to reset