Home / JAX / Flax > Vision Transformer Flax > Vision Transformer Official implementation of An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. Package 12.4k stars GitHub Back to JAX