29–31 Oct 2024
Padova
Europe/Rome timezone

Boost Quantum TEA performance via flexible choices for numerical libraries

31 Oct 2024, 11:40
20m
Sala Elettra (Palazzo della Salute)

Sala Elettra

Palazzo della Salute

Via San Francesco, 90 - Padova

Speaker

Daniel Jaschke (University of Ulm)

Description

We benchmark Quantum TEA, a simulation framework developed as well with the support of the INFN quantum initiative and INFN infrastructure. Quantum TEA supports both digital, analog, and quantum-inspired quantum simulation on classical hardware. The simulations of many-body quantum systems run on heterogeneous hardware platforms using CPUs, GPUs, and TPUs. We compare different linear algebra backends, e.g., numpy versus the torch, jax, or tensorflow library, as well as a mixed-precision-inspired approach and optimizations for the target hardware. Quantum red TEA out of the Quantum TEA library specifically addresses handling tensors with different libraries or hardware, where the tensors are the building block of tensor network algorithms. The benchmark problem is a variational search of a ground state in an interacting model. This is a ubiquitous problem in quantum many-body physics, which we solve using tensor network methods. This approximate state-of-the-art method compresses quantum correlations which is key to overcoming the exponential growth of the Hilbert space as a function of the number of particles. We present a way to obtain speedups of a factor of 34 when tuning parameters on the CPU, and an additional factor of 2.76 on top of the best CPU setup when migrating to GPUs.

Sessione Simulazione

Primary author

Daniel Jaschke (University of Ulm)

Co-authors

Marco Ballarin (Università degli Studi di Padova) Nora Reinić (Università degli Studi di Padova) Luka Pavešić (Università degli Studi di Padova) Simone Montangero (Università degli Studi di Padova)

Presentation materials