shawntan / stickbreaking-attention
Stick-breaking attention
☆41Updated this week
Alternatives and similar repositories for stickbreaking-attention:
Users that are interested in stickbreaking-attention are comparing it to the libraries listed below
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆25Updated 9 months ago
- ☆29Updated 10 months ago
- ☆46Updated 11 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆45Updated last month
- ☆78Updated 10 months ago
- ☆51Updated 7 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆66Updated 2 months ago
- DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆71Updated last month
- Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆73Updated 2 weeks ago
- Here we will test various linear attention designs.☆58Updated 8 months ago
- Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"☆65Updated 11 months ago
- ☆46Updated 6 months ago
- Long Context Extension and Generalization in LLMs☆40Updated 3 months ago
- [NeurIPS 2024 Spotlight] Code and data for the paper "Finding Transformer Circuits with Edge Pruning".☆42Updated last month
- Xmixers: A collection of SOTA efficient token/channel mixers☆11Updated 2 months ago
- Official implementation of Bootstrapping Language Models via DPO Implicit Rewards☆41Updated 5 months ago
- ☆27Updated 2 months ago
- [NeurIPS-2024] 📈 Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623☆75Updated 3 months ago
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆48Updated last year
- Repo for ACL2023 Findings paper "Emergent Modularity in Pre-trained Transformers"☆21Updated last year
- Sparse Backpropagation for Mixture-of-Expert Training☆27Updated 6 months ago
- Fast and memory-efficient exact attention☆52Updated last month
- Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆28Updated 2 months ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆80Updated last year
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆25Updated 8 months ago
- ☆98Updated 10 months ago
- Test-time-training on nearest neighbors for large language models☆37Updated 9 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆62Updated 8 months ago
- ☆44Updated last year
- ☆78Updated 3 months ago