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