ucbrise / actnn
ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
☆201Updated last year
Related projects ⓘ
Alternatives and complementary repositories for actnn
- ☆214Updated 2 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆129Updated last year
- [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework☆274Updated 11 months ago
- [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy☆156Updated 4 years ago
- Pytorch implementation of BRECQ, ICLR 2021☆253Updated 3 years ago
- ☆39Updated 3 years ago
- ☆44Updated 3 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆571Updated 6 months ago
- Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".☆104Updated last year
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆195Updated 2 years ago
- [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization☆95Updated 2 years ago
- This repository contains integer operators on GPUs for PyTorch.☆183Updated last year
- Group Fisher Pruning for Practical Network Compression(ICML2021)☆154Updated last year
- (CVPR 2021, Oral) Dynamic Slimmable Network☆228Updated 2 years ago
- PyTorch implementation of Data Free Quantization Through Weight Equalization and Bias Correction.☆258Updated last year
- ☆223Updated 3 years ago
- CVPR 2021 : Zero-shot Adversarial Quantization (ZAQ)☆65Updated 3 years ago
- ☆41Updated 2 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆159Updated 2 years ago
- Low Precision Arithmetic Simulation in PyTorch☆265Updated 6 months ago
- PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)☆75Updated last year
- [NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers☆171Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆296Updated 3 years ago
- code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"☆103Updated 3 years ago
- [ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark☆104Updated last year
- [CVPR 2019, Oral] HAQ: Hardware-Aware Automated Quantization with Mixed Precision☆370Updated 3 years ago
- PyTorch implementation for the APoT quantization (ICLR 2020)☆268Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated last year
- Code for ICML 2021 submission☆35Updated 3 years ago
- Pruning Neural Networks with Taylor criterion in Pytorch☆314Updated 5 years ago