improve einsum invocations