teelinsan / parallel-decoding
Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
☆116Updated last year
Alternatives and similar repositories for parallel-decoding:
Users that are interested in parallel-decoding are comparing it to the libraries listed below
- ☆143Updated last year
- ☆69Updated 2 months ago
- Sparse Backpropagation for Mixture-of-Expert Training☆29Updated 10 months ago
- 🔥 A minimal training framework for scaling FLA models☆119Updated this week
- Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"☆76Updated last year
- ☆103Updated last year
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year
- ☆49Updated 11 months ago
- Code for paper "Patch-Level Training for Large Language Models"☆84Updated 5 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆209Updated 8 months ago
- ☆36Updated 8 months ago
- ☆126Updated 2 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆110Updated 8 months ago
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆59Updated 7 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
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆51Updated 2 years ago
- ☆77Updated 2 weeks ago
- ☆44Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆59Updated 6 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆65Updated 9 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆212Updated 5 months ago
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆82Updated 11 months ago
- ☆103Updated 11 months ago
- The source code of "Merging Experts into One: Improving Computational Efficiency of Mixture of Experts (EMNLP 2023)":☆37Updated last year
- ☆84Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆90Updated last year
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆36Updated 8 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆58Updated last month
- ☆130Updated 9 months ago
- [NeurIPS 2024 Spotlight] Code and data for the paper "Finding Transformer Circuits with Edge Pruning".☆54Updated last month