Raincleared-Song / sparse_gpu_operator
GPU operators for sparse tensor operations
☆30Updated 10 months ago
Alternatives and similar repositories for sparse_gpu_operator:
Users that are interested in sparse_gpu_operator are comparing it to the libraries listed below
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆44Updated 9 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- ☆51Updated 9 months ago
- ☆96Updated 4 months ago
- 16-fold memory access reduction with nearly no loss☆63Updated 2 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆152Updated 6 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 4 months ago
- ☆27Updated 9 months ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆29Updated 6 months ago
- ☆23Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆290Updated 6 months ago
- Code for Palu: Compressing KV-Cache with Low-Rank Projection☆63Updated 2 months ago
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆90Updated 2 months ago
- An algorithm for static activation quantization of LLMs☆107Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆58Updated 2 months ago
- ☆107Updated 3 months ago
- PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".☆81Updated last year
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆32Updated 5 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆89Updated 8 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆234Updated last month
- PyTorch library for cost-effective, fast and easy serving of MoE models.☆112Updated last month
- A sparse attention kernel supporting mix sparse patterns☆93Updated 3 months ago
- [NeurIPS 2024] The official implementation of "Kangaroo: Lossless Self-Speculative Decoding for Accelerating LLMs via Double Early Exitin…☆47Updated 6 months ago
- Code Repository of Evaluating Quantized Large Language Models☆112Updated 4 months ago
- ☆22Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆97Updated 8 months ago
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆85Updated 3 months ago
- ☆55Updated 3 months ago