py-einops (spack): Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, tensorflow, and others. 
Keywords: chainer|cupy|deep-learning|gluon|jax|keras|numpy|pytorch|tensor|tensorflow