aiha-lab / Attention-Head-Pruning
Layer-wise Pruning of Transformer Heads for Efficient Language Modeling
☆21Updated 2 years ago
Alternatives and similar repositories for Attention-Head-Pruning:
Users that are interested in Attention-Head-Pruning are comparing it to the libraries listed below
- TernGEMM: General Matrix Multiply Library with Ternary Weights for Fast DNN Inference☆13Updated 2 years ago
- ☆197Updated 3 years ago
- The official PyTorch implementation of the NeurIPS2022 (spotlight) paper, Outlier Suppression: Pushing the Limit of Low-bit Transformer L…☆48Updated 2 years ago
- Code for the AAAI 2024 Oral paper "OWQ: Outlier-Aware Weight Quantization for Efficient Fine-Tuning and Inference of Large Language Model…☆56Updated 11 months ago
- Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".☆109Updated last year
- [NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers☆181Updated last year
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆267Updated 5 months ago
- DeiT implementation for Q-ViT☆24Updated 2 years ago
- QAQ: Quality Adaptive Quantization for LLM KV Cache☆45Updated 10 months ago
- An algorithm for static activation quantization of LLMs☆113Updated this week
- The official implementation of the ICML 2023 paper OFQ-ViT☆30Updated last year
- [EMNLP 2024] RoLoRA: Fine-tuning Rotated Outlier-free LLMs for Effective Weight-Activation Quantization☆29Updated 4 months ago
- ☆16Updated last year
- [ACL 2024] A novel QAT with Self-Distillation framework to enhance ultra low-bit LLMs.☆99Updated 8 months ago
- [TMLR] Official PyTorch implementation of paper "Efficient Quantization-aware Training with Adaptive Coreset Selection"☆30Updated 5 months ago
- SliM-LLM: Salience-Driven Mixed-Precision Quantization for Large Language Models☆26Updated 5 months ago
- ☆39Updated last year
- Code Repository of Evaluating Quantized Large Language Models☆115Updated 4 months ago
- Code repo for the paper BiT Robustly Binarized Multi-distilled Transformer☆102Updated last year
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆95Updated 2 weeks ago
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆47Updated last year
- AFPQ code implementation☆19Updated last year
- Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocks☆33Updated this week
- QUICK: Quantization-aware Interleaving and Conflict-free Kernel for efficient LLM inference☆116Updated 11 months ago
- Reproducing Quantization paper PACT☆60Updated 2 years ago
- ☆19Updated 2 months ago
- Official Pytorch Implementation of "Outlier Weighed Layerwise Sparsity (OWL): A Missing Secret Sauce for Pruning LLMs to High Sparsity"☆57Updated 7 months ago
- This project is the official implementation of our accepted ICLR 2022 paper BiBERT: Accurate Fully Binarized BERT.☆85Updated last year
- ☆74Updated 2 years ago
- ☆20Updated 11 months ago