cp-decomposition
Tensor Network Learning with PyTorch
Python Tensor Toolbox
MUSCO: MUlti-Stage COmpression of neural networks
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: Multi-Stage COmpression of neural networks