_version_ 1866915270941999104
author Zeng, Jinzhe
Zhang, Duo
Peng, Anyang
Zhang, Xiangyu
He, Sensen
Wang, Yan
Liu, Xinzijian
Bi, Hangrui
Li, Yifan
Cai, Chun
Zhang, Chengqian
Du, Yiming
Zhu, Jia-Xin
Mo, Pinghui
Huang, Zhengtao
Zeng, Qiyu
Shi, Shaochen
Qin, Xuejian
Yu, Zhaoxi
Luo, Chenxing
Ding, Ye
Liu, Yun-Pei
Shi, Ruosong
Wang, Zhenyu
Bore, Sigbjørn Løland
Chang, Junhan
Deng, Zhe
Ding, Zhaohan
Han, Siyuan
Jiang, Wanrun
Ke, Guolin
Liu, Zhaoqing
Lu, Denghui
Muraoka, Koki
Oliaei, Hananeh
Singh, Anurag Kumar
Que, Haohui
Xu, Weihong
Xu, Zhangmancang
Zhuang, Yong-Bin
Dai, Jiayu
Giese, Timothy J.
Jia, Weile
Xu, Ben
York, Darrin M.
Zhang, Linfeng
Wang, Han
author_facet Zeng, Jinzhe
Zhang, Duo
Peng, Anyang
Zhang, Xiangyu
He, Sensen
Wang, Yan
Liu, Xinzijian
Bi, Hangrui
Li, Yifan
Cai, Chun
Zhang, Chengqian
Du, Yiming
Zhu, Jia-Xin
Mo, Pinghui
Huang, Zhengtao
Zeng, Qiyu
Shi, Shaochen
Qin, Xuejian
Yu, Zhaoxi
Luo, Chenxing
Ding, Ye
Liu, Yun-Pei
Shi, Ruosong
Wang, Zhenyu
Bore, Sigbjørn Løland
Chang, Junhan
Deng, Zhe
Ding, Zhaohan
Han, Siyuan
Jiang, Wanrun
Ke, Guolin
Liu, Zhaoqing
Lu, Denghui
Muraoka, Koki
Oliaei, Hananeh
Singh, Anurag Kumar
Que, Haohui
Xu, Weihong
Xu, Zhangmancang
Zhuang, Yong-Bin
Dai, Jiayu
Giese, Timothy J.
Jia, Weile
Xu, Ben
York, Darrin M.
Zhang, Linfeng
Wang, Han
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2502_19161
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials
Zeng, Jinzhe
Zhang, Duo
Peng, Anyang
Zhang, Xiangyu
He, Sensen
Wang, Yan
Liu, Xinzijian
Bi, Hangrui
Li, Yifan
Cai, Chun
Zhang, Chengqian
Du, Yiming
Zhu, Jia-Xin
Mo, Pinghui
Huang, Zhengtao
Zeng, Qiyu
Shi, Shaochen
Qin, Xuejian
Yu, Zhaoxi
Luo, Chenxing
Ding, Ye
Liu, Yun-Pei
Shi, Ruosong
Wang, Zhenyu
Bore, Sigbjørn Løland
Chang, Junhan
Deng, Zhe
Ding, Zhaohan
Han, Siyuan
Jiang, Wanrun
Ke, Guolin
Liu, Zhaoqing
Lu, Denghui
Muraoka, Koki
Oliaei, Hananeh
Singh, Anurag Kumar
Que, Haohui
Xu, Weihong
Xu, Zhangmancang
Zhuang, Yong-Bin
Dai, Jiayu
Giese, Timothy J.
Jia, Weile
Xu, Ben
York, Darrin M.
Zhang, Linfeng
Wang, Han
Chemical Physics
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.
title DeePMD-kit v3: A Multiple-Backend Framework for Machine Learning Potentials
topic Chemical Physics
url https://arxiv.org/abs/2502.19161