# google-research/vision_transformer

Repository: https://github.com/google-research/vision_transformer
Canonical: https://ross.abutalabs.com/products/vision_transformer
Language: Jupyter Notebook
License: Apache-2.0
License Family: permissive
Last push: 2026-08-01T21:36:15+00:00

## Health v2 (maintenance only)
Score: 75/100 (v2, computed 2026-09-03T02:20:16.233290+00:00)
- activity 95, release rhythm 35, longevity 100
- inputs: {"age_days": 2142, "days_push": 32, "days_rel": null, "gap_med": null, "n_releases_24m": 0}
- flags: no_releases
- formula: round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)

## Adoption (not part of the score)
Stars 12683, forks 1476 (observed 2026-08-28T04:10:59.595986+00:00)

## What it is
Google Research's official JAX/Flax implementation of Vision Transformer (ViT) and MLP-Mixer architectures, with released pretrained checkpoints on ImageNet and ImageNet-21k. It provides code and Colab notebooks for fine-tuning the models on custom image classification tasks.

## Use cases
- fine-tune a pretrained vision transformer on my own image dataset
- run image classification inference with ViT checkpoints
- experiment with MLP-Mixer architectures in JAX
- explore thousands of ViT training checkpoints from the scaling study
- learn how vision transformers are implemented step by step
- zero-shot image classification with LiT models

## When to choose
- you want official, well-tested ViT/MLP-Mixer code with pretrained weights
- your stack is JAX/Flax and you need TPU or GPU fine-tuning
- you want to reproduce results from the ViT research papers

## When to avoid
- you need PyTorch implementations instead of JAX/Flax
- you need large-scale multi-host pretraining infrastructure (use big_vision)
- you need production image-classification serving rather than research code

## Facets
- artifact type: library
- maturity: stable
- function: machine-learning, deep-learning, image-processing
- domain: computer-vision, deep-learning, machine-learning
- platform: python, cloud
- tags: vision-transformer, jax, flax, mlp-mixer, pretrained-models, image-classification, fine-tuning, research-code, gpu

## Member repositories
- google-research/vision_transformer (main) score 75

## Provenance
- Observed fields: from GitHub, fetched 2026-08-28T04:10:59.595986+00:00.
- Health v2: computed from the inputs above; adoption is never an input.
- Inferred fields (summary, facets, guidance): AI-extracted, prompt v1, taxonomy v1, on 2026-08-29T17:13:51.914558+00:00, confidence not recorded.
  - readme: https://github.com/google-research/vision_transformer (fetched 2026-08-28T04:10:59.595986+00:00, sha 23ba5afd23a6)
- Data as of 2026-08-30T08:39:29.467469+00:00.