yuezhouhu / 2by4-pretrainLinks
Efficient 2:4 sparse training algorithms and implementations
☆54Updated 5 months ago
Alternatives and similar repositories for 2by4-pretrain
Users that are interested in 2by4-pretrain are comparing it to the libraries listed below
Sorting:
- Code for "Accelerating Transformer Pre-training with 2:4 Sparsity"☆23Updated 5 months ago
- ☆49Updated 10 months ago
- 16-fold memory access reduction with nearly no loss☆94Updated 2 months ago
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆158Updated 3 weeks ago
- A sparse attention kernel supporting mix sparse patterns☆222Updated 3 months ago
- 🔥 A minimal training framework for scaling FLA models☆146Updated 3 weeks ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆203Updated 2 weeks ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆68Updated 7 months ago
- ☆93Updated last week
- AFPQ code implementation☆21Updated last year
- Official Pytorch Implementation of "Outlier Weighed Layerwise Sparsity (OWL): A Missing Secret Sauce for Pruning LLMs to High Sparsity"☆66Updated 11 months ago
- ☆129Updated 3 months ago
- [COLM 2024] SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models☆20Updated 8 months ago
- [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More☆45Updated 3 months ago
- ☆74Updated 3 months ago
- [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation☆98Updated 2 months ago
- ☆21Updated 2 months ago
- The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"☆41Updated 2 months ago
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆82Updated 6 months ago
- An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization☆136Updated 2 weeks ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆163Updated 10 months ago
- ☆28Updated 10 months ago
- Code Repository of Evaluating Quantized Large Language Models☆123Updated 8 months ago
- Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)☆46Updated this week
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆38Updated 9 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆291Updated 6 months ago
- Quantized Attention on GPU☆44Updated 6 months ago
- [ICLR2025] Codebase for "ReMoE: Fully Differentiable Mixture-of-Experts with ReLU Routing", built on Megatron-LM.☆78Updated 5 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆93Updated last week
- [ICML 2025] SliM-LLM: Salience-Driven Mixed-Precision Quantization for Large Language Models☆31Updated 9 months ago