Spinsim: a GPU optimized python package for simulating spin-half and spin-one quantum systems

TL;DR

Spinsim is a GPU-accelerated Python package that efficiently simulates spin-half and spin-one quantum systems with high speed and accuracy, useful for various quantum physics and sensing applications.

Contribution

It introduces a GPU-optimized Python package using numba.cuda, achieving 3 to 4 orders of magnitude speedup over standard tools for simulating quantum spin systems.

Findings

Achieves 3-4 orders of magnitude speed improvement.

Maintains high accuracy comparable to industry standards.

Applicable to diverse quantum systems and experiments.

Abstract

The Spinsim python package simulates spin-half and spin-one quantum mechanical systems following a time dependent Shroedinger equation. It makes use of numba.cuda, which is an LLVM (Low Level Virtual Machine) compiler for Nvidia Cuda compatible systems using GPU parallelization. Along with other optimizations, this allows for speed improvements from 3 to 4 orders of magnitude while staying just as accurate, compared to industry standard packages. It is available for installation on PyPI, and the source code is available on github. The initial use-case for the Spinsim will be to simulate quantum sensing-based ultracold atom experiments for the Monash University School of Physics \& Astronomy spinor Bose-Einstein condensate (spinor BEC) lab, but we anticipate it will be useful in simulating any range of spin-half or spin-one quantum systems with time dependent Hamiltonians that cannot be solved analytically. These appear in the fields of nuclear magnetic resonance (NMR), nuclear quadrupole resonance (NQR) and magnetic resonance imaging (MRI) experiments and quantum sensing, and with the spin-one systems of nitrogen vacancy centres (NVCs), ultracold atoms, and BECs.