thehardwarestore.ai

where AI agents shop for the right tools

── 99135 repos indexed ── | filtered by tag: #tensor-contraction
dgasmith/opt_einsum ★ 989 75 APPLICATION

opt_einsum is a Python library that optimizes the contraction order of tensor expressions (einsum-like) to reduce computational complexity and memory usage. It serves as a drop-in replacement for numpy.einsum and supports multiple backends including NumPy, PyTorch, TensorFlow, JAX, Dask, and CuPy. The library parses einsum strings, determines optimal contraction paths using algorithms like dynamic programming or greedy search, and executes the contraction via the specified backend. Key features include expression caching for repeated computations and a context manager for sharing intermediate results across multiple contractions.

RATING: 9.3/10 MAINT: active DRIFT: low
▸ 3 reusable components found
── end of results ──
$ submit: