mitchellgordon95 / bert-prune
☆17Updated 4 years ago
Alternatives and similar repositories for bert-prune:
Users that are interested in bert-prune are comparing it to the libraries listed below
- [KDD'22] Learned Token Pruning for Transformers☆96Updated 2 years ago
- Method to improve inference time for BERT. This is an implementation of the paper titled "PoWER-BERT: Accelerating BERT Inference via Pro…☆61Updated last year
- Block Sparse movement pruning☆78Updated 4 years ago
- [NeurIPS 2020] "The Lottery Ticket Hypothesis for Pre-trained BERT Networks", Tianlong Chen, Jonathan Frankle, Shiyu Chang, Sijia Liu, Ya…☆140Updated 3 years ago
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆89Updated last year
- This package implements THOR: Transformer with Stochastic Experts.☆62Updated 3 years ago
- Code for ACL2022 publication Transkimmer: Transformer Learns to Layer-wise Skim☆21Updated 2 years ago
- Parameter Efficient Transfer Learning with Diff Pruning☆73Updated 4 years ago
- Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)☆101Updated 4 years ago
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆56Updated 5 months ago
- Code for the ACL-2022 paper "StableMoE: Stable Routing Strategy for Mixture of Experts"☆45Updated 2 years ago
- Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…☆21Updated 6 months ago
- Code for the paper "BERT Loses Patience: Fast and Robust Inference with Early Exit".☆65Updated 3 years ago
- This project is the official implementation of our accepted ICLR 2022 paper BiBERT: Accurate Fully Binarized BERT.☆87Updated last year
- DEMix Layers for Modular Language Modeling☆53Updated 3 years ago
- Pytorch library for factorized L0-based pruning.☆44Updated last year
- This pytorch package implements PLATON: Pruning Large Transformer Models with Upper Confidence Bound of Weight Importance (ICML 2022).☆43Updated 2 years ago
- DeeBERT: Dynamic Early Exiting for Accelerating BERT Inference☆155Updated 2 years ago
- Official PyTorch implementation of IntactKV: Improving Large Language Model Quantization by Keeping Pivot Tokens Intact☆42Updated 9 months ago
- ☆19Updated 4 years ago
- Source code for NAACL 2021 paper "TR-BERT: Dynamic Token Reduction for Accelerating BERT Inference"☆46Updated 2 years ago
- Code for EMNLP 2020 paper CoDIR☆41Updated 2 years ago
- ☆15Updated 3 years ago
- A plug-in of Microsoft DeepSpeed to fix the bug of DeepSpeed pipeline☆26Updated 3 years ago
- Pytorch implementation of paper "Efficient Nearest Neighbor Language Models" (EMNLP 2021)☆72Updated 3 years ago
- Efficient LLM Inference Acceleration using Prompting☆47Updated 4 months ago
- Staged Training for Transformer Language Models☆32Updated 2 years ago
- Repo for ICML23 "Why do Nearest Neighbor Language Models Work?"☆56Updated 2 years ago
- ☆47Updated 4 years ago