Open-source machine-learning software

4007 open-source libraries, tools and applications tagged machine-learning, ranked by how closely machine-learning is their primary purpose, then by adoption and maintenance health. Each entry links to a page with sources, use cases and when to choose or avoid it.

4007 products, primary matches first, then adoption-weighted; health v2 shown.

Product	Health v2	Stars	Maturity
Deep Lake Deep Lake is an open-source database for AI that stores multimodal data (images, video, audio, text, embeddings, annotations) in a format o…	75	9249	active
Pwnagotchi Pwnagotchi is an A2C deep reinforcement learning agent built on bettercap that learns from its surrounding WiFi environment to maximize cap…	63	9223	active
coqui-ai/TTS Coqui TTS is a deep learning toolkit for text-to-speech synthesis, providing pretrained models in over 1100 languages plus tools for traini…	23	46107	maintenance
modelscope/modelscope ModelScope is a Python library and ecosystem built on the 'Model-as-a-Service' concept, providing unified APIs to download, run inference o…	96	9149	active
CatBoost CatBoost is an open-source gradient boosting on decision trees library developed by Yandex, offering high-quality classification, regressio…	82	9133	stable
pyro-ppl/pyro Pyro is a deep universal probabilistic programming library built on Python and PyTorch, supporting Bayesian modeling with variational infer…	65	9059	stable
FareedKhan-dev/kimi-k3-in-c A dependency-free C99 inference engine that runs the 2.78-trillion-parameter Kimi K3 model on a single CPU with as little as 8 GB of RAM by…	79	9033	active
NSFWJS NSFWJS is a JavaScript library that uses TensorFlow.js to classify images into NSFW/safety categories (Drawing, Neutral, Sexy, Hentai, Porn…	80	9007	active
NVIDIA/apex NVIDIA-maintained PyTorch extension providing utilities for easy mixed precision and distributed training. It offers up-to-date CUDA and C+…	75	9000	active
MIC-DKFZ/nnUNet nnU-Net is a self-configuring deep learning framework for semantic image segmentation that automatically adapts preprocessing, U-Net archit…	65	8982	stable
DepthAnything/Depth-Anything-V2 Depth Anything V2 is a foundation model for monocular depth estimation, trained on 595K synthetic labeled images and 62M+ real unlabeled im…	54	8940	stable
LykosAI/StabilityMatrix Stability Matrix is a multi-platform desktop application that acts as a package manager for Stable Diffusion UIs and tools like ComfyUI, AU…	96	8932	active
apple/ml-sharp SHARP is a Python tool from Apple that synthesizes a photorealistic 3D Gaussian splat representation from a single photograph in under a se…	59	8931	active
Rig Rig is a Rust library for building modular, scalable LLM-powered applications and agents, offering a unified API across 20+ model providers…	87	8911	active
BentoML BentoML is a Python framework for building online serving systems for AI apps and model inference, turning model inference scripts into RES…	91	8909	stable
sebastianstarke/AI4Animation AI4Animation is a deep learning framework for data-driven character animation and control, built around Unity with a Python remake (AI4Anim…	63	8873	active
AI on the Edge Device A firmware application for ESP32-CAM boards that uses TensorFlow Lite CNNs on-device to digitize analog utility meters (water, gas, electri…	74	8865	active
elder-plinius/OBLITERATUS OBLITERATUS is a Python toolkit for abliteration — identifying and removing refusal behaviors from large language models by locating and in…	79	8861	active
MONAI MONAI is a PyTorch-based open-source framework for deep learning in healthcare imaging, providing domain-specific transforms, 3D architectu…	91	8788	stable
VowpalWabbit/vowpal_wabbit Vowpal Wabbit is a fast, open-source machine learning system focused on online, active, and interactive learning, including contextual band…	98	8727	active
bayesian-optimization/BayesianOptimization A pure Python library implementing Bayesian global optimization using Gaussian processes. It finds the maximum of expensive, unknown black-…	94	8720	stable
librosa/librosa Librosa is a Python library for audio and music signal processing, providing foundational algorithms for building music information retriev…	84	8690	stable
Jesse Jesse is an open-source Python framework for researching, backtesting, optimizing, and live-trading cryptocurrency trading strategies. It i…	77	8660	active
google-deepmind/alphafold3 Google DeepMind's official implementation of the AlphaFold 3 inference pipeline for predicting biomolecular structures and interactions. It…	81	8643	active
DeepDive slime is an open-source LLM post-training framework for reinforcement learning scaling, connecting Megatron-based training with SGLang-base…	79	8606	active
bitsandbytes-foundation/bitsandbytes bitsandbytes is a Python library providing k-bit quantization primitives for PyTorch, enabling 8-bit (LLM.int8()) and 4-bit (QLoRA) quantiz…	94	8514	active
OptimalScale/LMFlow LMFlow is an extensible Python toolkit for finetuning and inference of large foundation models such as LLaMA, GPT-2, and Galactica. It prov…	61	8487	active
qualcomm/GenieX GenieX is an on-device generative AI inference runtime for Qualcomm Snapdragon devices, running frontier LLMs and VLMs on the Hexagon NPU, …	87	8452	active
lucidrains/imagen-pytorch A PyTorch implementation of Imagen, Google's text-to-image neural network based on cascading DDPMs conditioned on T5 text embeddings. It pr…	23	8428	active
CASIA-LMC-Lab/FastSAM FastSAM is a CNN-based Segment Anything Model trained on only 2% of the SA-1B dataset, achieving comparable segmentation performance to SAM…	20	8421	active
ASRT SpeechRecognition ASRT is a deep-learning-based Chinese speech recognition (speech-to-text) system built with TensorFlow/Keras, using CNN, LSTM, attention me…	53	8391	active
py-why/dowhy DoWhy is a Python library for causal inference that supports explicit modeling and testing of causal assumptions. It unifies causal graphic…	72	8335	stable
mikel-brostrom/boxmot BoxMOT is a pluggable Python and C++ library providing state-of-the-art multi-object tracking (MOT) algorithms such as ByteTrack, BoT-SORT,…	91	8319	active
lmcinnes/umap UMAP is a Python library implementing Uniform Manifold Approximation and Projection, a non-linear dimensionality reduction technique usable…	86	8281	stable
jessevig/bertviz BertViz is an interactive Python library for visualizing attention mechanisms in Transformer language models such as BERT, GPT-2, and RoBER…	47	8200	stable
asg017/sqlite-vec sqlite-vec is a vector search extension for SQLite written in pure C with no dependencies, supporting float, int8, and binary vectors in ve…	71	8168	active
Pinokio Pinokio is a desktop application that acts as a one-click launcher and browser for open-source AI apps, letting users discover, install, ru…	91	8154	active
NVIDIA/Isaac-GR00T NVIDIA Isaac GR00T N1.7 is an open vision-language-action (VLA) foundation model for generalized humanoid robot skills, taking language and…	70	8116	stable
LibrePhotos LibrePhotos is a self-hosted, open-source photo management service built with Django and React. It offers AI-powered features like face rec…	100	8112	active
InternLM/lmdeploy LMDeploy is a toolkit for compressing, quantizing, deploying, and serving large language models, built around its high-performance TurboMin…	96	8111	active
nadermx/backgroundremover A free, open-source command line tool that removes backgrounds from images and videos using the U2Net AI model built on PyTorch. It is inst…	87	8061	active
shenweichen/DeepCTR DeepCTR is a Python library of easy-to-use, modular, and extendible deep-learning based CTR (click-through rate) prediction models built on…	71	8051	stable
PriorLabs/TabPFN TabPFN is a pre-trained tabular foundation model (transformer) that performs classification, regression, forecasting, and anomaly detection…	96	8048	active
MLX-Audio MLX-Audio is a Python library built on Apple's MLX framework for fast text-to-speech (TTS), speech-to-text (STT), and speech-to-speech (STS…	88	8012	active
alirezamika/autoscraper AutoScraper is a Python library that automatically learns scraping rules from a URL or HTML content plus a list of sample data you want to …	62	8003	stable
lanpa/tensorboardX A Python library for writing TensorBoard event files from PyTorch, Chainer, MXNet, NumPy, and other frameworks without needing TensorFlow. …	71	8002	active
evidentlyai/evidently Evidently is an open-source Python framework for evaluating, testing, and monitoring ML and LLM-powered systems, with 100+ built-in metrics…	97	7984	active
MochiDiffusion/MochiDiffusion A native macOS app built with SwiftUI for running Stable Diffusion and FLUX.2 Klein image generation locally on Apple Silicon Macs. It uses…	86	7957	active
Upsonic/Upsonic Upsonic is a Python framework for building autonomous and traditional AI agents, with support for tools, memory, knowledge bases, RAG, mult…	71	7956	active
open-mmlab/mmpose MMPose is an open-source pose estimation toolbox and benchmark built on PyTorch as part of the OpenMMLab ecosystem. It provides implementat…	35	7921	active
TheLastBen/fast-stable-diffusion A collection of Google Colab notebooks for quickly running Stable Diffusion UIs (AUTOMATIC1111, ComfyUI) and training DreamBooth models for…	53	7911	active
MaartenGr/BERTopic BERTopic is a Python library for topic modeling that leverages transformer-based sentence embeddings and class-based TF-IDF (c-TF-IDF) to d…	75	7902	stable
stanfordnlp/stanza Stanza is the Stanford NLP Group's official Python library for linguistic analysis of human language text. It provides a neural pipeline bu…	94	7886	stable
TensorRTx A C++ collection of popular deep learning networks (YOLO variants, ResNet, MobileNet, Swin Transformer, OCR models, and more) implemented f…	73	7838	active
Flyte Flyte is an open-source workflow orchestration platform for coordinating data, ML models, and AI agents at scale, authored in pure Python a…	96	7741	active
leejet/stable-diffusion.cpp A pure C/C++ inference engine for diffusion models (Stable Diffusion, FLUX, Wan, Qwen Image, Z-Image, and more) built on ggml in the style …	92	7734	active
google-deepmind/weathernext Google DeepMind's WeatherNext family of AI weather forecasting models, including WeatherNext 2, GraphCast, and GenCast, with code and pretr…	82	7709	active
alteryx/featuretools Featuretools is an open-source Python library for automated feature engineering, using Deep Feature Synthesis (DFS) to generate features fr…	65	7687	active
openai/gym Gym is a Python toolkit for developing and comparing reinforcement learning algorithms via a standard API between learning algorithms and e…	10	37236	maintenance
myshell-ai/MeloTTS MeloTTS is a high-quality multi-lingual text-to-speech library supporting English (multiple accents), Spanish, French, Chinese, Japanese, a…	17	7644	active
Kaggle/kaggle-cli The official Kaggle command-line interface for interacting with Kaggle competitions, datasets, models, notebooks, and discussion forums. It…	95	7607	active
OpenMathLib/OpenBLAS OpenBLAS is an optimized open-source implementation of the BLAS (Basic Linear Algebra Subprograms) library, based on GotoBLAS2, with LAPACK…	89	7587	stable
HIPS/autograd Autograd is a Python library that efficiently computes derivatives of native Python and NumPy code, supporting reverse-mode (backpropagatio…	82	7543	stable
H2O H2O-3 is an open-source, distributed, in-memory machine learning platform implementing algorithms such as GLM, GBM/XGBoost, Random Forest, …	75	7509	stable
hybridgroup/gocv GoCV is a Go language binding for the OpenCV 4 computer vision library, supporting Linux, macOS, Windows, and Docker. It includes support f…	66	7505	active
open-compass/opencompass OpenCompass is a Python-based evaluation platform and toolkit for benchmarking large language models across 100+ datasets, supporting model…	93	7471	active
EleutherAI/gpt-neox GPT-NeoX is EleutherAI's library for training large-scale autoregressive transformer language models on GPUs, built on NVIDIA's Megatron an…	67	7471	active
facebookresearch/SlowFast PySlowFast is a PyTorch-based open-source video understanding codebase from Facebook AI Research (FAIR). It provides implementations of sta…	61	7432	active
LargeWorldModel/LWM Large World Model (LWM) is a family of open-source 7B-parameter multimodal autoregressive transformer models trained on long videos and boo…	26	7425	active
apple/ml-fastvlm Official implementation of FastVLM, a vision language model with an efficient hybrid vision encoder (FastViTHD) that reduces token count an…	62	7423	active
arcee-ai/mergekit mergekit is a Python toolkit for merging pre-trained large language models directly in weight space, supporting many merge methods (SLERP, …	64	7389	active
google/flax Flax is a neural network library and ecosystem for JAX designed for flexibility, featuring the newer NNX API with first-class Python refere…	99	7348	active
Feast Feast is an open-source feature store for machine learning that manages offline stores for historical training data and low-latency online …	99	7327	stable
naver/dust3r DUSt3R is the official PyTorch implementation of a CVPR 2024 model that performs dense, unconstrained stereo and multi-view 3D reconstructi…	42	7317	active
vLLM vLLM-Omni is a Python framework extending vLLM for efficient inference and serving of omni-modality models, including diffusion transformer…	84	7293	active
ml-explore/mlx-lm A Python package for running and fine-tuning large language models on Apple silicon using the MLX framework. It integrates with the Hugging…	78	7259	active
Alluxio/alluxio Alluxio is an open-source distributed caching and data orchestration platform that sits between compute frameworks (Spark, Presto, Trino, P…	64	7248	stable
liuliu/ccv ccv is a modern, minimalist computer vision library written in C/C++ with an application-driven set of state-of-the-art algorithms includin…	77	7247	active
kohya-ss/sd-scripts A collection of Python training, generation, and utility scripts for Stable Diffusion and other image generation models, most widely used f…	96	7242	active
TensorBoard TensorBoard is TensorFlow's visualization toolkit, a suite of web applications for inspecting and understanding machine learning training r…	82	7231	active
stanfordnlp/GloVe Stanford's C implementation of the GloVe model for training distributed word representations (word embeddings), plus downloadable pre-train…	34	7230	stable
AI Scientist An end-to-end autonomous AI research system that generates hypotheses, runs machine learning experiments, analyzes data, and writes complet…	43	7222	active
NVIDIA/warp NVIDIA Warp is a Python framework that JIT-compiles regular Python functions into efficient CPU or CUDA GPU kernels. It provides differenti…	99	7215	active
deepseek-ai/DeepSpec DeepSpec is a full-stack Python codebase from DeepSeek for training and evaluating draft models used in speculative decoding of large langu…	51	7207	active
BVLC/caffe Caffe is a fast, modular deep learning framework developed by Berkeley AI Research, focused on convolutional neural networks for vision, sp…	23	34549	maintenance
OpenPose OpenPose is a real-time multi-person keypoint detection library that jointly estimates human body, face, hand, and foot keypoints (135 tota…	23	34493	maintenance
tensortrade-org/tensortrade TensorTrade is an open-source Python framework for building, training, and evaluating reinforcement learning agents for algorithmic trading…	58	7194	active
Background Matting V2 Official PyTorch implementation of the CVPR 2021 paper 'Real-Time High-Resolution Background Matting'. It produces state-of-the-art alpha m…	23	7193	stable
Lance Lance is an open lakehouse data format (file format, table format, and catalog spec) for multimodal AI, implemented in Rust with Python and…	95	7182	active
flwrlabs/flower Flower (flwr) is an open-source Python framework for building federated and collaborative AI systems, supporting any ML framework such as P…	99	7162	active
scikit-learn-contrib/imbalanced-learn imbalanced-learn is a Python library offering re-sampling techniques (over-sampling, under-sampling, and combinations) for datasets with st…	76	7120	stable
yangchris11/samurai SAMURAI is the official implementation of a zero-shot visual object tracker built on top of Segment Anything Model 2 (SAM 2), using a motio…	24	7116	active
ymcui/Chinese-LLaMA-Alpaca-2 A project releasing Chinese-adapted LLaMA-2 base models and Alpaca-2 instruction-tuned chat models with an extended Chinese vocabulary, inc…	51	7112	stable
zai-org/GLM-4 Official repository for the GLM-4 series of open-weight, multilingual (primarily Chinese/English) multimodal chat language models from Z.ai…	64	7069	active
google/gemma.cpp A lightweight, standalone C++ inference engine for Google's Gemma foundation models (Gemma 2/3, PaliGemma 2), with a small ~2K LoC core and…	73	7060	active
DeepChem DeepChem is a Python library that democratizes deep learning for drug discovery, quantum chemistry, materials science, and biology. It prov…	63	7050	active
TheTom/turboquant_plus TurboQuant+ is a Python reference implementation of the TurboQuant KV cache compression method (ICLR 2026), using PolarQuant codebooks and …	54	7036	active
apple/corenet CoreNet is Apple's deep neural network training toolkit for training standard and novel small and large-scale models, including foundation …	68	7008	active
gepa-ai/gepa GEPA is a Python framework that optimizes textual system parameters such as prompts, code, and agent configurations using LLM-based reflect…	81	6996	active
deeppavlov/DeepPavlov DeepPavlov is an open-source Python NLP library built on PyTorch and Hugging Face transformers for developing, training, and deploying stat…	35	6995	active