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

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

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

A toolkit for developing and comparing reinforcement learning algorithms.

结巴中文分词

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

Caffe: a fast open framework for deep learning.

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

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

MLX: An array framework for Apple silicon

Test your prompts, agents, and RAGs. Red teaming/pentesting/vulnerability scanning for AI. Compare performance of GPT, Claude, Gemini, DeepSeek, and more. Simple declarative configs with command line and CI/CD integration. Used by OpenAI and Anthropic.

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

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

Apache Hadoop

A computer algebra system written in pure Python

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

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

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

Bayesian Modeling and Probabilistic Programming in Python

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

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.

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.

Vowpal Wabbit is a machine learning system which pushes the frontier of machine learning with techniques such as online, hashing, allreduce, reductions, learning2search, active, and interactive learning.

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.

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.

Fit interpretable models. Explain blackbox machine learning.

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)