Awesome Machine Learning

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Deep Learning for humans

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

Milvus is a high-performance, cloud-native vector database built for scalable vector ANN search

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

Apache Spark - A unified analytics engine for large-scale data processing

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

Qdrant - High-performance, massive-scale Vector Database and Vector Search Engine for the next generation of AI. Also available in the cloud https://cloud.qdrant.io/

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

Data Apps & Dashboards for Python. No JavaScript Required.

Graph Neural Network Library for PyTorch

Debug, evaluate, and monitor your LLM applications, RAG systems, and agentic workflows with comprehensive tracing, automated evaluations, and production-ready dashboards.

Minimalist ML framework for Rust

A computer algebra system written in pure Python

A hyperparameter optimization framework

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

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

A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

Lime: Explaining the predictions of any machine learning classifier

Statsmodels: statistical modeling and econometrics in Python

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

💥 Fast State-of-the-Art Tokenizers optimized for Research and Production

Production ready toolkit to run AI locally

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

Open-source, low-code AutoML platform for Python. PyCaret 4.0: sklearn-native engine + React control plane.

Bayesian Modeling and Probabilistic Programming in Python

PraisonAI 🦞 — Hire a 24/7 AI Workforce. Stop writing boilerplate and start shipping autonomous self-improving agents that research, plan, code, and execute tasks. Deployed in 5 lines of code with built-in memory, RAG, and support for 100+ LLMs.

Out-of-Core hybrid Apache Arrow/NumPy DataFrame for Python, ML, visualization and exploration of big tabular data at a billion rows per second 🚀

Production infrastructure for machine learning at scale

Evidently is ​​an open-source ML and LLM observability framework. Evaluate, test, and monitor any AI-powered system or data pipeline. From tabular data to Gen AI. 100+ metrics.

Fit interpretable models. Explain blackbox machine learning.

Image processing in Python

A Python wrapper for Google Tesseract

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

A scikit-learn compatible neural network library that wraps PyTorch

A library of extension and helper modules for Python's data analysis and machine learning libraries.

The AI-native database built for LLM applications, providing incredibly fast hybrid search of dense vector, sparse vector, tensor (multi-vector), and full-text.

Fast data visualization and GUI tools for scientific / engineering applications

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

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.

OpenDILab Decision AI Engine. The Most Comprehensive Reinforcement Learning Framework B.P.

Main repository for Vispy

Algorithmic Trading in Python with Machine Learning

Accelerated deep learning R&D

Python Toolkit for Causal and Probabilistic Reasoning

🔅 Shapash: User-friendly Explainability and Interpretability to Develop Reliable and Transparent Machine Learning Models

a delightful machine learning tool that allows you to train, test, and use models without writing code

Transform ML models into a native code (Java, C, Python, Go, JavaScript, Visual Basic, C#, R, PowerShell, PHP, Dart, Haskell, Ruby, F#, Rust) with zero dependencies