TensorLy-Quantum: Quantum Machine Learning with Tensor Methods

• Main Authors: Patti, Taylor L, Kossaifi, Jean, Yelin, Susanne F, Anandkumar, Anima
• Format: Journal Article
• Language: English
• Published: 19.12.2021
• Subjects: Physics - Quantum Physics
• DOI: 10.48550/arxiv.2112.10239

Simulation is essential for developing quantum hardware and algorithms. However, simulating quantum circuits on classical hardware is challenging due to the exponential scaling of quantum state space. While factorized tensors can greatly reduce this overhead, tensor network-based simulators are relatively few and often lack crucial functionalities. To address this deficiency, we created TensorLy-Quantum, a Python library for quantum circuit simulation that adopts the PyTorch API. Our library leverages the optimized tensor methods of the existing TensorLy ecosystem to represent, simulate, and manipulate large-scale quantum circuits. Through compact tensor representations and efficient operations, TensorLy-Quantum can scale to hundreds of qubits on a single GPU and thousands of qubits on multiple GPUs. TensorLy-Quantum is open-source and accessible at https://github.com/tensorly/quantum