MIT HAN Lab

Repositories

• gan-compression

  [CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs

  compression pytorch gans pix2pix cyclegan image-to-image-translation conditional-gans
  Python 92 649 4 0 Updated Sep 16, 2020

• e3d
  Python MIT 3 50 1 0 Updated Sep 15, 2020

• torchsparse
  Python 1 42 0 0 Updated Sep 9, 2020

• neurips-micronet

  [JMLR 2020] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion

  natural-language-processing language-modeling pruning quantization knowledge-distillation efficient-model
  Jupyter Notebook 1 17 0 0 Updated Aug 19, 2020

• data-efficient-gans

  Differentiable Augmentation for Data-Efficient GAN Training

  tensorflow pytorch generative-adversarial-network image-generation gans data-efficient
  Python BSD-2-Clause 38 329 1 0 Updated Aug 8, 2020

• once-for-all

  [ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

  acceleration nas automl edge-ai efficient-model tinyml
  Python Apache-2.0 112 835 17 2 Updated Aug 3, 2020

• lite-transformer

  [ICLR 2020] Lite Transformer with Long-Short Range Attention
  Python 42 329 3 0 Updated Jul 25, 2020

• temporal-shift-module

  [ICCV 2019] TSM: Temporal Shift Module for Efficient Video Understanding

  acceleration low-latency video-understanding tsm nvidia-jetson-nano efficient-model temporal-modeling
  Python Apache-2.0 242 1,041 42 4 Updated Jul 3, 2020

• hardware-aware-transformers

  [ACL 2020] HAT: Hardware-Aware Transformers for Efficient Natural Language Processing

  natural-language-processing machine-translation transformer specialization efficient-model hardware-aware
  Python 12 162 2 0 Updated Jul 2, 2020

• haq

  [CVPR 2019, Oral] HAQ: Hardware-Aware Automated Quantization with Mixed Precision

  quantization automl mixed-precision efficient-model
  Python Apache-2.0 48 216 7 1 Updated Jul 1, 2020

• lpcvc

  [LPIRC 2019, ICCV 2019] Winner Solution for 4th LPCVC
  Python 3 30 0 0 Updated Jun 29, 2020

• pvcnn

  [NeurIPS 2019, Spotlight] Point-Voxel CNN for Efficient 3D Deep Learning

  point-cloud pytorch shapenet kitti pointnet pointnet2 frustum-pointnet
  Python MIT 61 283 9 0 Updated Jun 25, 2020

• apq

  [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy
  Python 8 53 1 0 Updated Jun 16, 2020

• proxylessnas

  [ICLR 2019] ProxylessNAS: Direct Neural Architecture Search on Target Task and Hardware

  acceleration automl specialization efficient-model on-device-ai hardware-aware
  Python Apache-2.0 230 1,135 0 0 Updated May 8, 2020

• VWW

  [LPIRC 2019] First place in the Visual Wake Words challenge (TF-lite track)

  demo tensorflow nas on-device-ml low-power-vision small-model-design efficient-model
  Python Apache-2.0 5 48 2 0 Updated Mar 3, 2020

• dlg

  [NeurIPS 2019] Deep Leakage From Gradients
  Python MIT 21 95 1 1 Updated Feb 12, 2020

• amc

  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices

  automl model-compression channel-pruning automl-for-compression efficient-model on-device-ai
  Python MIT 58 251 11 1 Updated Oct 25, 2019

• amc-models

  [ECCV 2018] AMC: AutoML for Model Compression and Acceleration on Mobile Devices

  automl model-compression efficient-model on-device-ai
  Python Apache-2.0 27 150 0 0 Updated Oct 25, 2019

• bnn-icestick

  Binary Neural Network on IceStick FPGA.

  fpga hardware-acceleration low-power-vision on-device-ai
  Jupyter Notebook MIT 10 41 0 0 Updated Jul 11, 2018