AI & ML

Listed by section, in editorial order. Click a column header to re-sort the whole list.

Press / to search. Tags open their category. Click a row for details.

Search and filter

Results

Row number Tags

AI and Agents 39 projects

Build a typed agent in Pydantic AI. Long-running, stateful agents suit LangGraph. Apps tied to one model vendor can use that vendor's Python AI agent library.

Django backend agent skills for Django, DRF, Celery, and Django-specific code review.
Agent skills for scientific research and analysis with Python libraries like scikit-learn, polars, and biopython.
Agent skills the Sentry team uses for code review, pull requests, and Django reviews.
getsentry/github.com/getsentry/skills /1,100 stars /
Security skills for vulnerability detection, auditing, and testing.
trailofbits/github.com/trailofbits/skills /7,878 stars /
Plugins, agents, and skills for AI coding tools, including Python skills for testing, async, packaging, and uv.
wshobson/github.com/wshobson/agents /41,988 stars /
A framework for building agents and LLM-powered applications.
langchain-ai/github.com/langchain-ai/langchain /148,676 stars /
Low-level orchestration framework for building stateful, long-running LLM agents.
A Python agent framework for building generative AI applications with structured schemas.
pydantic/github.com/pydantic/pydantic-ai /20,947 stars /
A framework for orchestrating role-playing autonomous AI agents for collaborative task solving.
crewAIInc/github.com/crewAIInc/crewAI /60,264 stars /
Anthropic's Python SDK for building AI agents on Claude Code's harness — custom tools, in-process MCP servers, hooks.
OpenAI's framework for building and managing AI agents.
Google's code-first toolkit for building, evaluating, and deploying AI agents.
google/github.com/google/adk-python /22,046 stars /
The official Python SDK for building Model Context Protocol servers and clients.
A high-level, Pythonic framework for building MCP servers and clients.
PrefectHQ/github.com/PrefectHQ/fastmcp /28,238 stars /
An adaptive personal AI assistant that grows with you.
A multi-platform AI assistant that connects LLMs to chat apps like Telegram, Slack, and QQ, extensible with Python plugins.
AstrBotDevs/github.com/AstrBotDevs/AstrBot /44,762 stars /
A framework for programming, not prompting, language models.
stanfordnlp/github.com/stanfordnlp/dspy /38,849 stars /
A library for extracting structured data from LLMs, powered by Pydantic.
567-labs/github.com/567-labs/instructor /14,077 stars /
A toolkit for building RAG pipelines and agents over your data.
run-llama/github.com/run-llama/llama_index /52,735 stars /
An intelligent memory layer for AI agents enabling personalized interactions.
mem0ai/github.com/mem0ai/mem0 /68,063 stars /
A context database for AI agents that unifies memory, resources, and skills.
volcengine/github.com/volcengine/OpenViking /41,150 stars /
A graph-native context and knowledge layer for AI agents with reasoning, provenance, and governance.
The model-definition framework for pretrained models in text, computer vision, audio, video, and multimodal tasks, for inference and training.
Compute embeddings, run retrieval, and rerank results with pretrained and fine-tuned models.
Python client for Ollama, which downloads and runs open-weight LLMs locally.
ollama/github.com/ollama/ollama-python /10,642 stars /
Python bindings for llama.cpp, for running GGUF models locally.
abetlen/github.com/abetlen/llama-cpp-python /10,667 stars /
Run and fine-tune large language models on Apple Silicon with MLX.
ml-explore/github.com/ml-explore/mlx-lm /7,505 stars /
A high-throughput and memory-efficient inference and serving engine for LLMs.
vllm-project/github.com/vllm-project/vllm /95,866 stars /
A high-performance serving framework for large language models and multimodal models.
sgl-project/github.com/sgl-project/sglang /38,364 stars /
DeepSeek's production inference engine for DeepSeek models, with MoE expert parallelism.
deepseek-ai/github.com/deepseek-ai/DeepInfer /24,242 stars /
Pre-trained diffusion models for generating and editing images, audio, and video.
huggingface/github.com/huggingface/diffusers /34,833 stars /
A library for parameter-efficient fine-tuning of large pretrained models.
huggingface/github.com/huggingface/peft /21,859 stars /
A library for post-training transformer language models with SFT, DPO, GRPO, and other trainers.
huggingface/github.com/huggingface/trl /19,630 stars /
Faster, lower-memory LLM fine-tuning, as a Python library or a desktop app.
unslothai/github.com/unslothai/unsloth /78,764 stars /
A framework for fine-tuning and post-training large language models.
A Whisper reimplementation on CTranslate2, up to 4 times faster than openai-whisper with less memory.
SYSTRAN/github.com/SYSTRAN/faster-whisper /26,303 stars /
A general-purpose automatic speech recognition model trained on 680k hours of multilingual and multitask supervised data.
openai/github.com/openai/whisper /111,545 stars /
Python library and CLI tool for converting text to speech using Google Translate TTS.
pndurette/github.com/pndurette/gTTS /2,638 stars /
Industrial-grade speech recognition toolkit with speaker diarization and emotion detection.
modelscope/github.com/modelscope/FunASR /20,874 stars /

Deep Learning 7 projects

