pyscf/gpu4pyscf

解决的问题

为 PySCF（一个基于 Python 的化学模拟框架）提供 GPU 加速功能。这使得计算成本高昂的量子化学计算——如哈特里-福克和密度泛函理论（DFT）——能够通过将繁重的数学运算卸载到 NVIDIA GPU 上，实现显著加速。

工作原理

作为 PySCF 的插件，允许用户通过 .to_gpu() 方法将 PySCF 对象移动到 GPU 上。它利用 CUDA 支持的硬件和 CuPy、cuTENSOR 等库，加速电子结构计算中必不可少的张量收缩及其他线性代数运算。

适用人群

需要缩短能量计算、解析梯度和 Hessian 矩阵计算时间的计算化学家和研究人员。

主要亮点

• 广泛的功能支持：通过 libXC 支持 LDA、GGA、mGGA、杂化和范围分离泛函。
• 全面的化学工具：支持解析梯度、Hessian 矩阵、几何优化和过渡态搜索。
• 高级模型：实现 PCM 和 SMD 溶剂模型，以及色散校正（DFTD3/DFTD4）。
• 高性能：在多种分子体系中，相较于 CPU 基础方案展现出显著的加速效果。