Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch - View it on GitHub