teelinsan / parallel-decodingLinks
Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
☆118Updated last year
Alternatives and similar repositories for parallel-decoding
Users that are interested in parallel-decoding are comparing it to the libraries listed below
Sorting:
- ☆147Updated 2 years ago
- Sparse Backpropagation for Mixture-of-Expert Training☆29Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆92Updated last year
- ☆76Updated 4 months ago
- ☆38Updated 10 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆91Updated last month
- Triton implementation of FlashAttention2 that adds Custom Masks.☆124Updated 11 months ago
- 🔥 A minimal training framework for scaling FLA models☆188Updated last month
- ☆105Updated last year
- ☆81Updated last year
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆57Updated 4 months ago
- [KDD'22] Learned Token Pruning for Transformers☆98Updated 2 years ago
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆52Updated 2 years 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
- The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free☆44Updated 2 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year
- Low-bit optimizers for PyTorch☆129Updated last year
- [ICLR 2024] CLEX: Continuous Length Extrapolation for Large Language Models☆78Updated last year
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆86Updated 3 weeks ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆101Updated last year
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆60Updated 9 months ago
- Code for paper "Patch-Level Training for Large Language Models"☆85Updated 7 months ago
- [EVA ICLR'23; LARA ICML'22] Efficient attention mechanisms via control variates, random features, and importance sampling☆86Updated 2 years ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆107Updated 3 months ago
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆146Updated 4 months ago
- ☆136Updated 5 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆224Updated 7 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆212Updated 10 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆70Updated last year
- The implementation for MLSys 2023 paper: "Cuttlefish: Low-rank Model Training without All The Tuning"☆45Updated 2 years ago