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

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.

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!

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

The fastai deep learning library

Open-source AI orchestration framework for building context-engineered, production-ready LLM applications. Design modular pipelines and agent workflows with explicit control over retrieval, routing, memory, and generation. Built for scalable agents, RAG, multimodal applications, semantic search, and conversational systems.

🤖 Python examples of popular machine learning algorithms with interactive Jupyter demos and math being explained

Data Apps & Dashboards for Python. No JavaScript Required.

Interactive Data Visualization in the browser, from Python

Zipline, a Pythonic Algorithmic Trading Library

A computer algebra system written in pure Python

A hyperparameter optimization framework

Approximate Nearest Neighbors in C++/Python optimized for memory usage and loading/saving to disk

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.

💡 All-in-one AI framework for semantic search, LLM orchestration and language model workflows

Statsmodels: statistical modeling and econometrics in Python

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

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

Web mining module for Python, with tools for scraping, natural language processing, machine learning, network analysis and visualization.

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

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.

Image processing in Python

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

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

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

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

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

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

StellarGraph - Machine Learning on Graphs

Deepnote is a drop-in replacement for Jupyter with an AI-first design, sleek UI, new blocks, and native data integrations. Use Python, R, and SQL locally in your favorite IDE, then scale to Deepnote cloud for real-time collaboration, Deepnote agent, and deployable data apps. https://deepnote.com/

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

Build, run and scale AI agents like API and microservices

🧠 Make your agents learn from experience. Now available as a hosted solution at kayba.ai

A modular active learning framework for Python