santient / sparse-transformerLinks
Sparse Transformer with limited attention span in PyTorch
☆14Updated 4 years ago
Alternatives and similar repositories for sparse-transformer
Users that are interested in sparse-transformer are comparing it to the libraries listed below
Sorting:
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆196Updated 3 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆71Updated 5 years ago
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆75Updated 5 years ago
- Implementation of Mogrifier LSTM in PyTorch☆34Updated 5 years ago
- Sparse Attention with Linear Units☆19Updated 4 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆228Updated 3 years ago
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆371Updated 2 years ago
- PyTorch implementation of Pay Attention to MLPs☆40Updated 4 years ago
- Recent Advances in MLP-based Models (MLP is all you need!)☆117Updated 3 years ago
- Implementation of Linformer for Pytorch☆303Updated last year
- An implementation of the efficient attention module.☆327Updated 5 years ago
- custom pytorch implementation of MoCo v3☆46Updated 4 years ago
- Official Pytorch Implementation for "Continual Transformers: Redundancy-Free Attention for Online Inference" [ICLR 2023]☆28Updated 2 years ago
- Unofficial PyTorch implementation of the paper "cosFormer: Rethinking Softmax In Attention".☆44Updated 4 years ago
- Multi-head attention in PyTorch☆156Updated 6 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆169Updated 4 years ago
- code for Explicit Sparse Transformer☆61Updated 2 years ago
- Fully featured implementation of Routing Transformer☆299Updated 4 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Updated 4 years ago
- categorical variational autoencoder using the Gumbel-Softmax estimator☆27Updated 6 years ago
- A pytorch realization of adafactor (https://arxiv.org/pdf/1804.04235.pdf )☆25Updated 6 years ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆126Updated 5 years ago
- [ICML 2020] code for "PowerNorm: Rethinking Batch Normalization in Transformers" https://arxiv.org/abs/2003.07845☆120Updated 4 years ago
- My take on a practical implementation of Linformer for Pytorch.☆422Updated 3 years ago
- ☆46Updated 4 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆70Updated 3 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆201Updated 3 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆111Updated 4 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆260Updated 4 years ago
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Updated 4 years ago