lucidrains / FLASH-pytorchLinks
Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"
☆370Updated 2 years ago
Alternatives and similar repositories for FLASH-pytorch
Users that are interested in FLASH-pytorch are comparing it to the libraries listed below
Sorting:
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆195Updated 2 years ago
- An implementation of local windowed attention for language modeling☆480Updated 2 months ago
- Implementation of Linformer for Pytorch☆300Updated last year
- Implement the paper "Self-Attention with Relative Position Representations"☆139Updated 4 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆798Updated last year
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆227Updated 3 years ago
- A pytorch &keras implementation and demo of Fastformer.☆189Updated 3 years ago
- My take on a practical implementation of Linformer for Pytorch.☆420Updated 3 years ago
- Rotary Transformer☆1,032Updated 3 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- Sequence modeling with Mega.☆300Updated 2 years ago
- ☆197Updated last year
- Implementation of "Attention Is Off By One" by Evan Miller☆197Updated 2 years ago
- Root Mean Square Layer Normalization☆254Updated 2 years ago
- Code for the ALiBi method for transformer language models (ICLR 2022)☆543Updated last year
- A Pytorch implementation of Sparsely-Gated Mixture of Experts, for massively increasing the parameter count of language models☆813Updated 2 years ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆757Updated 2 months ago
- The pure and clear PyTorch Distributed Training Framework.☆274Updated last year
- An implementation of the efficient attention module.☆321Updated 4 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆166Updated 4 years ago
- ACL'2023: DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models☆324Updated last year
- Fully featured implementation of Routing Transformer☆298Updated 3 years ago
- ☆680Updated 2 months ago
- Learning Rate Warmup in PyTorch☆412Updated 3 months ago
- Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.☆241Updated 3 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆386Updated 4 years ago
- Transformer are RNNs: Fast Autoregressive Transformer with Linear Attention☆23Updated 4 years ago
- Implementation of Forward Forward Network proposed by Hinton in NIPS 2022.☆170Updated 2 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆798Updated 3 months ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆216Updated 2 years ago