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

⅀

One More Einsum for Julia! With runtime order-specification and high-level adjoints for AD

Hyper optimized contraction trees for large tensor networks and einsums

It slices, it dices, it splices!

Stats, linear algebra and einops for xarray

Generic PyTorch implementation of einsum that supports different semirings

Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.

A generic implementation of tensor einsum in Fortran.

An implementation of EinsumNetworks in PyTorch.

Einsum optimization using opt_einsum and PyTorch FX graph rewriting

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

Convolutions and more as einsum for PyTorch

einsum via batch matrix multiply

Rust accelerated contraction ordering primitives for tensor networks and einsums

Einsum Expressions in Julia

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