yuezhouhu / 2by4-pretrainLinks
Efficient 2:4 sparse training algorithms and implementations
☆54Updated 6 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 6 months ago
- 16-fold memory access reduction with nearly no loss☆99Updated 3 months ago
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆166Updated this week
- ☆51Updated 11 months ago
- A sparse attention kernel supporting mix sparse patterns☆238Updated 4 months ago
- Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)☆47Updated 3 weeks ago
- [ICLR 2025] COAT: Compressing Optimizer States and Activation for Memory-Efficient FP8 Training☆210Updated last week
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆71Updated 8 months ago
- An algorithm for weight-activation quantization (W4A4, W4A8) of LLMs, supporting both static and dynamic quantization☆137Updated last month
- ☆21Updated 3 months ago
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆38Updated 10 months ago
- ☆52Updated 6 months ago
- 🔥 A minimal training framework for scaling FLA models☆178Updated 2 weeks ago
- ☆130Updated 4 months ago
- Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"☆36Updated 3 weeks ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆120Updated 10 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆163Updated 11 months ago
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆65Updated last year
- ☆114Updated 3 weeks ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆113Updated last month
- [ICLR 2025] Palu: Compressing KV-Cache with Low-Rank Projection☆123Updated 4 months ago
- Quantized Attention on GPU☆44Updated 7 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆99Updated 3 weeks ago
- [COLM 2024] SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models☆20Updated 8 months ago
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"☆233Updated 2 weeks ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆39Updated last year
- ☆75Updated 5 months ago
- ☆20Updated last year
- [NeurIPS 2024 Oral🔥] DuQuant: Distributing Outliers via Dual Transformation Makes Stronger Quantized LLMs.☆161Updated 8 months ago
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆113Updated last week