deep-spin / adasplashLinks
AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)
☆29Updated last month
Alternatives and similar repositories for adasplash
Users that are interested in adasplash are comparing it to the libraries listed below
Sorting:
- Kinetics: Rethinking Test-Time Scaling Laws☆82Updated 4 months ago
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆36Updated last year
- [ICLR 2025] Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxia…☆27Updated 3 months ago
- ☆16Updated last year
- Xmixers: A collection of SOTA efficient token/channel mixers☆29Updated 2 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆124Updated 4 months ago
- [ACL 2025] Squeezed Attention: Accelerating Long Prompt LLM Inference☆54Updated 11 months ago
- ☆35Updated 8 months ago
- Code for "RSQ: Learning from Important Tokens Leads to Better Quantized LLMs"☆19Updated 5 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year
- ☆14Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆102Updated last month
- Stick-breaking attention☆61Updated 4 months ago
- ☆10Updated last year
- ☆96Updated 8 months ago
- ☆25Updated 7 months ago
- The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…☆101Updated last month
- ☆26Updated 3 months ago
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆67Updated last year
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆80Updated last year
- Beyond KV Caching: Shared Attention for Efficient LLMs☆20Updated last year
- Fast and memory-efficient exact attention☆74Updated 8 months ago
- [EMNLP 2024] Quantize LLM to extremely low-bit, and finetune the quantized LLMs☆15Updated last year
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)☆35Updated last year
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆23Updated 2 months ago
- [ICML24] Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆95Updated 11 months ago
- Triton implement of bi-directional (non-causal) linear attention☆56Updated 9 months ago
- ☆30Updated last year
- ☆38Updated last year
- Muon fsdp 2☆44Updated 3 months ago