trainsum -- A Python package for quantics tensor trains
Mathematical Software
2026-02-25 v1 Computational Physics
Quantum Physics
Abstract
We present trainsum, a versatile Python package for doing computations with multidimensional quantics tensor trains: https://github.com/fh-igd-iet/trainsum. Using the Array API standard together with opt_einsum, trainsum allows the effortless approximation of tensors or functions by tensor trains independent of their shape or dimensionality. Once approximated, our package can perform normal arithmetic operations with quantics tensor trains, including addition, Einstein summations and element-wise transformations. It can be therefore used for generic computations with applications in simulation, data compression, machine learning and data analysis.
Keywords
Cite
@article{arxiv.2602.20226,
title = {trainsum -- A Python package for quantics tensor trains},
author = {Paul Haubenwallner and Matthias Heller},
journal= {arXiv preprint arXiv:2602.20226},
year = {2026}
}