A ranked list of awesome machine learning Python libraries
PyTorch code and models for V-JEPA self-supervised learning from video
An implementation of a deep learning recommendation model (DLRM)
Decentralized deep learning in PyTorch. Built to train models
Medical imaging toolkit for deep learning
Probabilistic reasoning and statistical analysis in TensorFlow
Making large AI models cheaper, faster and more accessible
An industrial grade federated learning framework
PyTorch code and models for VJEPA2 self-supervised learning from video
A Python library for audio data augmentation
Petastorm library enables single machine or distributed training
Spatiotemporal Signal Processing with Neural Machine Learning Models
Probabilistic time series modeling in Python
AutoGluon: AutoML for Image, Text, and Tabular Data
Library for serving Transformers models on Amazon SageMaker
Python package built to ease deep learning on graph
Best practices on recommendation systems
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
A unified framework for scalable computing