⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.

Hyper optimized contraction trees for large tensor networks and einsums

Stats, linear algebra and einops for xarray

Generic PyTorch implementation of einsum that supports different semirings

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

Convolutions and more as einsum for PyTorch

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

einsum via batch matrix multiply

A Tensor Network package for Machine Learning and Quantum Computing in Python.

Optimizing Einstein Sum Implementation in PyTorch with Specialization, Path Searching, and GPU-Native Contraction

A collection of state-of-the-art contraction ordering algorithms. https://arxiv.org/abs/2209.12332

NumPy's Einsum, but with named subscripts.

Personal fork of the official EinsumNetworks implementation with a few enhancements.

MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations