sramshetty / mixture-of-depths
An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
☆33Updated 3 months ago
Related projects: ⓘ
- Unofficial implementations of block/layer-wise pruning methods for LLMs.☆45Updated 4 months ago
- Implementation of the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆56Updated this week
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆43Updated 3 weeks ago
- ☆75Updated this week
- Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs☆68Updated 3 months ago
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆63Updated 3 months ago
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆55Updated 5 months ago
- ☆29Updated 7 months ago
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆28Updated last month
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆123Updated 2 months ago
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆134Updated 2 months ago
- Implementation of Kangaroo: Lossless Self-Speculative Decoding via Double Early Exiting☆39Updated 2 months ago
- Official Implementation of SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocks☆28Updated 2 months ago
- Get down and dirty with FlashAttention2.0 in pytorch, plug in and play no complex CUDA kernels☆90Updated last year
- ☆44Updated 11 months ago
- Pytorch implementation of our paper accepted by ICML 2024 -- CaM: Cache Merging for Memory-efficient LLMs Inference☆21Updated 3 months ago
- Code repo for the paper "SpinQuant LLM quantization with learned rotations"☆79Updated this week
- Triton implementation of Flash Attention2.0☆21Updated last year
- Low-bit optimizers for PyTorch☆109Updated 11 months ago
- ☆164Updated 4 months ago
- PB-LLM: Partially Binarized Large Language Models☆143Updated 9 months ago
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆87Updated 8 months ago
- ☆94Updated 6 months ago
- Code accompanying the paper "Massive Activations in Large Language Models"☆104Updated 6 months ago
- EE-LLM is a framework for large-scale training and inference of early-exit (EE) large language models (LLMs).☆44Updated 3 months ago
- The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆76Updated 2 weeks ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆25Updated 5 months ago
- AFPQ code implementation☆15Updated 10 months ago
- Code for Palu: Compressing KV-Cache with Low-Rank Projection☆39Updated this week
- [ACL 2024] Not All Experts are Equal: Efficient Expert Pruning and Skipping for Mixture-of-Experts Large Language Models☆59Updated 3 months ago