A ranked list of awesome machine learning Python libraries
Our first fully AI generated deep learning system
Pre-trained Deep Learning models and demos
An implementation of a deep learning recommendation model (DLRM)
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
Probabilistic reasoning and statistical analysis in TensorFlow
Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
PyTorch code and models for VJEPA2 self-supervised learning from video
ktrain is a Python library that makes deep learning AI more accessible
A Python library for audio data augmentation
Petastorm library enables single machine or distributed training
Spatiotemporal Signal Processing with Neural Machine Learning Models
AutoGluon: AutoML for Image, Text, and Tabular Data
Library for serving Transformers models on Amazon SageMaker
MII makes low-latency and high-throughput inference possible
The data structure for multimodal data
Open source framework for deep learning satellite and aerial imagery
Jittor is a high-performance deep learning framework
Fast image augmentation library and an easy-to-use wrapper
Hub of ready-to-use datasets for ML models
Build cross-modal and multimodal applications on the cloud