The-AI-Summer/self-attention-cv
A PyTorch library providing self-attention mechanism implementations for computer vision applications.

Not currently ranked — collecting fresh signals.
star history
This repository provides implementations of various self-attention building blocks for computer vision tasks in PyTorch. It includes modules such as multi-head self-attention, axial attention, and transformer encoders, implemented using einsum and einops for efficient computation. The library is designed as reusable components that can be integrated into larger vision models.
Frequently asked
- What is The-AI-Summer/self-attention-cv?
- A PyTorch library providing self-attention mechanism implementations for computer vision applications.
- Is self-attention-cv open source?
- Yes — The-AI-Summer/self-attention-cv is open source, released under the MIT license.
- What language is self-attention-cv written in?
- The-AI-Summer/self-attention-cv is primarily written in Python.
- How popular is self-attention-cv?
- The-AI-Summer/self-attention-cv has 1.2k stars on GitHub.
- Where can I find self-attention-cv?
- The-AI-Summer/self-attention-cv is on GitHub at https://github.com/The-AI-Summer/self-attention-cv.