Tim Rocktäschel
https://medium.com/ibm-data-ai/einsum-an-easy-intuitive-way-to-write-tensor-operation-9e12b8a80570
Understanding einsum for Deep learning: implement a transformer with multi-head self-attention from scratch | AI Summer
