A library for scientific machine learning & physics-informed learning
Implementation of Imagen, Google's Text-to-Image Neural Network
A library for deep learning end-to-end dialog systems and chatbots
Flower: A Friendly Federated Learning Framework
A unified, comprehensive and efficient recommendation library
Unified Model Serving Framework
A ranked list of awesome machine learning Python libraries
Modular quant framework
Deep learning optimization library making distributed training easy
AIMET is a library that provides advanced quantization and compression
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Powering Amazon custom machine learning chips
The fastest way to build data pipelines
Easy-to-use,Modular and Extendible package of deep-learning models
The Unified Machine Learning Framework
Geometric deep learning extension library for PyTorch
Jittor is a high-performance deep learning framework
Python examples of popular machine learning algorithms
An MLOps framework to package, deploy, monitor and manage models
Petastorm library enables single machine or distributed training
Train machine learning models within Docker containers
A python library for self-supervised learning on images
Fast image augmentation library and an easy-to-use wrapper
An agentic Machine Learning Engineer
A library of extension and helper modules for Python's data analysis