层次运动方程的GPU实现

层次运动方程(HEOM)方法是一种非微扰方法,用于模拟耗散量子系统的密度矩阵演化。基础方程是一组耦合的常微分方程(ODEs),可以在GPU上运行。这将允许研究更大的系统,如[1]中讨论的那样。本项目的目标是扩展QuTiP的HEOM方法[2]并在GPU上实现它。

由于该方法与模拟大型耦合常微分方程有关,它也可以非常通用,并扩展到其他求解器。

预期结果

  • 一个在GPU上运行的HEOM版本。

  • 与CPU版本的性能比较。

  • 实现动态扩展。

技能

  • Git、Python 以及对 Python 科学计算栈的熟悉

  • CUDA和OpenCL知识

难度

  • 困难

导师

参考文献