Abstract:In recent years, machine learning algorithms have been widely used for constructing force fields with an accuracy of ab initio methods and the efficiency of classical force fields. Here, we developed a python-based atom-centered machine-learning force field (PyAMFF) package to provide a simple and efficient platform for fitting and using machine learning force fields by implementing an atom-centered neural-network algorithm with Behler-Parrinello symmetry functions as structural fingerprints. The following three features are included in PyAMFF: (1) integrated Fortran modules for fast fingerprint calculations and Python modules for user-friendly integration through scripts and facile extension of future algorithms; (2) a pure Fortran backend to interface with the software, including the long-timescale dynamic simulation package EON, enabling both molecular dynamic simulations and adaptive kinetic Monte Carlo simulations with machine-learning force fields; and (3) integration with the Atomic Simulation Environment package for active learning and ML-based algorithm development. Here, we demonstrate an efficient parallelization of PyAMFF in terms of CPU and memory usage and show that the Fortran-based PyAMFF calculator exhibits a linear scaling relationship with the number of symmetry functions and the system size. Program summary Program title: python-based atom-centered machine-learning force field (PyAMFF) CPC Library link to program files: https://doi.org/10.17632/fsn6dkcvrv.1 Developer's repository link: https://gitlab.com/pyamff/pyamff Licensing provisions: Apache License, 2.0 Nature of problem: Determine an approximate (surrogate) model based upon atomic forces and energies from density functional theory (DFT). With a surrogate model that is less computationally expensive to evaluate than DFT, there can be a rapid exploration of the potential energy surface, accelerated optimization to minima and saddle points, and ultimately, accelerated design of active materials where the kinetics are key to the material function. Solution method: The atomic environments of training data are calculated in terms of Behler-Parrinello fingerprints. These fingerprints are passed to a neural network which is trained to reproduce the energy and force of the training data. A parallel implementation and Fortran backend allow for efficient training and calculation of the resulting surrogate model. Examples of long-time simulations of materials on the surrogate model surfaces are provided.

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

PhyNEO: A Neural-Network-Enhanced Physics-Driven Force Field Development Workflow for Bulk Organic Molecule and Polymer Simulations.

Incorporating Neural Networks into the AMOEBA Polarizable Force Field

Force-Field-Enhanced Neural Network Interactions: from Local Equivariant Embedding to Atom-in-Molecule properties and long-range effects

Scalable Hybrid Deep Neural Networks/Polarizable Potentials Biomolecular Simulations including long-range effects

A Scalable Graph Neural Network Method For Developing An Accurate Force Field Of Large Flexible Organic Molecules

QuantumBind-RBFE: Accurate Relative Binding Free Energy Calculations Using Neural Network Potentials

LAMMPS Implementation of Rapid Artificial Neural Network Derived Interatomic Potentials

Fast and Sample-Efficient Interatomic Neural Network Potentials for Molecules and Materials Based on Gaussian Moments

Scalable Training of Neural Network Potentials for Complex Interfaces Through Data Augmentation

Library-Based LAMMPS Implementation of High-Dimensional Neural Network Potentials.

Accurate and scalable multi-element graph neural network force field and molecular dynamics with direct force architecture

Accurate and scalable graph neural network force field and molecular dynamics with direct force architecture

Reweighting from Molecular Mechanics Force Fields to the ANI-2x Neural Network Potential

A Scalable Molecular Force Field Parameterization Method Based on Density Functional Theory and Quantum-Level Machine Learning

Atom-centered machine-learning force field package

AIMNet2: A Neural Network Potential to Meet your Neutral, Charged, Organic, and Elemental-Organic Needs

PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials