
self-attention-cv
by The-AI-Summer (organization)
Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository. (description from the project’s GitHub page)
Repository facts
- Stars
- 1,214
- Forks
- 152
- Open issues and pull requests
- 1
- Primary language
- Python
- License
- MIT License (MIT)
- Owner
- The-AI-Summer · Organization
- Created
- Jan 31, 2021
- Last push
- Sep 14, 2021
Topics
Related projects
Projects that share this one’s GitHub topics.
Rank 9,379: WenjieDu/SAITS
The official PyTorch implementation of the paper "SAITS: Self-Attention-based Imputation for Time Series". A fast and state-of-the-art (SOTA) deep-learning neural network model fo…
513 starsPythonMITLast push Aug 25, 2026
Rank 5,672: gordicaleksa/pytorch-original-transformer
My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing otherwise seemingly hard concepts. Currentl…
1,117 starsJupyter NotebookMITLast push Dec 27, 2020
Rank 5,476: lucidrains/performer-pytorch
An implementation of Performer, a linear attention-based transformer, in Pytorch
1,184 starsPythonMITLast push Feb 2, 2022
Rank 2,884: gordicaleksa/pytorch-GAT
My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention m…
2,688 starsJupyter NotebookMITLast push Nov 17, 2022
Rank 6,742: jeya-maria-jose/Medical-Transformer
Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021
860 starsPythonMITLast push Feb 23, 2023
Rank 551: BlinkDL/RWKV-LM
RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combin…
14,748 starsPythonApache-2.0Last push Sep 28, 2026