lucidrains / pause-transformer
Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount of time on any token
☆53Updated last year
Alternatives and similar repositories for pause-transformer:
Users that are interested in pause-transformer are comparing it to the libraries listed below
- ☆51Updated 11 months ago
- ☆37Updated last year
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆37Updated last year
- ☆47Updated last year
- ☆45Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆72Updated 6 months ago
- Language models scale reliably with over-training and on downstream tasks☆96Updated last year
- Implementation of Infini-Transformer in Pytorch☆110Updated 4 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆66Updated 7 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Stick-breaking attention☆52Updated last month
- ☆32Updated last year
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆64Updated last year
- ☆95Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated 8 months ago
- Code for the paper "The Impact of Positional Encoding on Length Generalization in Transformers", NeurIPS 2023☆135Updated last year
- This repo is based on https://github.com/jiaweizzhao/GaLore☆27Updated 7 months ago
- ☆31Updated last year
- ☆78Updated 8 months ago
- ☆52Updated 7 months ago
- ☆81Updated last year
- [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling☆36Updated last year
- ☆25Updated last year
- Here we will test various linear attention designs.☆60Updated last year
- ☆43Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆105Updated this week
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 10 months ago
- PyTorch implementation of Soft MoE by Google Brain in "From Sparse to Soft Mixtures of Experts" (https://arxiv.org/pdf/2308.00951.pdf)☆71Updated last year
- Randomized Positional Encodings Boost Length Generalization of Transformers☆81Updated last year