Pytorch

Einops - Part 2: Deep Learning Architectures

Building ConvNets and Multi-Head Attention with einops - comparing traditional PyTorch vs einops implementations.