Einstein Summation

I was curios about what makes Einstein summation run so fast. From what I understand it’s just a simpler way of writing the second matmul function. Thanks!