Vision Transformers with PyTorch colab: https://drive.google.com/file/d/1xEplxmYfCaXjF8ru4M-dBZtjlg6-re8_/view?usp=sharing