层次运动方程的GPU实现
层次运动方程(HEOM)方法是一种非微扰方法,用于模拟耗散量子系统的密度矩阵演化。基础方程是一组耦合的常微分方程(ODEs),可以在GPU上运行。这将允许研究更大的系统,如[1]中讨论的那样。本项目的目标是扩展QuTiP的HEOM方法[2]并在GPU上实现它。
由于该方法与模拟大型耦合常微分方程有关,它也可以非常通用,并扩展到其他求解器。
预期结果
一个在GPU上运行的HEOM版本。
与CPU版本的性能比较。
实现动态扩展。
技能
Git、Python 以及对 Python 科学计算栈的熟悉
CUDA和OpenCL知识
难度
困难
导师
尼尔·兰伯特 (nwlambert@gmail.com)
亚历克斯·皮奇福德 (alex.pitchford@gmail.com)
Shahnawaz Ahmed (shahnawaz.ahmed95@gmail.com)
西蒙·克罗斯 (hodgestar@gmail.com)