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 physics utility which can analyze any given metric solution to the Einstein field equations.

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

Fast(er?) Einsum