Keras implementation of ViT (Vision Transformer) - View it on GitHub