einops (conda): 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