Search papers, labs, and topics across Lattice.
This paper introduces a novel Jacobi-type tensor algorithm for tensor decomposition, implemented through a CUDA-based approach that enables the handling of high-dimensional tensors on a single GPU. The authors validate their method on NVIDIA H100 GPUs, demonstrating correct convergence for diagonalizable tensors up to nine dimensions and predictable runtime scaling with increasing tensor order. Remarkably, their implementation achieves performance improvements exceeding two orders of magnitude compared to the original MATLAB reference implementation, significantly enhancing computational efficiency in quantum system simulations.
Achieving over 100 times faster tensor decomposition on a single GPU could revolutionize simulations of complex quantum systems.
Quantum systems grow in complexity so rapidly that even modest models become difficult to simulate, creating a strong need for methods that can handle high-dimensional data, also known as tensors. In this work, we investigate a novel Jacobi-type tensor algorithm for tensor decomposition and develop a CUDA-based algorithm that supports tensors of arbitrary order on a single GPU. We test the implementation on NVIDIA H100 GPUs and show that the algorithm converges correctly for diagonalizable tensors up to nine dimensions, with runtime scaling in a predictable way as tensor order grows. Finally, our general algorithm outperforms the original MATLAB reference by more than two orders of magnitude.
