JAX-based neural network library
Graph Neural Network Library for PyTorch
Models and examples built with TensorFlow
Petastorm library enables single machine or distributed training
AI Toolkit for Healthcare Imaging
Learn how to develop, deploy and iterate on production-grade ML
A comprehensive quantitative trading system with AI-powered analysis
Flower: A Friendly Federated Learning Framework
Implementation of 'lightweight' GAN, proposed in ICLR 2021
Unified Model Serving Framework
Powering Amazon custom machine learning chips
Modular quant framework
Deep learning optimization library making distributed training easy
Reference implementations of MLPerf™ training benchmarks
A library for deep learning end-to-end dialog systems and chatbots
A Python library for audio
DeepMind's software stack for physics-based simulation
Implementation of Imagen, Google's Text-to-Image Neural Network
The Unified Machine Learning Framework
Geometric deep learning extension library for PyTorch
Open Source Differentiable Computer Vision Library
Fast image augmentation library and an easy-to-use wrapper
Django friendly finite state machine support
Python examples of popular machine learning algorithms
Train machine learning models within Docker containers