Let's train vision transformers (ViT) for cifar 10! - View it on GitHub