In this work, we present aenet-PyTorch, a PyTorch-based implementation for training artificial neural network-based machine learning interatomic potentials. Developed as an extension of the atomic energy network (aenet), aenet-PyTorch provides access to all the tools included in aenet for the application and usage of the potentials. The package has been designed as an alternative to the internal training capabilities of aenet, leveraging the power of graphic processing units to facilitate direct training on forces in addition to energies. This leads to a substantial reduction of the training time by one to two orders of magnitude compared to the CPU implementation, enabling direct training on forces for systems beyond small molecules. Here we demonstrate the main features of aenet-PyTorch and show its performance on open databases. Our results show that training on all the force information within a data set is not necessary, and including between 10% to 20% of the force information is sufficient to achieve optimally accurate interatomic potentials with the least computational resources.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.
customersupport@researchsolutions.com
10624 S. Eastern Ave., Ste. A-614
Henderson, NV 89052, USA
This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
Copyright © 2025 scite LLC. All rights reserved.
Made with 💙 for researchers
Part of the Research Solutions Family.