For more efficient einsum, see projects like https://optimized-einsum.readthedocs.io/en/stable/path_finding.html#introduction https://optimized-einsum.readthedo

For more efficient einsum, see projects like https://optimized-einsum.readthedocs.io/en/stable/path_finding.html#introduction https://optimized-einsum.readthedocs.io/en/stable/path_findi....