⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
Mar 19, 2026 - Python
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Hyper optimized contraction trees for large tensor networks and einsums
One More Einsum for Julia! With runtime order-specification and high-level adjoints for AD
Provides compile-time contraction pattern analysis to determine optimal tensor operation to perform.
An implementation of EinsumNetworks in PyTorch.
Stats, linear algebra and einops for xarray
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
Einsum Expressions in Julia
Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
A collection of state-of-the-art contraction ordering algorithms. https://arxiv.org/abs/2209.12332
Rust accelerated contraction ordering primitives for tensor networks and einsums
TBLIS FFI bindings, wrapper and build-from-source
Convolutions and more as einsum for PyTorch
A R package for efficient computation of U-statistics.
Add a description, image, and links to the einsum topic page so that developers can more easily learn about it.
To associate your repository with the einsum topic, visit your repo's landing page and select "manage topics."