Tensor Network Learning with PyTorch
Tensor contractions for AWS Trainium via NKI (cuTENSOR-equivalent) — einsum with contraction planning, CP/PARAFAC and Tucker decompositions, density-fitted post-Hartree-Fock patterns.
Python Tensor Toolbox
MUSCO: Multi-Stage COmpression of Neural Networks
MUSCO: Multi-Stage COmpression of neural networks