mirror of
https://git.FreeBSD.org/ports.git
synced 2025-01-05 06:27:37 +00:00
25b4607212
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. WWW: https://github.com/dgasmith/opt_einsum
10 lines
563 B
Plaintext
10 lines
563 B
Plaintext
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.
|
|
|
|
WWW: https://github.com/dgasmith/opt_einsum
|