qiuzh20 / gated_attention
The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free
☆17Updated this week
Alternatives and similar repositories for gated_attention
Users that are interested in gated_attention are comparing it to the libraries listed below
Sorting:
- Revisiting Mid-training in the Era of RL Scaling☆37Updated 3 weeks ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆72Updated last month
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆29Updated last month
- Offcial Repo of Paper "Eliminating Position Bias of Language Models: A Mechanistic Approach""☆14Updated 8 months ago
- Official implementation of AAAI 2025 paper "Augmenting Math Word Problems via Iterative Question Composing"(https://arxiv.org/abs/2401.09…☆20Updated 5 months ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆44Updated 6 months ago
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆29Updated last year
- ☆17Updated 11 months ago
- Implementation of "Decoding-time Realignment of Language Models", ICML 2024.☆18Updated 10 months ago
- Codebase for Instruction Following without Instruction Tuning☆34Updated 7 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Training☆21Updated 8 months ago
- GSM-Plus: Data, Code, and Evaluation for Enhancing Robust Mathematical Reasoning in Math Word Problems.☆61Updated 10 months ago
- ☆47Updated last year
- Simple and efficient pytorch-native transformer training and inference (batched)☆75Updated last year
- Stick-breaking attention☆53Updated 2 months ago
- ☆54Updated 10 months ago
- Self-Supervised Alignment with Mutual Information☆18Updated 11 months ago
- ☆31Updated last year
- ☆28Updated last month
- ☆22Updated last month
- Collections of RLxLM experiments using minimal codes☆12Updated 2 months ago
- [NeurIPS 2023] Repetition In Repetition Out: Towards Understanding Neural Text Degeneration from the Data Perspective☆30Updated last year
- Repo for ACL2023 Findings paper "Emergent Modularity in Pre-trained Transformers"☆23Updated last year
- ☆37Updated 8 months ago
- Awesome Triton Resources☆27Updated 2 weeks ago
- Long Context Extension and Generalization in LLMs☆55Updated 7 months ago
- The source code of "Merging Experts into One: Improving Computational Efficiency of Mixture of Experts (EMNLP 2023)":☆38Updated last year
- A framework for few-shot evaluation of autoregressive language models.☆24Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆59Updated 7 months ago