arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.04422quant-ph
收藏

在NWQ-Sim中实现张量网络模拟TN-Sim

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

University of Washington(华盛顿大学);PNNL University of Washington(太平洋西北国家实验室华盛顿大学)

Aaron C. Hoyt, Jonathan S. Bersson, Sean Garner, Chenxu Liu, Ang Li

AI总结:

在NWQ-Sim中实现TN-Sim,利用TAMM框架支持分布式HPC计算和本地模拟,展示并行化门收缩以提升大规模量子电路模拟效率。

摘要:

大规模的张量网络模拟对于开发针对经典量子模拟的鲁棒复杂度理论界界,使电路切割方法和优化电路编译成为可能,这些都有助于在有限的量子资源上实现高效的量子计算。现代exascale高性能计算平台为推进张量网络量子电路模拟能力提供了显著潜力。我们实现了TN-Sim,这是NWQ-Sim软件包中的一个张量网络模拟器后端,利用张量代数用于多体方法(TAMM)框架来支持分布式HPC规模计算和本地模拟。为了优化跨多个节点的计算规模,我们实现了基于任务的并行化方案,以展示用于具有许多门每层的宽量子电路的并行化门收缩。通过将TAMM框架与矩阵乘积态(MPS)张量网络方法相结合,我们提供了一个能够从本地系统扩展到HPC集群的模拟环境。我们演示了一个MPS张量网络模拟器在最先进的Perlmutter(NVIDIA)超级计算机上运行,并讨论了该软件在Frontier(AMD)和Aurora(Intel)等HPC集群上的潜在可移植性。我们还讨论了未来改进,包括对不同张量网络拓扑结构的支持和计算效率的增强。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