Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, - View it on GitHub