jack-willturner / gymnastics
A "gym" style toolkit for building lightweight NAS systems.
☆13Updated 2 years ago
Related projects: ⓘ
- [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion☆40Updated 3 years ago
- Official implementation of "UNAS: Differentiable Architecture Search Meets Reinforcement Learning", CVPR 2020 Oral☆59Updated 11 months ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆20Updated 4 years ago
- ☆13Updated 2 years ago
- Implementation of Kronecker Attention in Pytorch☆17Updated 4 years ago
- Code for BlockSwap (ICLR 2020).☆33Updated 3 years ago
- Train neural networks with joint quantization and pruning on both weights and activations using any pytorch modules☆40Updated 2 years ago
- ☆21Updated this week
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆28Updated last year
- ICML2019 Accepted Paper. Overcoming Multi-Model Forgetting☆13Updated 5 years ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- ☆25Updated 3 years ago
- MixPath: A Unified Approach for One-shot Neural Architecture Search☆28Updated 4 years ago
- Factorized Neural Layers☆27Updated last year
- All about acceleration and compression of Deep Neural Networks☆33Updated 4 years ago
- Arch-Net: Model Distillation for Architecture Agnostic Model Deployment☆22Updated 2 years ago
- Python pdb for multiple processes☆30Updated last year
- ☆22Updated 6 years ago
- Official PyTorch implementation of "Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets" (ICLR 2021)☆63Updated last month
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- ☆12Updated this week
- Memory Optimizations for Deep Learning (ICML 2023)☆58Updated 6 months ago
- ☆13Updated this week
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Updated 2 years ago
- a high performance system for customized-precision distributed deep learning☆12Updated 3 years ago
- Code for High-Capacity Expert Binary Networks (ICLR 2021).☆26Updated 2 years ago
- FLOPs and other statistics COunter for Pytorch neural networks☆23Updated 3 years ago
- The official implementation of You Only Compress Once: Towards Effective and Elastic BERT Compression via Exploit-Explore Stochastic Natu…☆48Updated 3 years ago
- Implementation of Spectral Leakage and Rethinking the Kernel Size in CNNs in Pytorch☆14Updated 3 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆48Updated 3 months ago