Quax:电子结构计算的自动微分 源码

Quax:量子化学,由JAX提供支持 您已找到Quax。 我们刚刚开源。 当我们整理东西时,请原谅我们的灰尘。 概述此软件包的论文是最近提交的,此回购将在发布后通过链接进行更新。 电子能的任意阶核衍生物 该库支持简单而干净的API,用于获取电子结构计算的高阶能量导数,例如Hartree-Fock,二阶Moller-Plesset扰动理论(MP2)以及具有单,双和扰动三重激发的耦合簇[ CCSD(T)]。 尽管大多数代码仅支持解析梯度,有时仅支持Hessian计算,但是此代码可以计算任意阶的解析导数。 我们使用来自动区分电子结构计算。 该代码可以轻松扩展以支持其他方法,例如使用提供的指导。 API很简单。 在给定的理论水平下,分子的全导数张量(具有所有笛卡尔坐标的完整梯度,或具有完整的Hessian或具有完整的三次方和四次导数张量,等等),其计算公式如下: import quax i