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.

Integrating physics in deep learning algorithms: a force field as a PyTorch module

DeepPhysics: a physics aware deep learning framework for real-time simulation

Bridging deep learning force fields and electronic structures with a physics-informed approach

TorchMD: A deep learning framework for molecular simulations

Synergistic integration of physical embedding and machine learning enabling precise and reliable force field

Enhanced force-field calibration via machine learning

Deep Lagrangian Networks: Using Physics as Model Prior for Deep Learning

Shared Data and Algorithms for Deep Learning in Fundamental Physics

Physics-Guided Deep Learning for Drag Force Prediction in Dense Fluid-Particulate Systems

A physics-informed operator regression framework for extracting data-driven continuum models

Physics-driven machine learning models coupling PyTorch and Firedrake

Combining physics and deep learning to learn continuous-time dynamics models

Integrating Machine Learning with Physics-Based Modeling

Deep learning-accelerated computational framework based on Physics Informed Neural Network for the solution of linear elasticity

Differentiable programming across the PDE and Machine Learning barrier

DeePMD-kit: A deep learning package for many-body potential energy representation and molecular dynamics

Deep Physics Corrector: A physics enhanced deep learning architecture for solving stochastic differential equations

Atom-centered machine-learning force field package

Physics-Aware Deep Learning on Multiphase Flow Problems

Physics-informed machine learning