opt_einsum
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
To install this package, run one of the following:
Einsum is a very powerful function for contracting tensors of arbitrary dimension and index. However, it is typically only optimized to contract two terms at a time resulting in non-optimal scaling. This package optimizes the contraction order for arbitrarily large speedups. See the docs for more information: dgasmith.github.io/opt_einsum/
Summary
Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
Last Updated
Sep 26, 2024 at 23:15
License
MIT
Total Downloads
3.8M
Supported Platforms
Unsupported Platforms
GitHub Repository
https://github.com/dgasmith/opt_einsum