Jet: Fast quantum circuit simulations with parallel task-based tensor-network contraction

TL;DR

Jet is an open-source software library that significantly accelerates classical tensor-network simulations of quantum circuits through parallel task-based contraction, shared work reuse, and concurrent hardware utilization.

Contribution

The paper introduces Jet, a novel tensor-network simulation library employing task-based parallelism and shared work reuse to enhance quantum circuit simulation speed.

Findings

Jet achieves faster simulation times compared to existing tools.

Benchmarking on Sycamore-53 and GBS circuits demonstrates superior performance.

Theoretical performance estimates align with empirical results.

Abstract

We introduce a new open-source software library Jet, which uses task-based parallelism to obtain speed-ups in classical tensor-network simulations of quantum circuits. These speed-ups result from i) the increased parallelism introduced by mapping the tensor-network simulation to a task-based framework, ii) a novel method of reusing shared work between tensor-network contraction tasks, and iii) the concurrent contraction of tensor networks on all available hardware. We demonstrate the advantages of our method by benchmarking our code on several Sycamore-53 and Gaussian boson sampling (GBS) supremacy circuits against other simulators. We also provide and compare theoretical performance estimates for tensor-network simulations of Sycamore-53 and GBS supremacy circuits for the first time.