A ranked list of awesome machine learning Python libraries
High-level, high-performance dynamic language for technical computing
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
A distributed system for embedding-based vector retrieval
Medical imaging toolkit for deep learning
Easy-to-use,Modular and Extendible package of deep-learning models
Probabilistic reasoning and statistical analysis in TensorFlow
A GPU-accelerated library containing highly optimized building blocks
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
An industrial grade federated learning framework
Package of deep-learning based CTR models
Geometric deep learning extension library for PyTorch
Industrial-strength Natural Language Processing (NLP)
PyTorch code and models for VJEPA2 self-supervised learning from video
Nx-powered Neural Networks
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
A refreshing functional take on deep learning