Awesome Machine Learning

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Deep Learning for humans

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

Visualizer for neural network, deep learning and machine learning models

The fastai deep learning library

Graph Neural Network Library for PyTorch

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

A scikit-learn compatible neural network library that wraps PyTorch

Rust bindings for the C++ api of PyTorch.

An Engine-Agnostic Deep Learning Framework in Java

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and models from research to production.

AI Infra / AI Orchestration / AI Control Plane

Fast, flexible and easy to use probabilistic modelling in Python.

🛠 All-in-one web-based IDE specialized for machine learning and data science.

Accelerated deep learning R&D

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

A toolkit for reproducible reinforcement learning research.

A PyTorch and TorchDrug based deep learning library for drug pair scoring. (KDD 2022)