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.
Expected outcomes
A version of HEOM which runs on a GPU.
Performance comparison with the CPU version.
Implement dynamic scaling.
Skills
Git, python and familiarity with the Python scientific computing stack
CUDA and OpenCL knowledge
Difficulty
Hard
Mentors
Neill Lambert (nwlambert@gmail.com)
Alex Pitchford (alex.pitchford@gmail.com)
Shahnawaz Ahmed (shahnawaz.ahmed95@gmail.com)
Simon Cross (hodgestar@gmail.com)