Implementation of Axial attention - attending to multi-dimensional data efficiently
☆396Aug 26, 2021Updated 4 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)☆461Jun 22, 2021Updated 4 years ago
- Axial Positional Embedding for Pytorch☆84Feb 25, 2025Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Jan 27, 2022Updated 4 years ago
- Official Code for ECCV2022: Learning Semantic Correspondence with Sparse Annotations☆18Aug 22, 2022Updated 3 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Dec 21, 2021Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- PyTorch Codes for Haar Graph Pooling☆11Feb 16, 2023Updated 3 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,484Mar 19, 2021Updated 5 years ago
- Locally Enhanced Self-Attention: Rethinking Self-Attention as Local and Context Terms☆11Nov 29, 2021Updated 4 years ago
- Another attempt at a long-context / efficient transformer by me☆38Apr 11, 2022Updated 3 years ago
- Implementation of Linformer for Pytorch☆305Jan 5, 2024Updated 2 years ago
- [ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"☆292Apr 25, 2022Updated 3 years ago
- An implementation of tensorflow-based TrajGRU model☆24Aug 21, 2019Updated 6 years ago
- An All-MLP solution for Vision, from Google AI☆1,056Jul 7, 2025Updated 8 months ago
- Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021☆858Feb 23, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Code and models for the paper Glance-and-Gaze Vision Transformer☆28Jun 7, 2021Updated 4 years ago
- ☆15Jun 1, 2022Updated 3 years ago
- ☆13Jan 9, 2018Updated 8 years ago
- ☆10Oct 20, 2023Updated 2 years ago
- Official implementation of PVT series☆1,889Oct 27, 2022Updated 3 years ago
- Two simple and effective designs of vision transformer, which is on par with the Swin transformer☆608Feb 14, 2023Updated 3 years ago
- Pytorch library for fast transformer implementations☆1,765Mar 23, 2023Updated 3 years ago
- MLP-Like Vision Permutator for Visual Recognition (PyTorch)☆192Mar 31, 2022Updated 3 years ago
- ☆246Jul 23, 2021Updated 4 years ago
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- A convolution-free, transformer-only version of the CycleGAN framework☆32Feb 12, 2022Updated 4 years ago
- Pytorch implementation of Axial-DeepLab: Stand-Alone Axial-Attention for Panoptic Segmentation☆12Oct 15, 2021Updated 4 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,176May 15, 2024Updated last year
- Tensorflow implementation for "Improved Transformer for High-Resolution GANs" (NeurIPS 2021).☆93Jul 30, 2024Updated last year
- Compare GAN code.☆17Apr 12, 2021Updated 4 years ago
- (ACL-IJCNLP 2021) Convolutions and Self-Attention: Re-interpreting Relative Positions in Pre-trained Language Models.☆21Jul 13, 2022Updated 3 years ago
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.☆1,215Sep 14, 2021Updated 4 years ago
- ☆110Sep 15, 2021Updated 4 years ago
- Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"☆1,611Aug 12, 2020Updated 5 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- implementing Weight Agnostic Neural Networks to Spiking Neural Networks☆10Jan 26, 2021Updated 5 years ago
- Code for ECCV 2020 paper (oral): Mining Cross-Image Semantics for Weakly Supervised Semantic Segmentation☆163Apr 23, 2021Updated 4 years ago
- ☆98Apr 27, 2022Updated 3 years ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆821Jul 14, 2022Updated 3 years ago
- SAIS: Supervising and Augmenting Intermediate Steps for Document-Level Relation Extraction☆20Oct 10, 2022Updated 3 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆292May 3, 2024Updated last year
- Taming Transformers for High-Resolution Image Synthesis☆6,455Jul 30, 2024Updated last year