Provide cuQuantum's cuDensityMat as a backend for QuTiP.
This backend is specifically designed for simulating large, composite quantum systems by performing tensor network contractions on the GPU.
qutip-cuquantum is not yet available on pypi. Installation must be done from source:
pip install git+https://github.com/qutip/qutip.git
pip install git+https://github.com/qutip/qutip-cuquantum.git
It require cuquantum to be installed at runtime, see nvidia;s documentation for installation instruction.
In simple case, simply calling set_as_default before a qutip script should be sufficient to use the backend common solver:
import qutip_cuquantum
from cuquantum.densitymat import WorkStream
qutip_cuquantum.set_as_default(WorkStream())
qutip-cuquantum work well to speed-up large simulation using mesolve or sesolve.
However this backend is not compatible with advanced qutip solvers (brmesolve, HEOM) and other various feature.