GitHub repository: protonx-engineering/vit
Primary language: Python
Description: Our implementation for paper: An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale