thu-ml / Adaptive-Sparse-TrainerLinks
Official implementation for "Pruning Large Language Models with Semi-Structural Adaptive Sparse Training" (AAAI 2025)
☆12Updated last month
Alternatives and similar repositories for Adaptive-Sparse-Trainer
Users that are interested in Adaptive-Sparse-Trainer are comparing it to the libraries listed below
Sorting:
- ☆14Updated 10 months ago
- KV cache compression via sparse coding☆12Updated 3 months ago
- ☆26Updated 2 weeks ago
- ☆14Updated 8 months ago
- A sparse attention kernel supporting mix sparse patterns☆267Updated 6 months ago
- Official code for the paper "HEXA-MoE: Efficient and Heterogeneous-Aware MoE Acceleration with Zero Computation Redundancy"☆13Updated 5 months ago
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆128Updated 5 months ago
- [ICLR2025]: OSTQuant: Refining Large Language Model Quantization with Orthogonal and Scaling Transformations for Better Distribution Fitt…☆72Updated 4 months ago
- An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization☆147Updated 2 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆317Updated last month
- Code Repository of Evaluating Quantized Large Language Models☆129Updated 11 months ago
- [ICML 2025] Official PyTorch implementation of "FlatQuant: Flatness Matters for LLM Quantization"☆152Updated 3 weeks ago
- [NeurIPS 2024 Oral🔥] DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs.☆165Updated 10 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆127Updated 2 months ago
- Code for Neurips24 paper: QuaRot, an end-to-end 4-bit inference of large language models.☆414Updated 8 months ago
- Official code implementation for 2025 ICLR accepted paper "Dobi-SVD : Differentiable SVD for LLM Compression and Some New Perspectives"☆36Updated 4 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆215Updated last month
- Official Pytorch Implementation of "Outlier Weighed Layerwise Sparsity (OWL): A Missing Secret Sauce for Pruning LLMs to High Sparsity"☆73Updated last month
- [ICML 2024] KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache☆313Updated 6 months ago
- A code sample demonstrating how to share and rebuild a PyTorch GPU tensor via its pointer/reference between different processes.☆13Updated 11 months ago
- 16-fold memory access reduction with nearly no loss☆104Updated 4 months ago
- ☆51Updated last year
- [ICML 2025] SliM-LLM: Salience-Driven Mixed-Precision Quantization for Large Language Models☆39Updated last year
- SeerAttention: Learning Intrinsic Sparse Attention in Your LLMs☆144Updated this week
- Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)☆56Updated 2 weeks ago
- Pytorch implementation of "Oscillation-Reduced MXFP4 Training for Vision Transformers" on DeiT Model Pre-training☆24Updated last month
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆56Updated last week
- LLM Inference with Microscaling Format☆27Updated 9 months ago
- DiTAS: Quantizing Diffusion Transformers via Enhanced Activation Smoothing (WACV 2025)☆10Updated 8 months ago
- Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation☆15Updated 2 months ago