FeNNol: an Efficient and Flexible Library for Building Force-field-enhanced Neural Network Potentials

Abstract

Neural network interatomic potentials (NNPs) have recently proven to be powerful tools to accurately model complex molecular systems while bypassing the high numerical cost of ab-initio molecular dynamics simulations. In recent years, numerous advances in model architectures as well as the development of hybrid models combining machine-learning (ML) with more traditional, physically-motivated, force-field interactions have considerably increased the design space of ML potentials. In this paper, we present FeNNol, a new library for building, training and running force-field-enhanced neural network potentials. It provides a flexible and modular system for building hybrid models, allowing to easily combine state-of-the-art embeddings with ML-parameterized physical interaction terms without the need for explicit programming. Furthermore, FeNNol leverages the automatic differentiation and just-in-time compilation features of the Jax Python library to enable fast evaluation of NNPs, shrinking the performance gap between ML potentials and standard force-fields. This is demonstrated with the popular ANI-2x model reaching simulation speeds nearly on par with the AMOEBA polarizable force-field on commodity GPUs (GPU=Graphics processing unit). We hope that FeNNol will facilitate the development and application of new hybrid NNP architectures for a wide range of molecular simulation problems.

References in corpus (33)

• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• 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
• PhysNet: A Neural Network for Predicting Energies, Forces, Dipole Moments and Partial Charges
• Towards Exact Molecular Dynamics Simulations with Machine-Learned Force Fields
• 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
• i-PI 2.0: A Universal Force Engine for Advanced Molecular Simulations
• SpookyNet: Learning Force Fields with Electronic Degrees of Freedom and Nonlocal Effects
• Hierarchical modeling of molecular energies using a deep neural network
• On the accuracy of the MB-pol many-body potential for water: Interaction energies, vibrational frequencies, and classical thermodynamic and dynamical properties from clusters to liquid water and ice
• Minimal Basis Iterative Stockholder: Atoms in Molecules for Force-Field Development
• A deep potential model with long-range electrostatic interactions
• Incorporating long-range physics in atomic-scale machine learning
• Rational Construction of Stochastic Numerical Methods for Molecular Sampling
• QM7-X: A comprehensive dataset of quantum-mechanical properties spanning the chemical space of small organic molecules
• Transferable atomic multipole machine learning models for small organic molecules
• TeaNet: universal neural network interatomic potential inspired by iterative electronic relaxations
• Gaussian Moments as Physically Inspired Molecular Descriptors for Accurate and Scalable Machine Learning Potentials
• End-to-End Differentiable Molecular Mechanics Force Field Construction
• Informing Geometric Deep Learning with Electronic Interactions to Accelerate Quantum Chemistry
• Tinker-HP : Accelerating Molecular Dynamics Simulations of Large Complex Systems with Advanced Point Dipole Polarizable Force Fields using GPUs and Multi-GPUs systems
• SchNetPack 2.0: A neural network toolbox for atomistic machine learning
• Accurate biomolecular simulations account for electronic polarization
• MLatom 3: Platform for machine learning-enhanced computational chemistry simulations and workflows
• Scalable Hybrid Deep Neural Networks/Polarizable Potentials Biomolecular Simulations including long-range effects
• Force-Field-Enhanced Neural Network Interactions: from Local Equivariant Embedding to Atom-in-Molecule properties and long-range effects
• Lifelong Machine Learning Potentials
• Development of the Quantum Inspired SIBFA Many-Body Polarizable Force Field: Enabling Condensed Phase Molecular Dynamics Simulations
• Nuclear Quantum Effects in liquid water at near classical computational cost using the adaptive Quantum Thermal Bath
• Routine Molecular Dynamics Simulations Including Nuclear Quantum Effects: from Force Fields to Machine Learning Potentials

Cited by in corpus (7)

• chemtrain: Learning Deep Potential Models via Automatic Differentiation and Statistical Physics
• Apax: A Flexible and Performant Framework For The Development of Machine-Learned Interatomic Potentials
• Lambda-ABF-OPES: Faster Convergence with High Accuracy in Alchemical Free Energy Calculations
• Automated Machine Learning Pipeline: Large Language Models-Assisted Automated Dataset Generation for Training Machine-Learned Interatomic Potentials
• Fast, Modular, and Differentiable Framework for Machine Learning-Enhanced Molecular Simulations
• Differentiable hybrid force fields support scalable autonomous electrolyte discovery
• Faster Molecular Dynamics with Neural Network Potentials via Distilled Multiple Time-Stepping and Non-Conservative Forces