teelinsan / parallel-decodingLinks
Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
☆119Updated 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:
- Sparse Backpropagation for Mixture-of-Expert Training☆30Updated last year
- ☆148Updated 2 years ago
- ☆106Updated last year
- ☆38Updated 11 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆95Updated last month
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆94Updated last year
- ☆80Updated 6 months ago
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆62Updated last month
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆62Updated 10 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆132Updated last year
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆91Updated 2 months ago
- The source code of "Merging Experts into One: Improving Computational Efficiency of Mixture of Experts (EMNLP 2023)":☆38Updated last year
- Code for paper "Patch-Level Training for Large Language Models"☆86Updated 9 months ago
- ☆139Updated last year
- ☆81Updated last year
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆30Updated last year
- Low-bit optimizers for PyTorch☆130Updated last year
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆148Updated 5 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆70Updated last year
- Official implementation of "DoRA: Weight-Decomposed Low-Rank Adaptation"☆124Updated last year
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆54Updated 2 years ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆102Updated last year
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆87Updated 8 months ago
- 🔥 A minimal training framework for scaling FLA models☆228Updated last week
- [NAACL 2025] A Closer Look into Mixture-of-Experts in Large Language Models☆54Updated 6 months ago
- Long Context Extension and Generalization in LLMs☆58Updated 11 months ago
- The official implementation for Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Free☆46Updated 3 months ago
- [ICLR 2024] CLEX: Continuous Length Extrapolation for Large Language Models☆78Updated last year
- Code accompanying the paper "Massive Activations in Large Language Models"☆176Updated last year