There are 2 repositories under einsum topic.
⚡️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
It slices, it dices, it splices!
Stats, linear algebra and einops for xarray
Generic PyTorch implementation of einsum that supports different semirings
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.
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
Julia rewrite of opt_einsum Python package
Personal fork of the official EinsumNetworks implementation with a few enhancements.
MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations