jack-willturner / gymnasticsLinks
A "gym" style toolkit for building lightweight NAS systems.
☆13Updated 3 years ago
Alternatives and similar repositories for gymnastics
Users that are interested in gymnastics are comparing it to the libraries listed below
Sorting:
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 4 years ago
- PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021☆56Updated 4 years ago
- Official implementation of "UNAS: Differentiable Architecture Search Meets Reinforcement Learning", CVPR 2020 Oral☆61Updated last year
- ☆70Updated 5 years ago
- ☆41Updated 4 years ago
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated last year
- Training neural networks in TensorFlow 2.0 with 5x less memory☆133Updated 3 years ago
- ☆10Updated 3 years ago
- Compression schema for gradients of activations in backward pass☆44Updated 2 years ago
- Official PyTorch implementation of "Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets" (ICLR 2021)☆64Updated last year
- Factorized Neural Layers☆30Updated 2 years ago
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 3 years ago
- PyTorch implementation of HashedNets☆36Updated 2 years ago
- MixPath: A Unified Approach for One-shot Neural Architecture Search☆29Updated 5 years ago
- Reference implementations of popular Binarized Neural Networks☆108Updated 2 months ago
- AlphaNet Improved Training of Supernet with Alpha-Divergence☆99Updated 4 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆29Updated 6 months ago
- PyTorch implementation of L2L execution algorithm☆108Updated 2 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆138Updated 4 years ago
- Code for ICML 2021 submission☆34Updated 4 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 2 years ago
- Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …☆52Updated 3 years ago
- ☆43Updated last year
- Official implementation of Neurips 2020 "Sparse Weight Activation Training" paper.☆28Updated 4 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 4 years ago
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆51Updated 2 years ago
- Torch Distributed Experimental☆117Updated last year
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆200Updated 2 years ago
- [ECCV 2022] SuperTickets: Drawing Task-Agnostic Lottery Tickets from Supernets via Jointly Architecture Searching and Parameter Pruning☆21Updated 3 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆169Updated 3 years ago