opt-einsum-torch (pypi): Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.
Keywords: einsum|gpu-computing|pytorch|scientific-computing