PyTorch suits new models, since it debugs like plain Python. Gymnasium environments pass arrays to whichever Python deep learning framework trains your agent.

Tensors and dynamic neural networks in Python with strong GPU acceleration.
pytorch/github.com/pytorch/pytorch /105,001 stars /
Deep learning framework to train, deploy, and ship AI products Lightning fast.
A library for high-performance numerical computing with autodiff and JIT compilation.
jax-ml/github.com/jax-ml/jax /36,524 stars /
A high-level deep learning library with support for JAX, TensorFlow, and PyTorch backends.
keras-team/github.com/keras-team/keras /64,397 stars /
An end-to-end machine learning platform from Google.
tensorflow/github.com/tensorflow/tensorflow /201,211 stars /
A standard API for reinforcement learning environments with popular reference environments (gym successor).
PyTorch implementations of Stable Baselines (deep) reinforcement learning algorithms.
DLR-RM/github.com/DLR-RM/stable-baselines3 /13,946 stars /

Machine Learning 13 projects

Fit models with scikit-learn, which covers a Python machine learning library's basic tasks, boosted trees included. Deep learning takes a framework of its own.

The most popular Python library for machine learning with extensive documentation and community support.
Resampling techniques for imbalanced datasets, compatible with scikit-learn.
A Python library for causal and probabilistic reasoning with graphical models.
pgmpy/github.com/pgmpy/pgmpy /3,364 stars /
sklearn-compatible API with the widest toolset for feature engineering and selection.
A scalable, portable, and distributed gradient boosting library.
dmlc/github.com/dmlc/xgboost /28,924 stars /
A fast, distributed, high-performance gradient boosting framework.
A fast, scalable, high-performance gradient boosting on decision trees library.
catboost/github.com/catboost/catboost /9,182 stars /
A tool for producing forecasts for time series with multiple seasonality and trend changes.
meta/github.com/facebook/prophet /20,464 stars /
Fast statistical forecasting models such as ARIMA, ETS, and Theta, compiled with numba.
Nixtla/github.com/Nixtla/statsforecast /4,954 stars /
A unified scikit-learn-style framework for forecasting and other time-series learning tasks.
sktime/github.com/sktime/sktime /10,081 stars /
A pretrained foundation model from Google Research for time-series forecasting, with non-commercial default weights.
A hyperparameter optimization framework with define-by-run search spaces and pruning of unpromising trials.
optuna/github.com/optuna/optuna /14,988 stars /
Distributed asynchronous hyperparameter optimization with Tree-structured Parzen Estimators.
hyperopt/github.com/hyperopt/hyperopt /7,598 stars /

Natural Language Processing 8 projects

Whether you're building an app or learning the field, there's a Python NLP library for it: spaCy for apps, NLTK for learning. Stanza covers many languages.

A leading platform for building Python programs to work with human language data.
nltk/github.com/nltk/nltk /14,760 stars /
A library for industrial-strength natural language processing in Python and Cython.
explosion/github.com/explosion/spaCy /34,003 stars /
Topic Modeling for Humans.
piskvorky/github.com/piskvorky/gensim /16,509 stars /
The Stanford NLP Group's official Python library, supporting 60+ languages.
stanfordnlp/github.com/stanfordnlp/stanza /7,904 stars /
The most popular Chinese text segmentation library.
fxsjy/github.com/fxsjy/jieba /35,221 stars /
Convert Chinese hanzi (漢字) to pinyin (拼音).
Traditional/Simplified Chinese conversion with regional variants and phrase-level rules.
BYVoid/github.com/BYVoid/OpenCC /10,085 stars /
Paranoid text spacing.
vinta/github.com/vinta/pangu.py /281 stars /

Computer Vision 7 projects

OpenCV processes images and video. Training your own detector needs more than a Python computer vision library scoped to inference, so reach for Ultralytics.

Open Source Computer Vision Library.
opencv/github.com/opencv/opencv-python /5,438 stars /
Ultralytics YOLO for object detection, segmentation, pose estimation, classification, and tracking.
Open-source differentiable computer vision library for PyTorch.
kornia/github.com/kornia/kornia /11,457 stars /
The open-source tool for building high-quality datasets and computer vision models.
voxel51/github.com/voxel51/fiftyone /11,250 stars /
A wrapper for the Tesseract OCR engine.
madmaze/github.com/madmaze/pytesseract /6,397 stars /
Ready-to-use OCR with 80+ languages supported.
JaidedAI/github.com/JaidedAI/EasyOCR /30,121 stars /
Multilingual OCR and document parsing toolkit based on PaddlePaddle.

Recommender Systems 3 projects

Match the Python recommender system library to your data: Surprise if users rate items, implicit if they only buy or watch. Annoy finds similar items fast.

Approximate Nearest Neighbors in C++/Python optimized for memory usage.
spotify/github.com/spotify/annoy /14,323 stars /
A fast Python implementation of collaborative filtering for implicit datasets.
benfred/github.com/benfred/implicit /3,833 stars /
A scikit for building and analyzing recommender systems.
NicolasHug/github.com/NicolasHug/Surprise /6,823 stars /

Know a project that belongs here?

Tell us what it does and why it stands out.