GMP-Featurizer:用于高效计算原子体系高斯多极特征的并行化 Python 包
化学物理
2023-03-24 v1 计算工程、金融与科学
摘要
GMP-Featurizer 是一个轻量、精确、高效且可扩展的软件包,用于计算涵盖元素周期表各类元素原子体系的高斯多极(GMP)特征 \cite{GMP}。从 AmpTorch \cite{amptorch} 的 GMP 特征计算模块发展而来,GMP-Featurizer 的能力此后得到大幅提升,包括精度与效率,以及在不同核心乃至机器上并行化的能力。此外,该 Python 包仅依赖极少数的标准 Python 库,外加用于 C++ 代码接口的 cffi 与用于并行化的 Ray \cite{Ray},使其轻量且稳健。我们设计了一套单元测试以确保输出可靠性。该包还包含丰富的示例与教程,以及两套赝势文件(用于指定 GMP 特征集),供用户使用。总体而言,本包旨在为有兴趣基于 GMP 特征开发模型的化学与材料科学家提供标准实现。其源代码依据 Apache 2.0 许可证公开发布。
引用
@article{arxiv.2303.12980,
title = {GMP-Featurizer: A parallelized Python package for efficiently computing the Gaussian Multipole features of atomic systems},
author = {Xiangyun Lei and Joseph Montoya},
journal= {arXiv preprint arXiv:2303.12980},
year = {2023}
}