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)
A GPU-accelerated library containing highly optimized building blocks
Making large AI models cheaper, faster and more accessible
Graph Neural Network Library for PyTorch
Geometric deep learning extension library for PyTorch
PyTorch code and models for VJEPA2 self-supervised learning from video
Petastorm library enables single machine or distributed training
Best practices on recommendation systems
Minimal Deep Q Learning (DQN & DDQN) implementations in Keras
The data structure for multimodal data
Enabling PyTorch on Google TPU
An engine-agnostic deep learning framework in Java
Jupyter notebooks that demonstrate how to build models using SageMaker
Unity machine learning agents toolkit
Python package built to ease deep learning on graph
Open source framework for deep learning satellite and aerial imagery
Open Source Computer Vision Library
A library for deep learning end-to-end dialog systems and chatbots
An industrial grade federated learning framework
Adaptive Intelligence also known as "Artificial General Intelligence"
ktrain is a Python library that makes deep learning AI more accessible
Deep universal probabilistic programming with Python and PyTorch
Audiocraft is a library for audio processing and generation