teelinsan / parallel-decodingLinks
Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"
☆121Updated 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☆29Updated last year
- ☆106Updated last year
- ☆150Updated 2 years ago
- [ICML'24 Oral] The official code of "DiJiang: Efficient Large Language Models through Compact Kernelization", a novel DCT-based linear at…☆104Updated last year
- Triton implementation of FlashAttention2 that adds Custom Masks.☆157Updated last year
- Code for paper "Patch-Level Training for Large Language Models"☆96Updated last month
- ☆39Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆74Updated last year
- ☆101Updated 9 months ago
- [EMNLP 2022] Official implementation of Transnormer in our EMNLP 2022 paper - The Devil in Linear Transformer☆64Updated 2 years ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆29Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆107Updated 2 months ago
- ☆142Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆95Updated last year
- LongSpec: Long-Context Lossless Speculative Decoding with Efficient Drafting and Verification☆69Updated 5 months ago
- ☆83Updated 2 years 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 9 months ago
- [ICLR 2024] CLEX: Continuous Length Extrapolation for Large Language Models☆78Updated last year
- Low-bit optimizers for PyTorch☆134Updated 2 years ago
- ☆22Updated 2 years ago
- [KDD'22] Learned Token Pruning for Transformers☆102Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆219Updated last year
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆102Updated last year
- ☆53Updated last year
- [ICML'24] The official implementation of “Rethinking Optimization and Architecture for Tiny Language Models”☆126Updated 11 months ago
- "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
- ☆60Updated 6 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆84Updated 5 months ago
- ☆51Updated last year