ucbrise / actnn
ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
☆200Updated 2 years ago
Alternatives and similar repositories for actnn:
Users that are interested in actnn are comparing it to the libraries listed below
- ☆225Updated 3 years ago
- ☆230Updated 2 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆132Updated last year
- [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy☆157Updated 4 years ago
- code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"☆104Updated 3 years ago
- Pytorch implementation of BRECQ, ICLR 2021☆272Updated 3 years ago
- Group Fisher Pruning for Practical Network Compression(ICML2021)☆156Updated last year
- ☆40Updated 3 years ago
- [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework☆278Updated last year
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- [ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark☆110Updated 2 years ago
- PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)☆74Updated 2 years ago
- [ICLR 2022 Oral] F8Net: Fixed-Point 8-bit Only Multiplication for Network Quantization☆94Updated 3 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆197Updated 3 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆168Updated 3 years ago
- AlphaNet Improved Training of Supernet with Alpha-Divergence☆98Updated 3 years ago
- Code for ICML 2021 submission☆34Updated 4 years ago
- PyTorch implementation for the APoT quantization (ICLR 2020)☆271Updated 4 months ago
- ☆44Updated 3 years ago
- (CVPR 2021, Oral) Dynamic Slimmable Network☆229Updated 3 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆604Updated last year
- Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models☆66Updated 2 years ago
- [ICLR 2022] The Unreasonable Effectiveness of Random Pruning: Return of the Most Naive Baseline for Sparse Training by Shiwei Liu, Tianlo…☆73Updated 2 years ago
- ☆42Updated 2 years ago
- Code for the NeurIPS 2022 paper "Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning".☆118Updated last year
- code for NASViT☆67Updated 3 years ago
- [ICML 2022] "DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks", by Yonggan …☆71Updated 2 years ago
- PyTorch implementation of Data Free Quantization Through Weight Equalization and Bias Correction.☆262Updated last year
- pytorch-profiler☆51Updated last year
- XNAS: An effective, modular, and flexible Neural Architecture Search (NAS) framework.☆48Updated 2 years ago