Tensor Decomposition Python Packages

tensorly

TensorLy: Tensor Learning in Python.

tntorch

Tensor Network Learning with PyTorch

tednet

tednet: a framework of tensor decomposition network.

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.

musco-pytorch

MUSCO: MUlti-Stage COmpression of neural networks

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.

hottbox

HOTTBOX: Higher Order Tensors ToolBOX.

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.

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

musco-tf

MUSCO: Multi-Stage COmpression of neural networks