TorchNEP: Ultra-Efficient and Accurate Training of Neuroevolution Potentials

Physics > Computational Physics [Submitted on 17 Jun 2026] Title:TorchNEP: Ultra-Efficient and Accurate Training of Neuroevolution Potentials View PDF HTML (experimental)Abstract:Neuroevolution Potential (NEP) is one of the most efficient machine-learned interatomic potential frameworks for large-scale atomistic simulations. However, its original training strategy remains computationally demanding, limiting systematic exploration of model architectures and training protocols. Here, we present TorchNEP, a PyTorch-based implementation of NEP that combines analytically derived gradients, adaptive optimization, and a two-stage training strategy. TorchNEP accelerates training by more than two orders of magnitude while maintaining full compatibility with existing NEP models. We further show that the improvement in predictive accuracy primarily originates from the two-stage training protocol rather than the optimization algorithm itself. Across diverse benchmark datasets, TorchNEP consistently improves force and stress predictions while maintaining comparable or improved energy accuracy. Benchmark evaluations on elemental and alloy systems demonstrate enhanced predictive performance for both atomic configurations and key materials properties. Furthermore, we show that increasing model complexity does not necessarily improve predictive performance despite reducing training errors. Overall, TorchNEP provides an efficient and flexible training framework for developing more accurate and robust machine-learned interatomic potentials. Current browse context: physics.comp-ph Change to browse by: References & Citations Loading... Bibliographic and Citation Tools Bibliographic Explorer (What is the Explorer?) Connected Papers (What is Connected Papers?) Litmaps (What is Litmaps?) scite Smart Citations (What are Smart Citations?) Code, Data and Media Associated with this Article alphaXiv (What is alphaXiv?) CatalyzeX Code Finder for Papers (What is CatalyzeX?) DagsHub (What is DagsHub?) Gotit.pub (What is GotitPub?) Hugging Face (What is Huggingface?) ScienceCast (What is ScienceCast?) Demos Recommenders and Search Tools Influence Flower (What are Influence Flowers?) CORE Recommender (What is CORE?) arXivLabs: experimental projects with community collaborators arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website. Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them. Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.