py-opt_einsum (python/py-opt_einsum) Updated: 1 year, 11 months ago Add to my watchlist
A tensor contraction order optimizer.Optimized einsum can significantly reduce the overall execution time of einsum-like expressions (e.g., np.einsum, dask.array.einsum, pytorch.einsum, tensorflow.einsum) by optimizing the expression's contraction order and dispatching many operations to canonical BLAS, cuBLAS, or other specialized routines. Optimized einsum is agnostic to the backend and can handle NumPy, Dask, PyTorch, Tensorflow, CuPy, Sparse, Theano, JAX, and Autograd arrays as well as potentially any library which conforms to a standard API.
Version: 3.3.0 License: MIT
GitHub
Statistics for selected duration
2024-Mar-29 to 2024-Apr-28
No stats available for this selection.
Try changing the range of days. Alternatively visit statistics page to have an overall look at the submitted statistics.