YASTN: Yet another symmetric tensor networks; A Python library for Abelian symmetric tensor network calculations
Marek M. Rams, Gabriela Wójtowicz, Aritra Sinha, Juraj Hasik
SciPost Phys. Codebases 52 (2025) · published 26 February 2025
- doi: 10.21468/SciPostPhysCodeb.52
- live repo (external)
- Submissions/Reports
-
This Publication is part of a bundle
When citing, cite all relevant items (e.g. for a Codebase, cite both the article and the release you used).
DOI | Type | |
---|---|---|
10.21468/SciPostPhysCodeb.52 | Article | |
10.21468/SciPostPhysCodeb.52-r1.2 | Codebase release |
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 a logical layer on top of a dense multidimensional array backend. This serves as the basis for higher-level tensor network algorithms operating on matrix product states and projected entangled pair states. An appropriate backend, such as PyTorch, gives direct access to automatic differentiation (AD) for cost-function gradient calculations and execution on GPU and other supported accelerators. We show the library performance in simulations with infinite projected entangled-pair states, such as finding the ground states with AD and simulating thermal states of the Hubbard model via imaginary time evolution. For these challenging examples, we identify and quantify sources of the numerical advantage exploited by the symmetric-tensor implementation.
Supplementary Information
External links to supplemental resources; opens in a new tab.
Cited by 2

Authors / Affiliations: mappings to Contributors and Organizations
See all Organizations.- 1 Marek Rams,
- 1 2 Gabriela Wojtowicz,
- 1 3 Aritra Sinha,
- 4 5 6 Juraj Hasik
- 1 Uniwersytet Jagielloński / Jagiellonian University
- 2 Universität Ulm / University of Ulm
- 3 Max-Planck-Institut für Physik komplexer Systeme / Max Planck Institute for the Physics of Complex Systems
- 4 Institute of Physics, University of Amsterdam [IoP, UvA]
- 5 Universität Zürich / University of Zurich [UZH]
- 6 Delta Institute for Theoretical Physics [Delta ITP]