Tensor Decomposition Python Packages

tensorly

TensorLy: Tensor Learning in Python.

tntorch

Tensor Network Learning with PyTorch

teneva

Compact implementation of basic operations in the tensor-train (TT) format, including TT-SVD, TT-ALS, TT-ANOVA, TT-CROSS, TT-truncate for approximation of multidimensional arrays and multivariate functions

tednet

TedNet: A Pytorch Toolkit for Tensor Decomposition Networks

musco-pytorch

MUSCO: MUlti-Stage COmpression of neural networks

protes

Method PROTES (PRobabilistic Optimizer with TEnsor Sampling) for derivative-free optimization of 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 solution of the multidimensional Fokker-Planck equation

scoit

SCOIT is an implementation of a probabilistic tensor decomposition framework for single-cell multi-omics data integration.

trntensor

Tensor contractions for AWS Trainium via NKI (cuTENSOR-equivalent) — einsum with contraction planning, CP/PARAFAC and Tucker decompositions, density-fitted post-Hartree-Fock patterns.

hottbox

HOTTBOX: Higher Order Tensors ToolBOX.

tensorclus

TensorClus, Tensor co-clustering, text mining, clustering, multiple graphs

mprod-package

Software implementation for tensor-tensor m-product

musco-tf

MUSCO: Multi-Stage COmpression of neural networks