Einops

"Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, tensorflow, jax, and others."

I'm interested in how this supports multiple backends, but not sure I do enough linear algebra anymore for the claimed readability of Einstein notation to sell itself to me.