github.com/arogozhnikov/einops (go): Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
Keywords: chainer|cupy|deep-learning|gluon|jax|keras|numpy|pytorch|tensor|tensorflow