⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
-
Updated
May 20, 2024 - 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.
Einsum optimization using opt_einsum and PyTorch FX graph rewriting
Stats, linear algebra and einops for xarray
An implementation of EinsumNetworks in PyTorch.
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
Julia rewrite of opt_einsum Python package
MyPy Type Checking for NumPy/Jax/PyTorch Einsum Operations
A generic implementation of tensor einsum in Fortran.
Personal fork of the official EinsumNetworks implementation with a few enhancements.
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."