/g-mlp-pytorch

Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch

Primary LanguagePythonMIT LicenseMIT

Watchers