GPU implementation of the Hierarchical Equations of Motion¶
The Hierarchical Equations of Motion (HEOM) method is a non-perturbative approach to simulate the evolution of the density matrix of dissipative quantum systems. The underlying equations are a system of coupled ODEs which can be run on a GPU. This will allow the study of larger systems as discussed in 1. The goal of this project would be to extend QuTiP’s HEOM method 2 and implement it on a GPU.
Since the method is related to simulating large, coupled ODEs, it can also be quite general and extended to other solvers.
A version of HEOM which runs on a GPU.
Performance comparison with the CPU version.
Implement dynamic scaling.
Git, python and familiarity with the Python scientific computing stack
CUDA and OpenCL knowledge