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

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.

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!

A toolkit for developing and comparing reinforcement learning algorithms.

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

Visualizer for neural network, deep learning and machine learning models

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

The fastai deep learning library

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.

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

Interactive Data Visualization in the browser, from Python

A computer algebra system written in pure Python

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.

💥 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

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.

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 🚀

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

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

A scikit-learn compatible neural network library that wraps PyTorch

Rust bindings for the C++ api of PyTorch.

An Engine-Agnostic Deep Learning Framework in Java

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.

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

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/