YASTN: Yet another symmetric tensor networks; A Python library for abelian symmetric tensor network calculations (2405.12196v1)
Abstract: We present an open-source tensor network Python library for quantum many-body simulations. At its core is an abelian-symmetric tensor, implemented as a sparse block structure managed by logical layer on top of dense multi-dimensional array backend. This serves as the basis for higher-level tensor networks algorithms, operating on matrix product states and projected entangled pair states, implemented here. Using appropriate backend, such as PyTorch, gives direct access to automatic differentiation (AD) for cost-function gradient calculations and execution on GPUs or other supported accelerators. We show the library performance in simulations with infinite projected entangled-pair states, such as finding the ground states with AD, or simulating thermal states of the Hubbard model via imaginary time evolution. We quantify sources of performance gains in those challenging examples allowed by utilizing symmetries.
- R. Orús, A practical introduction to tensor networks: Matrix product states and projected entangled pair states, Ann. Phys. 349, 117 (2014).
- R. Orús, Tensor networks for complex quantum systems, Nat. Rev. Phys. 1, 538 (2019).
- S. R. White, Density matrix formulation for quantum renormalization groups, Phys. Rev. Lett. 69, 2863 (1992).
- S. R. White, Density-matrix algorithms for quantum renormalization groups, Phys. Rev. B 48, 10345 (1993).
- S. Rommer and S. Östlund, Class of ansatz wave functions for one-dimensional spin systems and their relation to the density matrix renormalization group, Phys. Rev. B 55, 2164 (1997).
- G. Vidal, Efficient simulation of one-dimensional quantum many-body systems, Phys. Rev. Lett. 93, 040502 (2004).
- U. Schollwöck, The density-matrix renormalization group in the age of matrix product states, Ann. Phys. 326, 96 (2011).
- F. Verstraete and J. I. Cirac, Renormalization algorithms for quantum-many body systems in two and higher dimensions, arXiv:cond-mat/0407066 (2004).
- R. Orús, Exploring corner transfer matrices and corner tensors for the classical simulation of quantum lattice systems, Phys. Rev. B 85, 205117 (2012).
- P. C. G. Vlaar and P. Corboz, Simulation of three-dimensional quantum systems with projected entangled-pair states, Phys. Rev. B 103, 205137 (2021).
- G. Vidal, Classical simulation of infinite-size quantum lattice systems in one spatial dimension, Phys. Rev. Lett. 98, 070201 (2007).
- I. P. McCulloch, From density-matrix renormalization group to matrix product states, J. Stat. Mech. Theory Exp. 2007, P10014 (2007).
- S. Singh, R. N. C. Pfeifer, and G. Vidal, Tensor network decompositions in the presence of a global symmetry, Phys. Rev. A 82, 050301 (2010).
- S. Singh, R. N. C. Pfeifer, and G. Vidal, Tensor network states and algorithms in the presence of a global U(1) symmetry, Phys. Rev. B 83, 115125 (2011).
- S. Singh and G. Vidal, Tensor network states and algorithms in the presence of a global SU(2) symmetry, Phys. Rev. B 86, 195114 (2012).
- P. Corboz, Variational optimization with infinite projected entangled-pair states, Phys. Rev. B 94, 035133 (2016).
- M. Fishman, S. R. White, and E. M. Stoudenmire, The ITensor Software Library for Tensor Network Calculations, SciPost Phys. Codebases , 4 (2022).
- J. Hauschild and F. Pollmann, Efficient numerical simulations with Tensor Networks: Tensor Network Python (TeNPy), SciPost Phys. Lect. Notes , 5 (2018).
- C. R. Roberts et al., Tensornetwork: A library for easy and efficient manipulation of tensor networks, https://github.com/google/TensorNetwork.
- J. Haegeman and contributors, TensorKit.jl (2024).
- A. Weichselbaum, Qspace - an open-source tensor library for Abelian and non-Abelian symmetries, arXiv:2405.06632 (2024).
- J. Hasik and contributors, peps-torch: Solving two-dimensional spin models with tensor networks (powered by pytorch), https://github.com/jurajHasik/peps-torch.
- B. Ponsioen, ad-peps: ipeps ground- and excited-state implementation based on automatic differentiation, https://github.com/b1592/ad-peps.
- J. Dziarmaga, Time evolution of an infinite projected entangled pair state: Neighborhood tensor update, Phys. Rev. B 104, 094411 (2021).
- J. Dziarmaga, Simulation of many-body localization and time crystals in two dimensions with the neighborhood tensor update, Phys. Rev. B 105, 054203 (2022).
- P. Czarnik, J. Dziarmaga, and P. Corboz, Time evolution of an infinite projected entangled pair state: An efficient algorithm, Phys. Rev. B 99, 035115 (2019).
- A. Sinha, M. M. Rams, and J. Dziarmaga, Efficient representation of minimally entangled typical thermal states in two dimensions via projected entangled pair states, Phys. Rev. B 109, 045136 (2024).
- T. Nishino and K. Okunishi, Corner transfer matrix renormalization group method, J. Phys. Soc. Japan 65, 891 (1996).
- R. Orús and G. Vidal, Simulation of two-dimensional quantum systems on an infinite lattice revisited: Corner transfer matrix for tensor contraction, Phys. Rev. B 80, 094403 (2009).
- P. Corboz, T. M. Rice, and M. Troyer, Competing states in the t𝑡titalic_t-j𝑗jitalic_j model: Uniform d𝑑ditalic_d-wave state versus stripe state, Phys. Rev. Lett. 113, 046402 (2014).
- P. Corboz and G. Vidal, Fermionic multiscale entanglement renormalization ansatz, Phys. Rev. B 80, 165129 (2009).
- M. van Damme and contributors, PEPSKit.jl: Tools for working with projected entangled-pair states, https://github.com/quantumghent/PEPSKit.jl.
Sponsor
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.