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.

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.

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

A toolkit for developing and comparing reinforcement learning algorithms.

结巴中文分词

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

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

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.

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.

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

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

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

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.

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

Bayesian Modeling and Probabilistic Programming in Python

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

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

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.

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 🚀

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

Genetic Algorithm, Particle Swarm Optimization, Simulated Annealing, Ant Colony Optimization Algorithm,Immune Algorithm, Artificial Fish Swarm Algorithm, Differential Evolution and TSP(Traveling salesman)

Image processing in Python

A Python wrapper for Google Tesseract

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.

Algorithmic Trading in Python with Machine Learning

Accelerated deep learning R&D

Python Toolkit for Causal and Probabilistic Reasoning

NumPy and Pandas interface to Big Data

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

StellarGraph - Machine Learning on Graphs