MayDomine / Burst-Attention
Distributed IO-aware Attention algorithm
☆17Updated 3 weeks ago
Related projects: ⓘ
- Sequence-level 1F1B schedule for LLMs.☆16Updated 3 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆78Updated 4 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆50Updated 3 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting☆60Updated 6 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆47Updated 2 weeks ago
- PyTorch bindings for CUTLASS grouped GEMM.☆57Updated 2 months ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆183Updated last month
- Implementation of Kangaroo: Lossless Self-Speculative Decoding via Double Early Exiting☆39Updated 2 months ago
- ☆164Updated 4 months ago
- ☆105Updated last week
- An innovative method expediting LLMs via streamlined semi-autoregressive generation and draft verification.☆21Updated 7 months ago
- ☆60Updated last month
- ☆29Updated 4 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆134Updated 2 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆309Updated this week
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆37Updated 2 months ago
- Code associated with the paper **Draft & Verify: Lossless Large Language Model Acceleration via Self-Speculative Decoding**☆127Updated 3 months ago
- A LLaMA1/LLaMA12 Megatron implement.☆26Updated 9 months ago
- ☆83Updated 3 weeks ago
- ☆75Updated this week
- REST: Retrieval-Based Speculative Decoding, NAACL 2024☆158Updated 4 months ago
- ☆67Updated last week
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 9 months ago
- ☆102Updated 3 months ago
- Official implementation of the ICLR 2024 paper AffineQuant☆16Updated 5 months ago
- Low-bit optimizers for PyTorch☆109Updated 11 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆161Updated 2 months ago
- ☆32Updated 7 months ago
- Longitudinal Evaluation of LLMs via Data Compression☆25Updated 3 months ago
- Spec-Bench: A Comprehensive Benchmark and Unified Evaluation Platform for Speculative Decoding (ACL 2024 Findings)☆156Updated 3 months ago