Official codebase used to develop Vision Transformer, MLP-Mixer, LiT and more. - View it on GitHub