einops

github.com ↗

- Flexible and powerful tensor operations for readable and reliable code.