Tensor Decomposition Python Packages

tensorly

TensorLy: Tensor Learning in Python.

tntorch

Tensor Network Learning with PyTorch

teneva

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

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

tednet

TedNet: A Pytorch Toolkit for Tensor Decomposition Networks

hottbox

HOTTBOX: Higher Order Tensors ToolBOX.

scoit

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

tensorclus

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

mprod-package

Software implementation for tensor-tensor m-product

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.

musco-pytorch

MUSCO: MUlti-Stage COmpression of neural networks

musco-tf

MUSCO: Multi-Stage COmpression of neural networks