Toolkit for making machine learning and data analysis applications
Petastorm library enables single machine or distributed training
Elegant and Performant Deep Learning
Package of deep-learning based CTR models
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
Unity machine learning agents toolkit
Jittor is a high-performance deep learning framework
Medical imaging toolkit for deep learning
A distributed system for embedding-based vector retrieval
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
AutoGluon: AutoML for Image, Text, and Tabular Data
A python library for self-supervised learning on images
OpenVINO™ Toolkit repository
AI Toolkit for Healthcare Imaging
Spatiotemporal Signal Processing with Neural Machine Learning Models
Nx-powered Neural Networks
oneAPI Deep Neural Network Library (oneDNN)
Deep Learning Visualization Toolkit
Decentralized deep learning in PyTorch. Built to train models
Deep learning library
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
A Python library for audio data augmentation
A refreshing functional take on deep learning
On-device wake word detection powered by deep learning