openai / sparse_attention
Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"
☆1,550Updated 4 years ago
Alternatives and similar repositories for sparse_attention:
Users that are interested in sparse_attention are comparing it to the libraries listed below
- Reformer, the efficient Transformer, in Pytorch☆2,152Updated last year
- ☆3,635Updated 2 years ago
- Pytorch library for fast transformer implementations☆1,677Updated last year
- Single Headed Attention RNN - "Stop thinking with your head"☆1,180Updated 3 years ago
- 🐥A PyTorch implementation of OpenAI's finetuned transformer language model with a script to import the weights pre-trained by OpenAI☆1,508Updated 3 years ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,115Updated 3 years ago
- Efficient GPU kernels for block-sparse matrix multiplication and convolution☆1,035Updated last year
- Transformer training code for sequential tasks☆610Updated 3 years ago
- An open source framework for seq2seq models in PyTorch.☆1,506Updated 2 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆745Updated last year
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,108Updated 2 years ago
- DeLighT: Very Deep and Light-Weight Transformers☆467Updated 4 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,542Updated 3 years ago
- Integrating the Best of TF into PyTorch, for Machine Learning, Natural Language Processing, and Text Generation. This is part of the CAS…☆745Updated 2 years ago
- Longformer: The Long-Document Transformer☆2,082Updated 2 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,096Updated 2 years ago
- Mesh TensorFlow: Model Parallelism Made Easier☆1,604Updated last year
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆738Updated 9 months ago
- Unsupervised Data Augmentation (UDA)☆2,187Updated 3 years ago
- Fully featured implementation of Routing Transformer☆288Updated 3 years ago
- PyTorch original implementation of Cross-lingual Language Model Pretraining.☆2,899Updated 2 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆371Updated 4 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,910Updated last year
- Fast Block Sparse Matrices for Pytorch☆546Updated 4 years ago
- Training RNNs as Fast as CNNs (https://arxiv.org/abs/1709.02755)☆2,102Updated 3 years ago
- Multi-Task Deep Neural Networks for Natural Language Understanding☆2,245Updated 11 months ago
- PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM☆1,259Updated 3 years ago
- Code and model for the paper "Improving Language Understanding by Generative Pre-Training"☆2,189Updated 6 years ago
- Make huge neural nets fit in memory☆2,761Updated 4 years ago
- jiant is an nlp toolkit☆1,661Updated last year