Vision Transformer

timmPython

A comprehensive collection of PyTorch image models, layers, utilities, and training scripts for computer vision research and applications.

Vision TransformerJupyter Notebook

Official JAX/Flax implementation of Vision Transformer (ViT) and MLP-Mixer for image recognition, with pre-trained models.

ScenicPython

A JAX library for rapid prototyping of large-scale attention-based vision models across images, video, audio, and multimodal data.

Awesome Visual Transformer

A curated collection of research papers and resources on Vision Transformers (ViT) for computer vision tasks.