I’d like to present a new project called einops.
Einops provides flexible and powerful operations for readable and reliable code (developed right for deep learning).
Einops provides complete support for pytorch.
Project lives at github: http://github.com/arogozhnikov/einops