A ranked list of awesome machine learning Python libraries
PyTorch3D is FAIR's library of reusable components for deep learning
Easy-to-use,Modular and Extendible package of deep-learning models
Geometric deep learning extension library for PyTorch
State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX
A Python library for audio data augmentation
A game theoretic approach to explain the output of ml models
Ongoing research training transformer models at scale
2D and 3D Face alignment library build using pytorch
A python library for self-supervised learning on images
PyTorch code and models for V-JEPA self-supervised learning from video
Open source framework for deep learning satellite and aerial imagery
A library for deep learning end-to-end dialog systems and chatbots
AutoML library for deep learning
Our first fully AI generated deep learning system
Pre-trained Deep Learning models and demos
An implementation of a deep learning recommendation model (DLRM)
Probabilistic reasoning and statistical analysis in TensorFlow
An industrial grade federated learning framework
Package of deep-learning based CTR models
Open deep learning compiler stack for cpu, gpu, etc.
PyTorch code and models for VJEPA2 self-supervised learning from video
Petastorm library enables single machine or distributed training
Spatiotemporal Signal Processing with Neural Machine Learning Models
Probabilistic time series modeling in Python