Codebase release 0.0 for trainsum
We present trainsum, a Python package for computations with multidimensional quantics tensor trains. The package enables efficient approximation of tensors and functions by tensor trains, regardless of their shape or dimensionality, and supports analytic constructions of important objects, such as polynomials and Fourier transforms. Once represented in tensor train format, trainsum provides standard arithmetic operations, including addition, multiplication, and element-wise transformations. An Einstein-summation interface, closely aligned with NumPy’s einsum, enables concise specification of complex tensor train contractions and linear solvers. With support for multiple CPU and GPU backends (NumPy, CuPy, PyTorch), trainsum serves as a general-purpose framework for constructing and manipulating high-dimensional tensor representations in applications such as numerical simulation, data compression, machine learning, and data analysis.
Authors
- Matthias Heller (ORCID: https://orcid.org/0000-0002-4774-5072)
- Paul Haubenwallner
Institutions
- Fraunhofer Institute for Computer Graphics Research (DE)
- Technische Universität Darmstadt (DE)
Publication Details
- Journal
- SciPost Physics Codebases
- Published
- 2026-09-14
- DOI
- https://doi.org/10.21468/scipostphyscodeb.82-r0.0
- Primary Topic
- Parallel Computing and Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00