July 9, 2025
What is Einsum anyway?
Einstein summation lets us write an algebric expression in index-notation and have numpy/pytorch/jax generate the most effiecient set of tensor operations to compute it.
The only cons is that it can be confusing and comes with a learning curve. Einsum is backend agnostic tho, i.
read more
read more