Open-source machine-learning software
5374 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.
5374 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| Lightning-AI/litgpt LitGPT is a Python library providing from-scratch, hackable implementations of 20+ open-source large language models with recipes for pretr… | 87 | 13718 | active |
| CompVis/stable-diffusion The original reference implementation of Stable Diffusion, a latent text-to-image diffusion model trained on LAION-5B data with a CLIP text… | 32 | 73487 | maintenance |
| microsoft/TRELLIS TRELLIS is Microsoft's large-scale 3D asset generation model that creates high-quality 3D assets from text or image prompts. It uses a unif… | 58 | 13679 | active |
| divamgupta/diffusionbee-stable-diffusion-ui DiffusionBee is a free macOS desktop application that runs Stable Diffusion locally with a one-click installer and no technical setup. It p… | 23 | 13603 | active |
| Cactus Needle 2 is a 45M-parameter open language model packaged as a single 14MB binary for tool calling, device use, and structured extraction on… | 60 | 13581 | active |
| QwenLM/Qwen3-TTS Qwen3-TTS is a series of open-source text-to-speech models from Alibaba's Qwen team, supporting expressive and streaming speech generation,… | 45 | 13545 | active |
| NVIDIA/TensorRT NVIDIA TensorRT is an SDK for high-performance deep learning inference on NVIDIA GPUs, comprising an inference compiler that converts train… | 93 | 13362 | stable |
| modelscope/DiffSynth-Studio DiffSynth-Studio is an open-source diffusion model engine from the ModelScope community that integrates mainstream image, video, and audio … | 77 | 13229 | active |
| YaoFANGUK/video-subtitle-remover An AI-based desktop application that removes hard-coded subtitles and text-like watermarks from videos and images using deep learning inpai… | 66 | 13131 | active |
| ACE-Step ACE-Step 1.5 is an open-source music generation foundation model combining a language model planner with a Diffusion Transformer to create … | 77 | 13070 | active |
| deepseek-ai/FlashMLA FlashMLA is DeepSeek's library of optimized CUDA attention kernels implementing Multi-head Latent Attention (MLA), including dense and toke… | 64 | 13063 | active |
| Vaibhavs10/insanely-fast-whisper A CLI tool for transcribing audio files on-device using OpenAI's Whisper models, powered by Hugging Face Transformers, Optimum, and Flash A… | 46 | 13063 | active |
| zai-org/CogVideo CogVideo/CogVideoX is an open-source family of text-to-video and image-to-video generation models from Zhipu AI (THUDM), with inference and… | 42 | 13055 | active |
| jacobgil/pytorch-grad-cam A PyTorch library providing state-of-the-art pixel attribution (saliency) methods like GradCAM, ScoreCAM, and AblationCAM for explainable A… | 72 | 13002 | active |
| PaddlePaddle/PaddleFormers PaddleFormers is a Transformers-style library built on PaddlePaddle providing a model zoo of 100+ large language models and vision-language… | 88 | 12983 | active |
| libfacedetection An open-source C++ library for CNN-based face detection in images, with the model embedded as static C source so it has no external depende… | 59 | 12807 | stable |
| Vision Transformer Google Research's official JAX/Flax implementation of Vision Transformer (ViT) and MLP-Mixer architectures, with released pretrained checkp… | 76 | 12744 | stable |
| Farama-Foundation/Gymnasium Gymnasium is a Python library providing a standard API for single-agent reinforcement learning environments, maintained by the Farama Found… | 95 | 12663 | stable |
| sapientinc/HRM Official PyTorch implementation of the Hierarchical Reasoning Model (HRM), a 27M-parameter recurrent architecture with high-level and low-l… | 49 | 12642 | active |
| bmaltais/kohya_ss A Gradio-based GUI and CLI wrapper around Kohya's Stable Diffusion training scripts for fine-tuning diffusion image generation models. It s… | 89 | 12616 | active |
| axolotl-ai-cloud/axolotl Axolotl is a free, open-source, config-driven framework for fine-tuning large language models, supporting SFT, preference learning (DPO/KTO… | 98 | 12591 | active |
| Tencent-Hunyuan/HunyuanVideo HunyuanVideo is Tencent's open-source framework for large-scale video generation, providing PyTorch model definitions, pre-trained weights,… | 59 | 12577 | active |
| datalab-to/chandra Chandra OCR 2 is a state-of-the-art open-weight OCR model from Datalab that converts images and PDFs into structured HTML, Markdown, or JSO… | 65 | 12487 | active |
| cupy/cupy CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python, running on NVIDIA CUDA or AMD ROCm. It acts as a … | 92 | 12362 | stable |
| DayBreak-u/chineseocr_lite An ultra-lightweight Chinese OCR toolkit combining DBNet text detection, CRNN text recognition, and an angle classifier, with total model s… | 66 | 12347 | active |
| guoyww/AnimateDiff Official implementation of AnimateDiff, a plug-and-play motion modeling module that turns personalized text-to-image diffusion models (e.g.… | 30 | 12274 | active |
| FlagOpen/FlagEmbedding FlagEmbedding is the official Python toolkit for BAAI's BGE family of embedding models and rerankers, covering inference, evaluation, and f… | 82 | 12242 | active |
| PKU-YuanGroup/Open-Sora-Plan Open-Sora Plan is an open-source effort to reproduce OpenAI's Sora text-to-video model, providing training and inference code for video gen… | 45 | 12209 | active |
| xmu-xiaoma666/External-Attention-pytorch A PyTorch library (fightingcv-attention) providing clean, minimal implementations of numerous attention mechanisms, MLP variants, re-parame… | 61 | 12186 | active |
| Autoware Autoware is the world's leading open-source, production-ready software stack for autonomous driving, built on ROS 2 and hosted by the Autow… | 90 | 12155 | stable |
| google/sentencepiece SentencePiece is a fast, lightweight unsupervised text tokenizer and detokenizer for neural network-based text generation systems, implemen… | 83 | 12132 | stable |
| ostris/ai-toolkit An all-in-one open-source training toolkit for finetuning diffusion models (image and video) on consumer-grade hardware. It supports many r… | 72 | 12107 | active |
| Tongyi-MAI/Z-Image Z-Image is a 6B-parameter text-to-image generation foundation model family built on a single-stream diffusion transformer, with a distilled… | 43 | 12057 | active |
| nerfstudio Nerfstudio is a Python library and CLI toolkit providing a simple, modular API for creating, training, and testing Neural Radiance Fields (… | 34 | 12019 | active |
| InstantID InstantID is a tuning-free, zero-shot identity-preserving image generation method built on diffusion models, generating customized images i… | 27 | 11994 | active |
| facebookresearch/sam3 Official code for Meta's Segment Anything Model 3 (SAM 3), a unified foundation model for promptable segmentation in images and videos. It … | 63 | 11957 | active |
| NVIDIA/cosmos NVIDIA Cosmos is an open platform of omnimodal world foundation models, datasets, and tools for building Physical AI systems such as robots… | 69 | 11892 | active |
| facebookresearch/seamless_communication A library of foundational multilingual multimodal AI models from Meta for speech and text translation, including SeamlessM4T, SeamlessExpre… | 71 | 11889 | active |
| SpeechBrain SpeechBrain is an open-source PyTorch-based speech toolkit for building conversational AI systems. It provides training recipes, pretrained… | 85 | 11845 | active |
| ludwig-ai/ludwig Ludwig is a declarative, low-code deep learning framework for training, fine-tuning, and deploying AI models — from LLMs to tabular, image,… | 96 | 11775 | active |
| qubvel-org/segmentation_models.pytorch A PyTorch library providing neural networks for image semantic segmentation with a simple high-level API. It includes 12 encoder-decoder ar… | 68 | 11745 | stable |
| cleanlab Cleanlab is a Python library for data-centric AI that automatically detects issues in ML datasets, such as label errors, outliers, duplicat… | 56 | 11692 | stable |
| milesial/Pytorch-UNet A PyTorch implementation of the U-Net architecture for semantic segmentation of high-resolution images, originally built for Kaggle's Carva… | 23 | 11668 | active |
| facebookresearch/dinov3 Reference PyTorch implementation and pretrained models for DINOv3, Meta's self-supervised vision transformer backbone family. It includes t… | 56 | 11536 | active |
| rerun-io/rerun Rerun is an open-source SDK and viewer for logging, storing, querying, and visualizing multi-rate multimodal data such as images, point clo… | 96 | 11498 | active |
| Kornia Kornia is a differentiable computer vision library built on PyTorch, offering GPU-accelerated image processing, augmentations, and geometri… | 84 | 11436 | active |
| THU-MIG/yolov10 YOLOv10 is a real-time end-to-end object detection model family that removes NMS post-processing via consistent dual assignments and optimi… | 17 | 11331 | active |
| Weights & Biases Weights & Biases (wandb) is a Python SDK and platform for tracking, visualizing, and managing machine learning experiments, including metri… | 100 | 11303 | active |
| microsoft/TRELLIS.2 TRELLIS.2 is a 4B-parameter state-of-the-art 3D generative model from Microsoft for high-fidelity image-to-3D asset generation. It uses a f… | 54 | 11284 | active |
| salesforce/LAVIS LAVIS is a Python library from Salesforce AI Research providing a unified toolkit for language-vision (multimodal) intelligence, including … | 10 | 11264 | active |
| MiniCPM MiniCPM is a family of small, state-of-the-art on-device language models from OpenBMB, with MiniCPM5-1B being a dense 1B Transformer for lo… | 74 | 11232 | active |
| huggingface/tokenizers Hugging Face Tokenizers is a fast, Rust-based library implementing state-of-the-art tokenization algorithms (BPE, WordPiece, Unigram) with … | 95 | 11181 | stable |
| Numba Numba is an open-source NumPy-aware JIT compiler that translates a subset of Python and NumPy code into fast machine code using LLVM. It su… | 94 | 11175 | stable |
| pathwaycom/arc-task-gen A Python tool that generates original ARC-AGI-1-style puzzle tasks distribution-matched to the public evaluation set. It produces tasks in … | 53 | 11149 | active |
| LTX-Video Official repository for LTX-Video, a DiT-based open-weights video generation model from Lightricks that generates high-fidelity video (up t… | 45 | 11138 | active |
| LTX-Video Official Python package from Lightricks providing inference pipelines and LoRA training for LTX-2/LTX-2.5, an open-weights DiT-based founda… | 45 | 11138 | active |
| Triton Inference Server NVIDIA Triton Inference Server is an open-source inference serving software that deploys AI models from multiple frameworks (TensorRT, PyTo… | 98 | 11075 | stable |
| thu-ml/tianshou Tianshou is a modular, high-performance deep reinforcement learning library built on pure PyTorch and Gymnasium. It offers both low-level h… | 66 | 11000 | active |
| NopeCHA NopeCHA is an AI-powered CAPTCHA solving service distributed as a browser extension (Chrome/Firefox/Edge) plus Python and Node.js client li… | 87 | 10997 | active |
| face_recognition A Python library and command-line tool providing a simple API for face detection, facial landmark extraction, and face recognition, built o… | 59 | 56796 | maintenance |
| google/dopamine Dopamine is a research framework from Google for fast prototyping of reinforcement learning algorithms, built around a small, easily readab… | 52 | 10925 | active |
| Natural Natural is a general natural language processing library for Node.js offering tokenizing, stemming, part-of-speech tagging, sentiment analy… | 58 | 10881 | active |
| karpathy/minbpe A minimal, clean Python implementation of the byte-level Byte Pair Encoding (BPE) algorithm used for tokenization in modern LLMs like GPT-4… | 26 | 10830 | stable |
| multimodal-art-projection/YuE YuE is a family of open-source foundation models based on the LLaMA2 architecture that generate full songs (up to five minutes) with vocals… | 87 | 10822 | active |
| cumulo-autumn/StreamDiffusion StreamDiffusion is a Python pipeline for real-time interactive diffusion-based image generation, achieving 100+ fps on modern GPUs. It opti… | 18 | 10817 | active |
| doccano Doccano is an open-source, web-based text annotation tool for building labeled datasets for machine learning. It supports text classificati… | 60 | 10794 | active |
| Tabnine Tabnine is an AI code completion and coding assistant that provides all-language autocompletion across major IDEs (VS Code, JetBrains, Subl… | 10 | 10767 | active |
| autogluon/autogluon AutoGluon is an AutoML library that automates machine learning on tabular data, time series, text, and images with just a few lines of Pyth… | 93 | 10766 | active |
| lucidrains/denoising-diffusion-pytorch A PyTorch implementation of Denoising Diffusion Probabilistic Models (DDPM) for generative image modeling, published on PyPI as denoising_d… | 86 | 10707 | active |
| IDEA-Research/GroundingDINO Official PyTorch implementation of Grounding DINO, an open-set object detector that fuses a Transformer-based DINO detector with grounded v… | 22 | 10682 | stable |
| Megvii-BaseDetection/YOLOX YOLOX is a high-performance anchor-free YOLO object detection model family implemented in PyTorch, with pretrained weights and export suppo… | 30 | 10628 | stable |
| bigscience-workshop/petals Petals is a Python library that lets you run and fine-tune large language models (Llama 3.1, Mixtral, Falcon, BLOOM) on a BitTorrent-style … | 23 | 10614 | active |
| pyannote.audio pyannote.audio is an open-source Python toolkit built on PyTorch for speaker diarization, providing neural building blocks like voice activ… | 92 | 10593 | active |
| NVIDIA/cutlass CUTLASS is NVIDIA's collection of CUDA C++ template abstractions and Python DSLs for implementing high-performance GEMM and related linear … | 98 | 10580 | active |
| facebookresearch/xformers xFormers is a PyTorch-based library of hackable, optimized Transformer building blocks with custom CUDA kernels for fast, memory-efficient … | 87 | 10565 | active |
| vwxyzjn/cleanrl CleanRL is a deep reinforcement learning library providing high-quality, single-file implementations of algorithms like PPO, DQN, DDPG, TD3… | 54 | 10481 | active |
| zyddnys/manga-image-translator A Python tool that automatically detects, OCRs, translates, inpaints, and re-typesets text in images, primarily for manga and comics. It ru… | 66 | 10477 | active |
| lllyasviel/Fooocus Fooocus is an offline, open-source image generation application built on Stable Diffusion XL with a Gradio interface. It simplifies text-to… | 41 | 53616 | maintenance |
| niedev/RTranslator RTranslator is a free, open-source, offline real-time translation app for Android that runs speech recognition (Whisper) and translation (M… | 77 | 10449 | active |
| CyberTimon/RapidRAW RapidRAW is a free, open-source, non-destructive RAW photo editor and image library manager built with Rust, wgpu, React, and Tauri. It off… | 82 | 10446 | active |
| FreeMoCap FreeMoCap is a free, open-source, markerless motion capture system that uses ordinary cameras (webcams, GoPros, smartphones) to record and … | 98 | 10437 | active |
| snakers4/silero-vad Silero VAD is a pre-trained, enterprise-grade Voice Activity Detector model available via PyPI, runnable with PyTorch or ONNX Runtime. It d… | 93 | 10317 | stable |
| open-mmlab/Amphion Amphion is an open-source Python toolkit for audio, music, and speech generation, supporting tasks like text-to-speech, voice conversion, s… | 48 | 10304 | active |
| Metaflow Metaflow is a human-centric Python framework from Netflix for building, managing, and deploying real-life AI/ML and data science systems. I… | 92 | 10280 | stable |
| deepseek-ai/DeepEP DeepEP is a high-performance GPU communication library for expert parallelism (EP) in MoE training and inference, providing high-throughput… | 55 | 10263 | active |
| xai-org/grok-1 xAI's open release of the Grok-1 open-weights model (314B-parameter Mixture-of-Experts LLM) with JAX example code for loading and running i… | 26 | 52232 | maintenance |
| InternVL InternVL is a family of open-source multimodal large language models (vision-language models) that combine vision transformers with LLMs to… | 34 | 10163 | active |
| Moondream Moondream is an open-weight family of small, efficient vision language models (2B to 9B MoE) that perform image captioning, visual question… | 58 | 10127 | active |
| stanfordnlp/CoreNLP Stanford CoreNLP is a Java suite of natural language processing tools that annotates raw text with tokenization, POS tags, named entities, … | 89 | 10122 | stable |
| TencentARC/PhotoMaker PhotoMaker is a personalized text-to-image generation method that encodes multiple reference face photos into a stacked ID embedding to gen… | 27 | 10088 | stable |
| sktime sktime is a unified Python framework for machine learning with time series, offering 500+ models behind a single scikit-learn-compatible AP… | 98 | 10062 | stable |
| TPOT TPOT (Tree-based Pipeline Optimization Tool) is a Python automated machine learning library that optimizes scikit-learn machine learning pi… | 40 | 10053 | active |
| OpenRLHF/OpenRLHF OpenRLHF is a high-performance, production-ready open-source RLHF framework built on Ray + vLLM + DeepSpeed for scalable reinforcement lear… | 89 | 10040 | active |
| OpenMined/PySyft PySyft is a Python library that lets data scientists run computations on private data that stays on the data owner's server, with results s… | 73 | 10034 | active |
| yzhao062/pyod PyOD is the most comprehensive Python library for anomaly detection, offering 60+ detectors across tabular, time series, graph, text, image… | 96 | 10026 | stable |
| opendatalab/PDF-Extract-Kit PDF-Extract-Kit is a Python model toolbox for high-quality PDF content extraction, integrating state-of-the-art models for layout detection… | 15 | 10023 | active |
| ESPnet ESPnet is an end-to-end speech processing toolkit built on PyTorch covering speech recognition, text-to-speech, speech translation, enhance… | 94 | 9989 | active |
| facebookresearch/pytorch3d PyTorch3D is Facebook AI Research's library of efficient, reusable components for deep learning with 3D data, built on PyTorch. It provides… | 70 | 9983 | active |
| open-mmlab/mmsegmentation MMSegmentation is a PyTorch-based toolbox and benchmark for semantic segmentation, part of the OpenMMLab ecosystem. It provides implementat… | 23 | 9956 | stable |
| kyutai-labs/pocket-tts Pocket TTS is a lightweight 100M-parameter text-to-speech model and Python library from Kyutai that runs efficiently on CPUs without a GPU.… | 83 | 9923 | active |