GuoTianYu2000 / Active-Dormant-AttentionLinks
codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"
☆10Updated last year
Alternatives and similar repositories for Active-Dormant-Attention
Users that are interested in Active-Dormant-Attention are comparing it to the libraries listed below
Sorting:
- Unofficial Implementation of Selective Attention Transformer☆20Updated last year
- ☆35Updated last year
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)☆39Updated last year
- ☆19Updated 9 months ago
- [ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)☆152Updated 6 months ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆63Updated 10 months ago
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆26Updated last month
- Stick-breaking attention☆62Updated 6 months ago
- ☆20Updated 2 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆85Updated 6 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆109Updated 3 months ago
- ☆52Updated last month
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆87Updated last year
- Implementation of CoLA: Compute-Efficient Pre-Training of LLMs via Low-Rank Activation☆25Updated 11 months ago
- ☆104Updated 10 months ago
- [ICML 2024] SPP: Sparsity-Preserved Parameter-Efficient Fine-Tuning for Large Language Models☆21Updated last year
- Code for the EMNLP24 paper "A simple and effective L2 norm based method for KV Cache compression."☆18Updated last year
- A Sober Look at Language Model Reasoning☆92Updated 2 months ago
- ☆51Updated last year
- Efficient Scaling laws and collaborative pretraining.☆19Updated 4 months ago
- ☆73Updated 6 months ago
- LISA: Layerwise Importance Sampling for Memory-Efficient Large Language Model Fine-Tuning☆36Updated last year
- official implementation of ICLR'2025 paper: Rethinking Bradley-Terry Models in Preference-based Reward Modeling: Foundations, Theory, and…☆70Updated 9 months ago
- ☆53Updated last year
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆128Updated 6 months ago
- Position Coupling: Improving Length Generalization of Arithmetic Transformers Using Task Structure (NeurIPS 2024) + Arithmetic Transfor…☆11Updated 2 months ago
- Benchmarking Optimizers for LLM Pretraining☆48Updated 2 weeks ago
- Code for "Reasoning to Learn from Latent Thoughts"☆124Updated 9 months ago
- Test-time-training on nearest neighbors for large language models☆49Updated last year
- [ICML‘2024] "LoCoCo: Dropping In Convolutions for Long Context Compression", Ruisi Cai, Yuandong Tian, Zhangyang Wang, Beidi Chen☆18Updated last year