Tensor Contraction Python Packages

cotengra

Hyper optimized contraction trees for large tensor networks and einsums

tensorly-torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

spacetimeengine

A Python utility for analyzing a given solution to the Einstein's field equations. Built on Sympy.

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.

feinsum

Database approach for generating near-roofline Einstein Summation kernels.