Implementation of Tensor Network Simulation TN-Sim under NWQ-Sim

Abstract

Large-scale tensor network simulations are crucial for developing robust complexity-theoretic bounds on classical quantum simulation, enabling circuit cutting approaches, and optimizing circuit compilation, all of which aid efficient quantum computation on limited quantum resources. Modern exascale high-performance computing platforms offer significant potential for advancing tensor network quantum circuit simulation capabilities. We implement TN-Sim, a tensor network simulator backend within the NWQ-Sim software package that utilizes the Tensor Algebra for Many-body Methods (TAMM) framework to support both distributed HPC-scale computations and local simulations with ITensor. To optimize the scale up in computation across multiple nodes we implement a task based parallelization scheme to demonstrate parallelized gate contraction for wide quantum circuits with many gates per layer. Through the integration of the TAMM framework with Matrix Product State (MPS) tensor network approaches, we deliver a simulation environment that can scale from local systems to HPC clusters. We demonstrate an MPS tensor network simulator running on the state-of-the-art Perlmutter (NVIDIA) supercomputer and discuss the potential portability of this software to HPC clusters such as Frontier (AMD) and Aurora (Intel). We also discuss future improvements including support for different tensor network topologies and enhanced computational efficiency.

References in corpus (16)

• Quantum Computing in the NISQ era and beyond
• The density-matrix renormalization group in the age of matrix product states
• Area laws for the entanglement entropy - a review
• The ITensor Software Library for Tensor Network Calculations
• Time-evolving a matrix product state with long-ranged interactions
• Simulating Large Quantum Circuits on a Small Quantum Computer
• Hyper-optimized tensor network contraction
• Perfect Sampling with Unitary Tensor Networks
• Fast quantum circuit cutting with randomized measurements
• Efficient tensor network simulation of IBM's largest quantum processors
• Boundaries of quantum supremacy via random circuit sampling
• Investigating the effect of circuit cutting in QAOA for the MaxCut problem on NISQ devices
• Experimental Simulation of Larger Quantum Circuits with Fewer Superconducting Qubits
• Hyper-optimized approximate contraction of tensor networks with arbitrary geometry
• TAMM: Tensor Algebra for Many-body Methods
• Approximate Quantum Compiling for Quantum Simulation: A Tensor Network based approach