Tensor Train Python Packages

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.

tntorch

Tensor Network Learning with PyTorch

teneva

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

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.

fpcross

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

ttopt

Multivariate function optimizer based on the tensor train approach.

mpnum

Matrix Product Representation library for Python

t3f

Tensor Train decomposition on TensorFlow