> Personally, I think np.einsum is one of the rare parts of NumPy that’s actually good.
einsum only being able to do multiplication makes it quite limited. If we leaned into the Einstein notation (e.g. [1]), we could make something that is both quite nice and quite performant.