idiap / fast-transformers
Pytorch library for fast transformer implementations
☆1,687Updated 2 years ago
Alternatives and similar repositories for fast-transformers:
Users that are interested in fast-transformers are comparing it to the libraries listed below
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,116Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆751Updated 10 months ago
- Reformer, the efficient Transformer, in Pytorch☆2,155Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆747Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,132Updated last year
- Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"☆1,560Updated 4 years ago
- An All-MLP solution for Vision, from Google AI☆1,015Updated 6 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,092Updated last year
- My take on a practical implementation of Linformer for Pytorch.☆413Updated 2 years ago
- Profiling and inspecting memory in pytorch☆1,049Updated 7 months ago
- Longformer: The Long-Document Transformer☆2,092Updated 2 years ago
- DeLighT: Very Deep and Light-Weight Transformers☆468Updated 4 years ago
- PyTorch Extension Library of Optimized Scatter Operations☆1,622Updated 2 months ago
- PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations☆1,048Updated 2 months ago
- Structured state space sequence models☆2,581Updated 8 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆651Updated 3 months ago
- An implementation of local windowed attention for language modeling☆428Updated 2 months ago
- [ICLR 2020] Lite Transformer with Long-Short Range Attention☆606Updated 8 months ago
- PyTorch extensions for high performance and large scale training.☆3,278Updated 2 months ago
- PyTorch implementation of Contrastive Learning methods☆1,966Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆779Updated 8 months ago
- [NeurIPS'19] Deep Equilibrium Models☆741Updated 2 years ago
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,798Updated 8 months ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,094Updated 3 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,610Updated 2 years ago
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- ☆376Updated last year
- Transformers for Longer Sequences☆596Updated 2 years ago
- list of efficient attention modules☆996Updated 3 years ago