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

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.

spacetimeengine

A python physics utility which can analyze any given metric solution to the Einstein field equations.

feinsum

Database approach for generating near-roofline Einstein Summation kernels.