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 基礎方案展現出顯著的加速效果。