sparse_attention

Attention library

Provides primitives for sparse attention mechanisms used in transformer models to improve computational efficiency and scalability

Examples of using sparse attention, as in "Generating Long Sequences with Sparse Transformers"

GitHub

2k stars
43 watching
192 forks
Language: Python
last commit: about 6 years ago

Related projects:

RepositoryDescriptionStars
openai/finetune-transformer-lmThis project provides code and model for improving language understanding through generative pre-training using a transformer-based architecture.2,167
hiroyuki-kasai/sparsegdlibraryA collection of unconstrained optimization algorithms for sparse modeling in MATLAB53
microsoft/focal-transformerA vision transformer architecture that uses a novel attention mechanism to capture local-global interactions in images547
openai/lm-human-preferencesTraining methods and tools for fine-tuning language models using human preferences1,240
neuralmagic/sparsemlEnables the creation of smaller neural network models through efficient pruning and quantization techniques2,083
huyz1117/bamAn implementation of the Bottleneck Attention Module in TensorFlow using attention mechanism12
koichiro11/residual-attention-networkAn image classification neural network implementation using attention mechanisms and residual learning94
openai/generating-reviews-discovering-sentimentGenerates reviews and discovers sentiment using a language model1,512
tusimple/sparse-structure-selectionRe-implements sparse structure selection algorithm for deep neural networks in a modified MXNet framework.87
pistony/residualattentionnetworkA Gluon implementation of Residual Attention Network for image classification tasks108
openai/simple-evalsEvaluates language models using standardized benchmarks and prompting techniques.2,059
64bit/async-openaiA Rust library providing an asynchronous interface to the OpenAI API1,202
macpaw/openaiAn OpenAI API wrapper for Swift applications2,149
yaroslavyaroslav/openai-sublime-textAn AI-powered Sublime Text plugin for generating text and answering questions using OpenAI models208
openai/openai-openapiA specification for an API providing access to AI capabilities1,332