JAX-based neural network library
Open deep learning compiler stack for cpu, gpu, etc.
Python library for defining and optimizing mathematical expressions
AutoML library for deep learning
Implementation of Imagen, Google's Text-to-Image Neural Network
Implementation of 'lightweight' GAN, proposed in ICLR 2021
This repos contains notebooks for the Advanced Solutions Lab
End-to-End Library for Continual Learning based on PyTorch
The Unified Machine Learning Framework
Geometric deep learning extension library for PyTorch
The most intuitive, flexible, way for researchers to build models
Python examples of popular machine learning algorithms
An MLOps framework to package, deploy, monitor and manage models
Train machine learning models within Docker containers
Python package built to ease deep learning on graph
Reference implementations of MLPerf™ training benchmarks
Django friendly finite state machine support
A python library for self-supervised learning on images
Jittor is a high-performance deep learning framework
Open Source Differentiable Computer Vision Library
Fast image augmentation library and an easy-to-use wrapper
A unified framework for scalable computing
A library for deep learning end-to-end dialog systems and chatbots
The Triton Inference Server provides an optimized cloud
A python library for easy manipulation and forecasting of time series