Raincleared-Song / sparse_gpu_operator
GPU operators for sparse tensor operations
☆29Updated 7 months ago
Related projects ⓘ
Alternatives and complementary repositories for sparse_gpu_operator
- Code for Palu: Compressing KV-Cache with Low-Rank Projection☆54Updated this week
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆25Updated 4 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 10 months ago
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆42Updated 7 months ago
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆57Updated this week
- ☆20Updated last month
- Odysseus: Playground of LLM Sequence Parallelism☆53Updated 4 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆146Updated 3 months ago
- Implementation of Kangaroo: Lossless Self-Speculative Decoding via Double Early Exiting☆41Updated 4 months ago
- ☆88Updated 2 months ago
- ☆42Updated 7 months ago
- A sparse attention kernel supporting mix sparse patterns☆52Updated 3 weeks ago
- A minimal implementation of vllm.☆30Updated 3 months ago
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆32Updated 2 months ago
- ☆24Updated 7 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆195Updated last week
- PyTorch bindings for CUTLASS grouped GEMM.☆51Updated last week
- Repository for CPU Kernel Generation for LLM Inference☆24Updated last year
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆34Updated 8 months ago
- PyTorch library for cost-effective, fast and easy serving of MoE models.☆101Updated 2 months ago
- Official Repo for SparseLLM: Global Pruning of LLMs (NeurIPS 2024)☆34Updated 2 weeks ago
- ☆43Updated this week
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆196Updated last week
- An algorithm for static activation quantization of LLMs☆67Updated last month
- ☆22Updated 10 months ago
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆79Updated 5 months ago
- AFPQ code implementation☆18Updated last year
- ☆130Updated 3 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆51Updated 2 months ago
- [ICML 2024 Oral] Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs☆80Updated 2 months ago