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.


  • Git, python and familiarity with the Python scientific computing stack

  • CUDA and OpenCL knowledge