Recursively embedded atom neural network
Introduction:
Recursively embedded atom neural network (REANN) is a PyTorch-based end-to-end multi-functional Deep Neural Network Package for Molecular, Reactive and Periodic Systems. Currently, REANN can be used to train interatomic potentials, dipole moments, transition dipole moments, and polarizabilities. Taking advantage of Distributed DataParallel features embedded in PyTorch, the training process is highly parallelized on both GPU and CPU. For the convenience of MD simulation, an interface to LAMMPS has been constructed by creating a new pair_style invoking this representation for highly efficient MD simulations. More details can be found in the manual saved in the "manual" folder and an example has been placed in "reann/example/" for constructing the ML potential of the co2+Ni(100) system using the data located in the "data/" folder.
As a calculator, REANN can be used with Atomic Simulation Environment (ASE), you can read the readme in ASE/ folder for interface and the website to know how to use ASE: https://wiki.fysik.dtu.dk/ase/
Requirements:
- PyTorch 1.10.0
- LibTorch 1.10.0
- cmake 3.1.0
- opt_einsum 3.2.0
Common problem:
*RuntimeError: "Distributed package doesn't have NCCL built in" This error is induced by using the "nccl" backend when you run the package on the gpu. More details can be seen in the introduction about the parameters "DDP_backend" in manual. In the version updated on 2022-5-9, this parameters will be forced to "gloo" when the package can not find the gpu irrelative what you have set in the "input_nn".
References:
If you use this package, please cite these works.
- The original EANN model: Yaolong Zhang, Ce Hu and Bin Jiang J. Phys. Chem. Lett. 10, 4962-4967 (2019).
- The EANN model for dipole/transition dipole/polarizability: Yaolong Zhang Sheng Ye, Jinxiao Zhang, Jun Jiang and Bin Jiang J. Phys. Chem. B 124, 7284โ7290 (2020).
- The theory of REANN model: Yaolong Zhang, Junfan Xia and Bin Jiang Phys. Rev. Lett. 127, 156002 (2021).
- The details about the implementation of REANN: Yaolong Zhang, Junfan Xia and Bin Jiang J. Chem. Phys. 156, 114801 (2022).