- Beijing
Starred repositories
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Universal LLM Deployment Engine with ML Compilation
Open standard for machine learning interoperability
Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
Accessible large language models via k-bit quantization for PyTorch.
MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, …
Production First and Production Ready End-to-End Speech Recognition Toolkit
FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs o…
A high performance and generic framework for distributed DNN training
Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!
An open-source framework for machine learning and other computations on decentralized data.
A Flow-based Generative Network for Speech Synthesis
DeepMind's Tacotron-2 Tensorflow implementation
SOTA low-bit LLM quantization (INT8/FP8/INT4/FP4/NF4) & sparsity; leading model compression techniques on TensorFlow, PyTorch, and ONNX Runtime
HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis
General code to convert a trained keras model into an inference tensorflow model
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.
Channel Pruning for Accelerating Very Deep Neural Networks (ICCV'17)
GAN-based Mel-Spectrogram Inversion Network for Text-to-Speech Synthesis
Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019
Fast, Accurate and Lightweight Super-Resolution models
A parser, editor and profiler tool for ONNX models.
A tutorial for getting started with the Deep Learning Accelerator (DLA) on NVIDIA Jetson