teelinsan / parallel-decodingLinks
Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
☆123Updated 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:
- ☆39Updated last year
- ☆150Updated 2 years ago
- ☆106Updated last year
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆29Updated last year
- Sparse Backpropagation for Mixture-of-Expert Training☆29Updated last year
- Triton implementation of FlashAttention2 that adds Custom Masks.☆160Updated last year
- ☆103Updated 10 months ago
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆73Updated 6 months ago
- Low-bit optimizers for PyTorch☆137Updated 2 years ago
- ☆53Updated last year
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆31Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆96Updated last year
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆104Updated last year
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆65Updated last year
- ☆156Updated 11 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆107Updated 3 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Updated last year
- Kinetics: Rethinking Test-Time Scaling Laws☆85Updated 6 months ago
- The evaluation framework for training-free sparse attention in LLMs☆108Updated 3 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆220Updated last year
- Code for paper "Patch-Level Training for Large Language Models"☆96Updated 2 months ago
- Simple and efficient pytorch-native transformer training and inference (batched)☆79Updated last year
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- [ICLR‘24 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆103Updated 6 months ago
- Long Context Extension and Generalization in LLMs☆62Updated last year
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆152Updated 10 months ago
- ☆83Updated 2 years ago
- Official implementation of "DoRA: Weight-Decomposed Low-Rank Adaptation"☆124Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆79Updated last year
- ☆143Updated last year