Tensor Train Python Packages

tntorch

Tensor Network Learning with PyTorch

teneva

A framework based on the tensor train decomposition for working with multivariate functions and multidimensional arrays

semiflow-pde

Fast no_std Rust library for evolution equations and PDEs via Chernoff approximation of operator semigroups — heat, Schrödinger, diffusion, graph & manifold operators. Bindings for C, Python, and WebAssembly.

tensor-network-visualization

2D and 3D tensor network visualization, designed for torch, numpy, quimb, tensorkrowch, tenpy and tensornetwork compatibility. Explained in https://arxiv.org/abs/2606.08760

protes

Gradient-free optimization method for the multidimensional arrays and discretized multivariate functions based on the tensor train (TT) format.

ttopt

Gradient-free optimization method for multivariable functions based on the low rank tensor train (TT) format and maximal-volume principle.

fpcross

Solver in the low-rank tensor train format with cross approximation approach for the multidimensional Fokker-Planck equation

mpnum

Matrix Product Representation library for Python

t3f

Tensor Train decomposition on TensorFlow

tensorquantlib

Comprehensive quantitative finance library with tensor-train compression, autodiff, and stochastic models

tnkm

tnkm: A JAX Framework for Tensor Network Kernel Machines