JAX-based neural network library
A library for scientific machine learning & physics-informed learning
TensorFlow is an open source library for machine learning
Petastorm library enables single machine or distributed training
Open Source Differentiable Computer Vision Library
A unified, comprehensive and efficient recommendation library
Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
A python library for easy manipulation and forecasting of time series
End-to-End Library for Continual Learning based on PyTorch
Python library for defining and optimizing mathematical expressions
AIMET is a library that provides advanced quantization and compression
Fast image augmentation library and an easy-to-use wrapper
Deep learning optimization library making distributed training easy
A python library for self-supervised learning on images
Machine learning, conversational dialog engine for creating chat bots
A cross-platform Python library for differentiable programming
Library for OCR-related tasks powered by Deep Learning
Geometric deep learning extension library for PyTorch
AutoML library for deep learning
Models and examples built with TensorFlow
The Unified Machine Learning Framework
DeepMind's software stack for physics-based simulation
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Python examples of popular machine learning algorithms