An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Images to inference with no labeling (use foundation models to train supervised models).

Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".

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

A modern hardware definition language and toolchain based on Python

This is a collection of our NAS and Vision Transformer work.

Machine learning on FPGAs using HLS

Vitis AI is Xilinx’s development stack for AI inference on Xilinx hardware platforms, including both edge devices and Alveo cards.

Brevitas: neural network quantization in PyTorch

The MyHDL development repository

Modular hardware build system

Dataflow compiler for QNN inference on FPGAs

LLM model quantization (compression) toolkit with hw acceleration support for Nvidia CUDA, AMD ROCm, Intel XPU and Intel/AMD/Apple CPU via HF, vLLM, and SGLang.

NASLib is a Neural Architecture Search (NAS) library for facilitating NAS research for the community by providing interfaces to several state-of-the-art NAS search spaces and optimizers.

On-Device Training Under 256KB Memory [NeurIPS'22]

Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.

A Python compiler design toolkit.

Conversion of PyTorch Models into TFLite

A DNN inference latency prediction toolkit for accurately modeling and predicting the latency on diverse edge devices.

Naszilla is a Python library for neural architecture search (NAS)

Save, Load Frozen Graph and Run Inference From Frozen Graph in TensorFlow 1.x and 2.x

Communication framework for RTL simulation and emulation.

A collection of classes providing simple hardware specification, simulation, tracing, and testing suitable for teaching and research. Simplicity, usability, clarity, and extensibility are the overa…

DC3: A Learning Method for Optimization with Hard Constraints

[CVPR 2020] APQ: Joint Search for Network Architecture, Pruning and Quantization Policy

[ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark

Multiclass classification of vibration signals of faulty bearings

A tensorflow based Faster RCNN inception v2 python model to detect and count humans in real time images, videos & camera.

Smart Meter Data Collector