MLP-Mixer: An all-MLP Architecture for Vision comprehensive and step-by-step implementation using PyTorch
In this repository, a very informative and comprehensive implementation of MLP-Mixer is provided for educational purposes using PyTorch. Paper link: https://arxiv.org/abs/2105.01601
A very comprehensive youtube video is also provided, I strongly recommend you to watch to understand the neural network architecture proposed in this paper better.