Home Projects vit-pytorch
vit-pytorch
Python

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

by lucidrains · GitHub
Top 5% most starred in the catalogue
Stars
Forks
License
Created
Last commit
Language
artificial-intelligenceattention-mechanismcomputer-visionMITPython
View on GitHub
In plain words

Build and implement advanced image classification models using a simple framework in Python.

From the README

Install
$ pip install vit-pytorch

Excerpts from the project README on GitHub. Copyright and licensing remain with the respective authors.

You maintain this project?

Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.

Claim this page →
vit-pytorch — GitHub preview card
📈 Star history
25.42k25.36k
2026-06-272026-08-31
📈 Track vit-pytorch

Get an email alert on its next release or when it starts trending — never miss the moment.

Free · no card · unsubscribe anytime
Get email alerts →
📄 About

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

vit-pytorch has 25.4k stars on GitHub. It has been forked 3.5k times. vit-pytorch is written mainly in Python. It has been in active development since 2020. vit-pytorch is available under the MIT license. Its main topics are artificial-intelligence, attention-mechanism, computer-vision, image-classification.

Frequently asked questions

What is vit-pytorch?

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Is vit-pytorch open source?

vit-pytorch is an open-source project. It is released under the MIT license.

Is vit-pytorch free?

Yes. vit-pytorch is free and open source — you can use, modify and self-host it.

What license does vit-pytorch use?

vit-pytorch is available under the MIT license.

What language is vit-pytorch written in?

vit-pytorch is written mainly in Python.

🏅 Maintainer of this project?
olud.ai badge — vit-pytorch

Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.

[![olud.ai](https://olud.ai/badge.php?tool=lucidrains-vit-pytorch)](https://olud.ai/project/lucidrains-vit-pytorch.html)
More badge options →
🧬 Shares DNA with🧬 View the DNA map →
bottleneck-transformer-pytorch
Implementation of Bottleneck Transformer in Pytorch
678 · artificial-intelligence
sharesimage-classificationattention-mechanism
transformer-in-transformer
Implementation of Transformer in Transformer, pixel level attention paired with patch level att…
306 · artificial-intelligence
sharesimage-classification
MultiModalMamba
A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-…
473 · ai
sharesattention-mechanism
computer-vision
Programming Assignments and Lectures for Stanford's CS 231: Convolutional Neural Networks for V…
508 · computer-vision
sharesimage-classification
sign-language-translator
Python library & framework to build custom translators for the hearing-impaired and translate b…
359 · artificial-intelligence
sharesattention-mechanism
AlphaTree-graphic-deep-neural-network
AI Roadmap:机器学习(Machine Learning)、深度学习(Deep Learning)、对抗神经网络(GAN),图神经网络(GNN),NLP,大数据相关的发展路书(roa…
3k · deep-learning
sharesimage-classification
Star-Attention
Efficient LLM Inference over Long Sequences
392 · attention-mechanism
sharesattention-mechanism
catalyst
Accelerated deep learning R&D
3.4k · computer-vision
sharesimage-classification
Awesome-Backbones
Integrate deep learning models for image classification | Backbone learning/comparison/magic mo…
2k · cnn
sharesimage-classification
ultralytics
Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmenta…
59.6k · computer-vision
sharesimage-classification

Measured from GitHub topics shared by both projects, weighted by how rare each topic is.