Loading [MathJax]/extensions/Safe.js
SciPost logo

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

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).

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

Crossref Cited-by

Authors / Affiliations: mappings to Contributors and Organizations

See all Organizations.
Funders for the research work leading to this publication