Papers › ænet-PyTorch: a GPU-supported implementation for machine learning atomic potentials training
ænet-PyTorch: a GPU-supported implementation for machine learning atomic potentials training
Jon Lopez-Zorrilla, Xabier M. Aretxabaleta, Inwon Yue, Inigo Etxebarria, Hegoi Manzano, Nongnuch Artrith
The archive published only this paper's code-link row. Authors, date and abstract are from arXiv's metadata (CC0), read from the Kaggle arXiv metadata snapshot of 2026-09-12 where its title matched the archive's; the title is the archive's.
In this work, we present {\ae}net-PyTorch, a PyTorch-based implementation for training artificial neural network-based machine learning interatomic potentials. Developed as an extension of the atomic energy network ({\ae}net), {\ae}net-PyTorch provides access to all the tools included in {\ae}net for the application and usage of the potentials. The package has been designed as an alternative to the internal training capabilities of {\ae}net, 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 {\ae}net-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.
Code
Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.
Code Syntology ran Syntology
Not run by Syntology. Nothing on this page verifies that the listed code works.
Results from the paper archive 2025-07-28
No leaderboard rows for this paper in the archive.
Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections