NonvolatileMemory / flash_attn_gqaLinks
triton ver of gqa flash attn, based on the tutorial
☆11Updated 11 months ago
Alternatives and similar repositories for flash_attn_gqa
Users that are interested in flash_attn_gqa are comparing it to the libraries listed below
Sorting:
- Implementation of the model: "Reka Core, Flash, and Edge: A Series of Powerful Multimodal Language Models" in PyTorch☆30Updated 2 weeks ago
- ☆14Updated 2 years ago
- ☆35Updated last year
- A probabilitic model for contextual word representation. Accepted to ACL2023 Findings.☆23Updated last year
- ☆20Updated last year
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated last year
- ☆56Updated last year
- Code for the preprint "Cache Me If You Can: How Many KVs Do You Need for Effective Long-Context LMs?"☆38Updated this week
- The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free☆45Updated 2 months ago
- Long Context Extension and Generalization in LLMs☆57Updated 9 months ago
- ☆19Updated 2 years ago
- Codebase for Instruction Following without Instruction Tuning☆35Updated 9 months ago
- [ICLR'24 spotlight] Tool-Augmented Reward Modeling☆50Updated last month
- This repository is the official implementation of our EMNLP 2022 paper ELMER: A Non-Autoregressive Pre-trained Language Model for Efficie…☆26Updated 2 years ago
- Code for ICML 25 paper "Metadata Conditioning Accelerates Language Model Pre-training (MeCo)"☆41Updated 2 weeks ago
- Use the tokenizer in parallel to achieve superior acceleration☆16Updated last year
- Official code repo for paper "Great Memory, Shallow Reasoning: Limits of kNN-LMs"☆23Updated 2 months ago
- Official Implementation of ACL2023: Don't Parse, Choose Spans! Continuous and Discontinuous Constituency Parsing via Autoregressive Span …☆13Updated last year
- ☆21Updated 2 years ago
- ☆22Updated last year
- ☆53Updated 3 weeks ago
- The source code of "Merging Experts into One: Improving Computational Efficiency of Mixture of Experts (EMNLP 2023)":☆38Updated last year
- ☆12Updated last year
- ☆32Updated last year
- Awesome Triton Resources☆32Updated 2 months ago
- Code for "[COLM'25] RepoST: Scalable Repository-Level Coding Environment Construction with Sandbox Testing"☆22Updated 3 months ago
- ☆24Updated last year
- [EMNLP 2023]Context Compression for Auto-regressive Transformers with Sentinel Tokens☆24Updated last year
- Scaling Sparse Fine-Tuning to Large Language Models☆16Updated last year
- Code for the ACL-2022 paper "StableMoE: Stable Routing Strategy for Mixture of Experts"☆47Updated 3 years ago