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 |
|---|---|---|---|
| huggingface/accelerate Hugging Face Accelerate is a Python library that lets you run the same PyTorch training and inference code on any device or distributed con… | 94 | 9916 | stable |
| gorse-io/gorse Gorse is an AI-powered open-source recommender system engine written in Go that ingests items, users, and interaction feedback and automati… | 95 | 9876 | active |
| PyCaret PyCaret is an open-source, low-code AutoML library for Python that wraps scikit-learn to automate training, tuning, and comparison of model… | 79 | 9846 | active |
| yihong0618/bilingual_book_maker A Python CLI tool that uses AI translation services (ChatGPT, Claude, Gemini, DeepL, and others) to create bilingual versions of epub, txt,… | 96 | 9825 | active |
| Tiiny-AI/PowerInfer PowerInfer is a CPU/GPU hybrid LLM inference engine that exploits activation locality to run large language models fast on consumer-grade h… | 60 | 9814 | active |
| PyMC PyMC is a Python library for Bayesian statistical modeling and probabilistic programming, built on PyTensor. It provides intuitive model sp… | 96 | 9774 | stable |
| roboflow/rf-detr RF-DETR is a real-time transformer-based model architecture from Roboflow for object detection, instance segmentation, and keypoint detecti… | 86 | 9678 | active |
| francescopace/espectre ESPectre is an open-source motion detection system that analyzes Wi-Fi Channel State Information (CSI) on ESP32 devices to detect movement … | 84 | 9636 | active |
| arogozhnikov/einops einops is a Python library providing readable, framework-agnostic tensor operations via mini-language functions like rearrange, reduce, and… | 71 | 9615 | stable |
| zhouxiaoka/autoclip AutoClip is a self-hosted AI-powered video clipping system that downloads videos from YouTube and Bilibili, uses LLM-based content analysis… | 86 | 9577 | active |
| WongKinYiu/yolov9 Official PyTorch implementation of the YOLOv9 object detection paper, featuring Programmable Gradient Information for improved accuracy. It… | 17 | 9562 | active |
| YaoFANGUK/video-subtitle-extractor A GUI application that extracts hard-coded (burned-in) subtitles from videos and generates SRT subtitle files using local deep-learning-bas… | 64 | 9555 | active |
| sloria/TextBlob TextBlob is a Python library providing a simple, consistent API for common NLP tasks such as part-of-speech tagging, noun phrase extraction… | 67 | 9551 | stable |
| unit8co/darts Darts is a Python library for user-friendly time series forecasting and anomaly detection. It provides a unified scikit-learn-like fit/pred… | 92 | 9533 | active |
| PeterL1n/RobustVideoMatting Robust Video Matting (RVM) is a deep learning model and library for real-time human video matting, using a recurrent neural network with te… | 23 | 9533 | stable |
| modelscope/facechain FaceChain is a deep-learning toolchain from ModelScope for generating identity-preserved personal portraits (digital twins) from a single p… | 27 | 9508 | active |
| MakazhanAlpamys/Soup Soup is a Python CLI that fine-tunes and post-trains LLMs from a single YAML config, supporting 23 methods (SFT, DPO, ORPO, SimPO, KTO, etc… | 79 | 9486 | active |
| Cog Cog is an open-source CLI tool that packages machine learning models into production-ready Docker containers using a simple cog.yaml config… | 97 | 9484 | active |
| blue-yonder/tsfresh tsfresh is a Python package that automatically extracts hundreds of features from time series using algorithms from statistics, signal proc… | 74 | 9470 | stable |
| OneFlow OneFlow is an open-source deep learning framework written in C++ with a PyTorch-like Python API, focused on scalable and efficient distribu… | 44 | 9437 | active |
| oumi-ai/oumi Oumi is an open-source Python framework and platform for the end-to-end lifecycle of open-weight LLMs: data synthesis, fine-tuning (SFT, Lo… | 91 | 9407 | active |
| PaddlePaddle/PaddleSeg PaddleSeg is an end-to-end image segmentation toolkit built on PaddlePaddle, offering a model zoo with dozens of pre-trained models for sem… | 49 | 9395 | active |
| QwenAudio/SenseVoice SenseVoice is an open-source speech foundation model (SenseVoiceSmall) providing multilingual ASR, spoken language identification, speech e… | 87 | 9386 | active |
| ML.NET ML.NET is an open-source, cross-platform machine learning framework for .NET that lets developers build, train, and deploy custom ML models… | 78 | 9356 | active |
| Kokoro TTS An inference library for Kokoro-82M, an open-weight text-to-speech model with 82 million parameters that delivers quality comparable to lar… | 34 | 9340 | active |
| keras-team/autokeras AutoKeras is an AutoML library for deep learning built on Keras, developed by DATA Lab at Texas A&M University. It automates model architec… | 46 | 9336 | active |
| bytedance/monolith Monolith is a deep learning framework built on TensorFlow for large-scale recommendation modeling. It provides collisionless embedding tabl… | 10 | 9286 | active |
| X-PLUG/MobileAgent Mobile-Agent is a family of multimodal GUI agents from Tongyi Lab that perceive and operate mobile and desktop user interfaces using vision… | 64 | 9269 | 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 |
| polakowo/vectorbt VectorBT is a Python library for backtesting and analyzing trading strategies at scale, operating on pandas and NumPy objects accelerated b… | 88 | 9175 | active |
| studio-dots-ai/dots.ocr dots.ocr is a 1.7B-parameter vision-language model for multilingual document layout parsing, converting documents into structured output wi… | 48 | 9157 | active |
| 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 |
| NVlabs/Sana SANA is an efficiency-oriented PyTorch codebase for high-resolution text-to-image and text-to-video generation built on Linear Diffusion Tr… | 70 | 9129 | active |
| 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 |
| 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 |
| deepseek-ai/DeepGEMM DeepGEMM is a high-performance CUDA BLAS kernel library for NVIDIA tensor cores, providing FP8, FP4, and BF16 GEMMs plus fused MoE and othe… | 82 | 8937 | active |
| 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 |
| 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 |
| MIC-DKFZ/nnUNet nnU-Net is a self-configuring deep learning framework for semantic image segmentation that automatically adapts preprocessing, U-Net archit… | 65 | 8901 | 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 |
| FoundationVision/VAR Official PyTorch implementation of Visual Autoregressive Modeling (VAR), a NeurIPS 2024 Best Paper-winning method for scalable image genera… | 45 | 8751 | active |
| 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 |
| fudan-generative-vision/hallo Hallo is a Python research library implementing hierarchical audio-driven visual synthesis for animating portrait images into talking-head … | 15 | 8667 | active |
| 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 |
| lllyasviel/IC-Light IC-Light is a Python tool for manipulating the illumination of images using diffusion models, offering text-conditioned and background-cond… | 25 | 8551 | active |
| netease-youdao/EmotiVoice EmotiVoice is an open-source text-to-speech engine supporting English and Chinese with over 2000 voices and prompt-controlled emotional syn… | 60 | 8529 | 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 |
| Gonum Gonum is a suite of numerical and scientific computing libraries for Go, covering matrices and linear algebra, statistics and probability, … | 73 | 8441 | 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 |
| isaac-sim/IsaacLab Isaac Lab is a GPU-accelerated open-source framework for robot learning built on NVIDIA Isaac Sim, unifying workflows like reinforcement le… | 96 | 8397 | 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 |
| XPixelGroup/BasicSR BasicSR is an open-source PyTorch toolbox for image and video restoration tasks such as super-resolution, denoising, deblurring, and JPEG a… | 23 | 8391 | stable |
| QwenLM/Qwen-Image Qwen-Image is a 20B MMDiT image generation foundation model from the Qwen team, with strong complex text rendering (especially Chinese) and… | 45 | 8377 | active |
| meituan-longcat/LongCat-Video LongCat-Video is a 13.6B-parameter foundational video generation model from Meituan that unifies text-to-video, image-to-video, and video-c… | 51 | 8358 | active |
| CompreFace Exadel CompreFace is a free, open-source face recognition system that provides REST APIs for face recognition, verification, detection, lan… | 23 | 8344 | stable |
| 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 |
| zai-org/ChatGLM-6B ChatGLM-6B is an open-source bilingual (Chinese-English) 6.2B-parameter dialogue language model based on the GLM architecture, optimized fo… | 31 | 40938 | maintenance |
| 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 | 8301 | active |
| lmcinnes/umap UMAP is a Python library implementing Uniform Manifold Approximation and Projection, a non-linear dimensionality reduction technique usable… | 86 | 8281 | stable |
| lightningpixel/modly Modly is an open-source desktop application that generates textured 3D meshes from images or text prompts using open-source AI models (Huny… | 82 | 8255 | active |
| Ucas-HaoranWei/GOT-OCR2.0 Official implementation of GOT-OCR2.0, a unified end-to-end vision-language model for general OCR that converts images of text, documents, … | 23 | 8228 | active |
| carson-katri/dream-textures A Blender add-on that integrates Stable Diffusion for generating textures, concept art, and background assets directly inside Blender. It s… | 66 | 8213 | active |
| lltcggie/waifu2x-caffe A Windows GUI/CLI application that reimplements the waifu2x image upscaling and noise-reduction tool using the Caffe deep learning framewor… | 52 | 8212 | 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 |
| brycedrennan/imaginAIry A Python library and CLI tool (imaginairy/aimg) for generating images and videos with Stable Diffusion and Stable Video Diffusion models. I… | 60 | 8186 | active |
| AI4Finance-Foundation/FinRobot FinRobot is an open-source AI agent platform for financial applications built on large language models, unifying LLMs, reinforcement learni… | 78 | 8140 | active |
| RapidAI/RapidOCR RapidOCR is an open-source, multi-language OCR toolkit that performs text detection and recognition using models converted to run on ONNX R… | 99 | 8134 | 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 |
| 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 |
| suno-ai/bark Bark is Suno's open-source transformer-based text-to-audio model that generates highly realistic multilingual speech, music, background noi… | 31 | 39276 | maintenance |
| 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 |
| 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 |
| 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 |
| 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 |
| MeiGen-AI/InfiniteTalk InfiniteTalk is an open-source model and framework for unlimited-length audio-driven talking video generation, supporting both image-to-vid… | 52 | 7888 | active |
| 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 |
| OpenFace OpenFace is a facial behavior analysis toolkit that performs facial landmark detection, head pose estimation, facial action unit recognitio… | 23 | 7774 | active |
| TencentARC/GFPGAN GFPGAN is a Python library built on PyTorch that restores and enhances real-world degraded face photos using GAN-based priors. It provides … | 23 | 37683 | maintenance |
| ERNIE Official repository for Baidu's ERNIE 4.5 family of large multimodal models and ERNIEKit, an industrial-grade training toolkit built on Pad… | 62 | 7737 | active |
| geekyutao/Inpaint-Anything Inpaint Anything combines Segment Anything (SAM) with inpainting models like LaMa and Stable Diffusion to remove, fill, or replace objects … | 62 | 7725 | active |