Awesome Machine Learning

An Open Source Machine Learning Framework for Everyone

🤗 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.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Open Source Computer Vision Library

Deep Learning for humans

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

Streamlit — A faster way to build and share data apps.

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

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Making large AI models cheaper, faster and more accessible

Caffe: a fast open framework for deep learning.

OpenPose: Real-time multi-person keypoint detection library for body, face, hands, and foot estimation

💫 Industrial-strength Natural Language Processing (NLP) in Python

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.

Deep learning library featuring a higher-level API for TensorFlow.

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

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

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

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

Accelerated deep learning R&D

Rust native ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT2,...)

StellarGraph - Machine Learning on Graphs

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

Python library for time series forecasting using scikit-learn compatible models, statistical methods, and foundation models

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

ML powered analytics engine for outlier detection and root cause analysis.

Little Ball of Fur - A graph sampling extension library for NetworKit and NetworkX (CIKM 2020)

The official implementation of "The Shapley Value of Classifiers in Ensemble Games" (CIKM 2021).