Recently I’ve open-sourced einops — a new (and better) way to write deep learning code.

Einops introduces a new notation and new operations.

It perfectly complements existing frameworks (pytorch, tensorflow, gluon, chainer, numpy and others) allowing you to write better deep learning code (see examples for pytorch).

Einops at Github

Tutorials: part 1 and part 2.