vineeths96 / Compressed-Transformers
In this repository, we explore model compression for transformer architectures via quantization. We specifically explore quantization aware training of the linear layers and demonstrate the performance for 8 bits, 4 bits, 2 bits and 1 bit (binary) quantization.
☆23Updated 3 years ago
Alternatives and similar repositories for Compressed-Transformers:
Users that are interested in Compressed-Transformers are comparing it to the libraries listed below
- [ICASSP'22] Integer-only Zero-shot Quantization for Efficient Speech Recognition☆31Updated 3 years ago
- Code for High-Capacity Expert Binary Networks (ICLR 2021).☆27Updated 3 years ago
- Position-based Scaled Gradient for Model Quantization and Pruning Code (NeurIPS 2020)☆26Updated 4 years ago
- Pytorch implementation of BiFSMN, IJCAI 2022☆21Updated 2 years ago
- This repository containts the pytorch scripts to train mixed-precision networks for microcontroller deployment, based on the memory contr…☆49Updated 9 months ago
- [ICML 2022] "Coarsening the Granularity: Towards Structurally Sparse Lottery Tickets" by Tianlong Chen, Xuxi Chen, Xiaolong Ma, Yanzhi Wa…☆32Updated last year
- Implementation of NeurIPS 2019 paper "Normalization Helps Training of Quantized LSTM"☆30Updated 7 months ago
- Post-training sparsity-aware quantization☆34Updated 2 years ago
- BitSplit Post-trining Quantization☆49Updated 3 years ago
- This is an implementation of YOLO using LSQ network quantization method.☆23Updated 2 years ago
- Reproducing Quantization paper PACT☆61Updated 2 years ago
- An 8bit automated quantization conversion tool for the pytorch (Post-training quantization based on KL divergence)☆33Updated 5 years ago
- A Out-of-box PyTorch Scaffold for Neural Network Quantization-Aware-Training (QAT) Research. Website: https://github.com/zhutmost/neuralz…☆26Updated 2 years ago
- ☆17Updated 2 years ago
- DeiT implementation for Q-ViT☆24Updated 2 years ago
- Pytorch implementation of BiFSMNv2, TNNLS 2023☆30Updated 2 years ago
- [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization☆94Updated 2 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
- ☆28Updated 3 years ago
- Manually implemented quantization-aware training☆21Updated 2 years ago
- OTOv1-v3, NeurIPS, ICLR, TMLR, DNN Training, Compression, Structured Pruning, Erasing Operators, CNN, Diffusion, LLM☆31Updated 4 months ago
- PyTorch implementation of "Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference"☆55Updated 5 years ago
- Layer-wise Pruning of Transformer Heads for Efficient Language Modeling☆21Updated 3 years ago
- Implementation of a Quantized Transformer Model☆18Updated 5 years ago
- Quantize pytorch model, support post-training quantization and quantization aware training methods☆13Updated last year
- PyTorch Quantization Aware Training Example☆130Updated 9 months ago
- [TMLR] Official PyTorch implementation of paper "Efficient Quantization-aware Training with Adaptive Coreset Selection"☆30Updated 6 months ago
- [ICML 2023] This project is the official implementation of our accepted ICML 2023 paper BiBench: Benchmarking and Analyzing Network Binar…☆54Updated 11 months ago
- A collection of research papers on efficient training of DNNs☆70Updated 2 years ago
- ☆13Updated last year