Deep universal probabilistic programming with Python and PyTorch
Geometric deep learning extension library for PyTorch
PaddlePaddle End-to-End Development Toolkit
A ranked list of awesome machine learning Python libraries
Industrial-strength Natural Language Processing (NLP)
Library for serving Transformers models on Amazon SageMaker
Open deep learning compiler stack for cpu, gpu, etc.
Jittor is a high-performance deep learning framework
A set of Docker images for training and serving models in TensorFlow
Audiocraft is a library for audio processing and generation
A Python toolbox for scalable outlier detection
A python library for easy manipulation and forecasting of time series
Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Fast image augmentation library and an easy-to-use wrapper
A library for deep learning end-to-end dialog systems and chatbots
Powering Amazon custom machine learning chips
AutoML library for deep learning
A python library for self-supervised learning on images
AI Toolkit for Healthcare Imaging
Petastorm library enables single machine or distributed training
Python package built to ease deep learning on graph
A unified framework for scalable computing
Our first fully AI generated deep learning system
Deep learning optimization library: makes distributed training easy