Einsum Python Packages

xarray-einstats

Stats, linear algebra and einops for xarray

opt-einsum-fx

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

cotengra

Hyper optimized contraction trees for large tensor networks and einsums

einconv

Convolutions and more as einsum for PyTorch

tensor-network-visualization

2D and 3D tensor network visualization, designed for torch, numpy, quimb, tensorkrowch, tenpy and tensornetwork compatibility. Explained in https://arxiv.org/abs/2606.08760

torch-semiring-einsum

Generic PyTorch implementation of einsum that supports different semirings

tensor-network-editor

Interactive local editor for building tensor networks visually and generating readable Python code for tensornetwork, quimb, tensorkrowch, and einsum. Explained in https://arxiv.org/abs/2606.08760

neinsum

NumPy's Einsum, but with named subscripts.

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.

u-stats

A Python package for efficient computation of U-statistics via Einsum.

opt-einsum-torch

Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.

tl-pjrt

Tensor Logic as a StableHLO-emitting DSL for PJRT runtimes

mypy-einsum

MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations

easy-ast

This package `easy-ast` contains several utility about AST transformer.

python-einsum

Simple, readable Einstein Summation in Python.