The most intuitive, flexible, way for researchers to build models
End-to-End Library for Continual Learning based on PyTorch
Tribuo - A Java machine learning library
Training PyTorch models with differential privacy
Test Suites for validating ML models & data
JAX-based neural network library
OneFlow is a deep learning framework designed to be user-friendly
Graph Neural Network Library for PyTorch
Open deep learning compiler stack for cpu, gpu, etc.
A library for scientific machine learning & physics-informed learning
A cross-platform Python library for differentiable programming
A general-purpose probabilistic programming system
AutoML library for deep learning
Implementation of Imagen, Google's Text-to-Image Neural Network
A PyTorch-based Speech Toolkit
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Powering Amazon custom machine learning chips
Simple and distributed Machine Learning
A unified, comprehensive and efficient recommendation library
A ranked list of awesome machine learning Python libraries
Modular quant framework
A GPU-accelerated library containing highly optimized building blocks
The Unified Machine Learning Framework
Geometric deep learning extension library for PyTorch
Open source platform for the machine learning lifecycle