huggingface / nn_pruningLinks
Prune a model while finetuning or training.
☆403Updated 3 years ago
Alternatives and similar repositories for nn_pruning
Users that are interested in nn_pruning are comparing it to the libraries listed below
Sorting:
- Parallelformers: An Efficient Model Parallelization Toolkit for Deployment☆790Updated 2 years ago
- Library for 8-bit optimizers and quantization routines.☆779Updated 3 years ago
- Implementation of a Transformer, but completely in Triton☆276Updated 3 years ago
- Root Mean Square Layer Normalization☆256Updated 2 years ago
- FastFormers - highly efficient transformer models for NLU☆707Updated 8 months ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆197Updated 2 years ago
- Repository containing code for "How to Train BERT with an Academic Budget" paper☆316Updated 2 years ago
- A library for researching neural networks compression and acceleration methods.☆140Updated 2 months ago
- [ACL'20] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing☆336Updated last year
- Fast Block Sparse Matrices for Pytorch☆548Updated 4 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆431Updated 3 years ago
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆612Updated 3 years ago
- Sequence modeling with Mega.☆301Updated 2 years ago
- [NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers☆192Updated 2 years ago
- Efficient, check-pointed data loading for deep learning with massive data sets.☆209Updated 2 years ago
- Accelerate PyTorch models with ONNX Runtime☆366Updated 8 months ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆228Updated 3 years ago
- A simple and working implementation of Electra, the fastest way to pretrain language models from scratch, in Pytorch☆235Updated 2 years ago
- Long Range Arena for Benchmarking Efficient Transformers☆767Updated last year
- Understanding the Difficulty of Training Transformers☆332Updated 3 years ago
- ⚡ boost inference speed of T5 models by 5x & reduce the model size by 3x.☆589Updated 2 years ago
- This is a repository with the code for the ACL 2019 paper "Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, t…☆315Updated 4 years ago
- Code for the ALiBi method for transformer language models (ICLR 2022)☆544Updated 2 years ago
- Block Sparse movement pruning☆81Updated 4 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆236Updated 2 years ago
- Pretrain and finetune ELECTRA with fastai and huggingface. (Results of the paper replicated !)☆330Updated last year
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆156Updated last year
- Repository for the paper "Optimal Subarchitecture Extraction for BERT"☆470Updated 3 years ago
- Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint☆415Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆301Updated 5 months ago