Ruosong Shi | Searcharxiv

arXiv2025

DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials

In recent years, machine learning potentials (MLPs) have become indispensable tools in physics, chemistry, and materials science, driving the development of software packages for molecular dynamics (MD) simulations and related applications. These packages, typically built on specific machine learning frameworks such as TensorFlow, PyTorch, or JAX, face integration challenges when advanced applications demand communication across different frameworks. The previous TensorFlow-based implementation of DeePMD-kit exemplified these limitations. In this work, we introduce DeePMD-kit version 3, a significant update featuring a multi-backend framework that supports TensorFlow, PyTorch, JAX, and PaddlePaddle backends, and demonstrate the versatility of this architecture through the integration of other MLPs packages and of Differentiable Molecular Force Field. This architecture allows seamless backend switching with minimal modifications, enabling users and developers to integrate DeePMD-kit with other packages using different machine learning frameworks. This innovation facilitates the development of more complex and interoperable workflows, paving the way for broader applications of MLPs in scientific research.

physics.chem-ph↗

arXiv2023

Deep Learning Illuminates Spin and Lattice Interaction in Magnetic Materials

Atomistic simulations hold significant value in clarifying crucial phenomena such as phase transitions and energy transport in materials science. Their success stems from the presence of potential energy functions capable of accurately depicting the relationship between system energy and lattice changes. In magnetic materials, two atomic scale degrees of freedom come into play: the lattice and the spin. However, accurately tracing the simultaneous evolution of both lattice and spin in magnetic materials at an atomic scale is a substantial challenge. This is largely due to the complexity involved in depicting the interaction energy precisely, and its influence on lattice and spin-driving forces, such as atomic force and magnetic torque, which continues to be a daunting task in computational science. Addressing this deficit, we present DeepSPIN, a versatile approach that generates high-precision predictive models of energy, atomic forces, and magnetic torque in magnetic systems. This is achieved by integrating first-principles calculations of magnetic excited states with deep learning techniques via active learning. We thoroughly explore the methodology, accuracy, and scalability of our proposed model in this paper. Our technique adeptly connects first-principles computations and atomic-scale simulations of magnetic materials. This synergy presents opportunities to utilize these calculations in devising and tackling theoretical and practical obstacles concerning magnetic materials.

cond-mat.mtrl-sci↗