Implementation of Axial attention - attending to multi-dimensional data efficiently
☆396Aug 26, 2021Updated 5 years ago
Alternatives and similar repositories for axial-attention
Users that are interested in axial-attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆459Jun 22, 2021Updated 5 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆173Nov 25, 2022Updated 3 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Jan 27, 2022Updated 4 years ago
- Axial Positional Embedding for Pytorch☆82Feb 25, 2025Updated last year
- Another attempt at a long-context / efficient transformer by me☆38Apr 11, 2022Updated 4 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆85Oct 30, 2021Updated 5 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 4 years ago
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆77Jun 8, 2021Updated 5 years ago
- Implementation of Kronecker Attention in Pytorch☆20Sep 12, 2020Updated 6 years ago
- A Pytorch implementation of Attention on Attention module (both self and guided variants), for Visual Question Answering☆43Nov 8, 2020Updated 5 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Aug 3, 2021Updated 5 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,487Mar 19, 2021Updated 5 years ago
- Official Code for ECCV2022: Learning Semantic Correspondence with Sparse Annotations☆18Aug 22, 2022Updated 4 years ago
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆39Aug 3, 2021Updated 5 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆56Jan 5, 2023Updated 3 years ago
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- Implementation of Fast Transformer in Pytorch☆176Aug 26, 2021Updated 5 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Sep 27, 2021Updated 5 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆140Dec 21, 2021Updated 4 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆306Dec 27, 2021Updated 4 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,222Oct 26, 2026Updated last week
- Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning☆166Feb 12, 2024Updated 2 years ago
- Implementation of Uformer, Attention-based Unet, in Pytorch☆97Oct 26, 2021Updated 5 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Jun 2, 2024Updated 2 years ago
- Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021☆859Feb 23, 2023Updated 3 years ago
- Implementation of TransGanFormer, an all-attention GAN that combines the finding from the recent GanFormer and TransGan paper☆155Apr 27, 2021Updated 5 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 5 years ago
- A simple implementation of a deep linear Pytorch module☆21Oct 16, 2020Updated 6 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆73Jun 8, 2026Updated 4 months ago
- Two simple and effective designs of vision transformer, which is on par with the Swin transformer☆609Feb 14, 2023Updated 3 years ago
- Pytorch library for fast transformer implementations☆1,785Mar 23, 2023Updated 3 years ago
- Locally Enhanced Self-Attention: Rethinking Self-Attention as Local and Context Terms☆11Nov 29, 2021Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Official implementation of PVT series☆1,905Oct 27, 2022Updated 4 years ago
- Implementation of Linformer for Pytorch☆307Jan 5, 2024Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆145Mar 24, 2025Updated last year
- Implementation of Uniformer, a simple attention and 3d convolutional net that achieved SOTA in a number of video classification tasks, de…☆102Apr 22, 2022Updated 4 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,183Feb 2, 2022Updated 4 years ago
- Fully featured implementation of Routing Transformer☆300Nov 6, 2021Updated 4 years ago
- An All-MLP solution for Vision, from Google AI☆1,066Jul 7, 2025Updated last year