Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Python examples of popular machine learning algorithms
The unified and scalable ML library for large-scale training
NVIDIA Federated Learning Application Runtime Environment
A toolkit to optimize ML models for deployment for Keras & TensorFlow
High quality, fast, modular reference implementation of SSD in PyTorch
Enabling PyTorch on Google TPU
Determined, deep learning training platform
Graph Neural Network Library for PyTorch
Serve machine learning models within a Docker container
An open source python library for automated feature engineering
An open-source, low-code machine learning library in Python
OneFlow is a deep learning framework designed to be user-friendly
Deep Learning Visualization Toolkit
Build MLOps Pipelines in Minutes
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
A unified, comprehensive and efficient recommendation library
ONNX-TensorRT: TensorRT backend for ONNX
Proofs, cases, concept supplements, and reference explanations
A library for scientific machine learning & physics-informed learning
Training PyTorch models with differential privacy
Usable Implementation of "Bootstrap Your Own Latent" self-supervised
Hyperparameter Optimization for TensorFlow, Keras and PyTorch
High-level training, data augmentation, and utilities for Pytorch