MIT HAN Lab

Repositories

• data-efficient-gans

  [NeurIPS 2020] Differentiable Augmentation for Data-Efficient GAN Training

  tensorflow pytorch generative-adversarial-network image-generation gans data-efficient neurips-2020
  Python BSD-2-Clause 116 829 12 0 Updated Jul 27, 2021

• spvnas

  [ECCV 2020] Searching Efficient 3D Architectures with Sparse Point-Voxel Convolution

  computer-vision deep-learning efficiency point-cloud pytorch lidar 3d-deep-learning
  Python MIT 55 286 3 2 Updated Jul 27, 2021

• gan-compression

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

  compression pytorch gans pix2pix cyclegan image-to-image-translation conditional-gans
  Python 116 851 10 1 Updated Jul 25, 2021

• calo-cluster
  Jupyter Notebook 1 0 0 6 Updated Jul 23, 2021

• torchsparse

  A high-performance neural network library for point cloud processing.

  point-cloud pytorch
  Python MIT 42 261 8 0 Updated Jul 21, 2021

• anycost-gan

  [CVPR 2021] Anycost GANs for Interactive Image Synthesis and Editing

  computer-vision deep-learning computer-graphics pytorch image-editing generative-adversarial-network gan
  Python MIT 55 523 3 0 Updated Jul 8, 2021

• 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 MIT 212 1,273 34 2 Updated Jul 1, 2021

• lite-transformer

  [ICLR 2020] Lite Transformer with Long-Short Range Attention
  Python 56 454 7 0 Updated Jun 12, 2021

• pytorch-quantum

  A PyTorch-centric hybrid classical-quantum machine learning framework

  machine-learning neural-network quantum pytorch quantum-computing quantum-machine-learning pytorch-quantum
  Python MIT 4 52 0 0 Updated Jun 6, 2021

• tinyml
  Python MIT 44 279 7 0 Updated Jun 5, 2021

• pvcnn

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

  point-cloud pytorch shapenet kitti pointnet pointnet2 frustum-pointnet
  Python MIT 92 394 2 0 Updated Jun 1, 2021

• gan-compression-dynamic
  Python 0 1 0 0 Updated May 16, 2021

• data-efficient-gans-dynamic
  Python BSD-2-Clause 0 0 0 0 Updated May 15, 2021

• 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 MIT 332 1,421 75 7 Updated Apr 30, 2021

• inter-operator-scheduler

  [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration

  acceleration cnn parallelism inference-optimization
  C++ MIT 11 86 0 0 Updated Apr 6, 2021

• e3d

  Efficient 3D Deep Learning
  0 15 0 0 Updated Mar 19, 2021

• 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 MIT 3 32 0 0 Updated Feb 25, 2021

• proxylessnas

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

  acceleration automl specialization efficient-model on-device-ai hardware-aware
  Python MIT 253 1,251 0 0 Updated Feb 26, 2021

• amc-models

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

  automl model-compression efficient-model on-device-ai
  Python MIT 26 153 0 0 Updated Feb 26, 2021

• haq

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

  quantization automl mixed-precision efficient-model
  Python MIT 69 263 12 1 Updated Feb 26, 2021

• 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 76 317 13 2 Updated Feb 20, 2021

• 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 34 230 0 0 Updated Nov 10, 2020

• apq

  [CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy
  Python 21 105 3 0 Updated Jun 16, 2020

• dlg

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

• bnn-icestick

  Binary Neural Network on IceStick FPGA.

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