deep-spin / entmax
The entmax mapping and its loss, a family of sparse softmax alternatives.
☆434Updated 10 months ago
Alternatives and similar repositories for entmax:
Users that are interested in entmax are comparing it to the libraries listed below
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆262Updated 3 years ago
- Implementation of Sparsemax activation in Pytorch☆159Updated 4 years ago
- Fully featured implementation of Routing Transformer☆291Updated 3 years ago
- Transformer with Untied Positional Encoding (TUPE). Code of paper "Rethinking Positional Encoding in Language Pre-training". Improve exis…☆251Updated 3 years ago
- Understanding the Difficulty of Training Transformers☆329Updated 2 years ago
- ☆376Updated last year
- Fast, general, and tested differentiable structured prediction in PyTorch☆1,110Updated 3 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆374Updated 4 years ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆318Updated 5 years ago
- ☆218Updated 4 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆751Updated last year
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,126Updated 3 years ago
- My take on a practical implementation of Linformer for Pytorch.☆414Updated 2 years ago
- Fast Differentiable Sorting and Ranking☆603Updated last year
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆151Updated last year
- Implementation of Universal Transformer in Pytorch☆259Updated 6 years ago
- Official PyTorch Repo for "ReZero is All You Need: Fast Convergence at Large Depth"☆408Updated 9 months ago
- Implementation of gMLP, an all-MLP replacement for Transformers, in Pytorch☆425Updated 3 years ago
- Implementation for the Lookahead Optimizer.☆241Updated 3 years ago
- Pretrain and finetune ELECTRA with fastai and huggingface. (Results of the paper replicated !)☆328Updated last year
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.☆105Updated 3 years ago
- Transformer training code for sequential tasks☆612Updated 3 years ago
- Pytorch implementation of Hyperspherical Variational Auto-Encoders☆362Updated 5 years ago
- Source code for the paper "Hyperbolic Neural Networks", https://arxiv.org/abs/1805.09112☆178Updated 4 years ago
- PyTorch implementation of "Lagging Inference Networks and Posterior Collapse in Variational Autoencoders" (ICLR 2019)☆184Updated 4 years ago
- LAnguage Modelling Benchmarks☆137Updated 5 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆225Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Pytorch implementation of Compressive Transformers, from Deepmind☆157Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆763Updated last year