PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" - View it on GitHub