中文

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}
}