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 |
|---|---|---|---|
| Geniusay/ChopperBot ChopperBot is a fully automated AI bot that monitors popular live streams across platforms like Douyu, Huya, Bilibili, Douyin, and Twitch, … | 42 | 2826 | active |
| OpenDCAI/DataFlex DataFlex is a data-centric training framework built on LLaMA-Factory that dynamically schedules LLM training data during optimization. It p… | 63 | 2824 | active |
| Camb-ai/MARS5-TTS MARS5 is an open-source English text-to-speech model from CAMB.AI that uses a two-stage AR-NAR pipeline to generate expressive speech with … | 15 | 2821 | active |
| huawei-noah/HEBO A research monorepo from Huawei Noah's Ark Lab containing official implementations of Bayesian optimization, reinforcement learning, and ge… | 67 | 2812 | active |
| kpu/kenlm KenLM is a C++ toolkit for estimating, filtering, and querying n-gram language models with modified Kneser-Ney smoothing. It is designed fo… | 34 | 2812 | stable |
| kha-white/manga-ocr Manga OCR is a Python library providing optical character recognition for Japanese text, focused on Japanese manga. It uses a custom end-to… | 77 | 2809 | stable |
| nutonomy/nuscenes-devkit The official Python devkit for the nuScenes dataset, a large-scale autonomous driving dataset from Motional. It provides dataset loading, v… | 62 | 2807 | stable |
| satijalab/seurat Seurat is an R toolkit for single-cell genomics developed by the Satija Lab, providing a complete pipeline for analyzing single-cell RNA-se… | 88 | 2803 | stable |
| facebookresearch/HyperAgents A research framework from Meta AI implementing self-referential, self-improving LLM agents that can optimize their own code for arbitrary c… | 54 | 2803 | active |
| pytorch/xla PyTorch/XLA is a Python package that connects the PyTorch deep learning framework to XLA devices such as Google Cloud TPUs via the XLA deep… | 63 | 2801 | active |
| rom1504/clip-retrieval A Python toolkit for computing CLIP embeddings for images and text and building a semantic search/retrieval system on top of them. It inclu… | 52 | 2797 | active |
| Mega4alik/ollm oLLM is a lightweight Python library for large-context LLM inference built on Hugging Face Transformers and PyTorch. It offloads weights an… | 57 | 2795 | active |
| lucidrains/DALLE2-pytorch A PyTorch implementation of OpenAI's DALL-E 2 text-to-image synthesis model, focusing on the diffusion prior network that predicts image em… | 23 | 11302 | maintenance |
| rhasspy/piper Piper is a fast, local neural text-to-speech system that runs offline on modest hardware, including Raspberry Pi devices. It offers many pr… | 10 | 11290 | maintenance |
| torinmb/mediapipe-touchdesigner A GPU-accelerated, self-contained MediaPipe plugin for TouchDesigner that runs MediaPipe vision models (face detection, face/hand/pose trac… | 83 | 2784 | active |
| Autodistill Autodistill is a Python library that uses large foundation vision models (like Grounding DINO, Grounded SAM, and CLIP) to automatically lab… | 64 | 2784 | active |
| FasterDecoding/Medusa Medusa is a framework that accelerates LLM text generation by adding multiple decoding heads to an existing model, avoiding the need for a … | 19 | 2776 | active |
| Stan Stan is a C++ probabilistic programming library for full Bayesian inference via NUTS/Hamiltonian Monte Carlo, approximate inference via ADV… | 88 | 2774 | stable |
| icereed/paperless-gpt A self-hosted companion application for paperless-ngx that uses LLMs and vision models to auto-generate document titles, tags, and dates, a… | 90 | 2773 | active |
| pyro-ppl/numpyro NumPyro is a lightweight probabilistic programming library providing a NumPy backend for Pyro, built on JAX for automatic differentiation a… | 94 | 2770 | active |
| google-research/tabfm TabFM is a scikit-learn compatible tabular foundation model from Google Research that performs zero-shot classification and regression on t… | 63 | 2769 | active |
| NVlabs/stylegan2 The official TensorFlow implementation of StyleGAN2, NVIDIA's improved style-based generative adversarial network for high-quality uncondit… | 32 | 11180 | maintenance |
| NVIDIA/FastPhotoStyle FastPhotoStyle is NVIDIA's official PyTorch implementation of the ECCV 2018 paper 'A Closed-form Solution to Photorealistic Image Stylizati… | 23 | 11171 | maintenance |
| apple/turicreate Turi Create is a Python library from Apple that simplifies building custom machine learning models for tasks like image classification, obj… | 10 | 11155 | maintenance |
| huggingface/swift-coreml-diffusers A native SwiftUI application demonstrating how to run Stable Diffusion text-to-image generation on-device using Apple's Core ML Stable Diff… | 10 | 2755 | active |
| voxelmorph/voxelmorph VoxelMorph is a Python library for learning-based image registration and alignment, using unsupervised deep learning to model deformations … | 75 | 2754 | active |
| thewh1teagle/kokoro-onnx A Python library that runs the Kokoro text-to-speech model via ONNX Runtime, supporting CPU and GPU inference. It provides multi-language T… | 65 | 2750 | active |
| lightseekorg/tokenspeed TokenSpeed is a high-performance LLM inference engine designed for agentic workloads, aiming for TensorRT-LLM-level performance with vLLM-l… | 78 | 2743 | active |
| TWIST GMR is a Python library for general motion retargeting that converts human motions into joint commands for diverse humanoid robots in real … | 49 | 2741 | active |
| artidoro/qlora QLoRA is the official implementation of the QLoRA paper, an efficient finetuning approach that backpropagates through a frozen 4-bit quanti… | 30 | 11035 | maintenance |
| qualcomm/aimet AIMET (AI Model Efficiency Toolkit) is a Python library from Qualcomm providing advanced quantization and compression techniques for traine… | 98 | 2723 | active |
| intel/neural-compressor Intel Neural Compressor is an open-source Python library providing state-of-the-art low-bit quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4),… | 95 | 2723 | active |
| black-forest-labs/flux2 Official inference repository for Black Forest Labs' FLUX.2 family of open-weight image generation and editing models. It provides minimal … | 45 | 2720 | active |
| prophesier/diff-svc Diff-SVC is a deep learning project that performs singing voice conversion using diffusion models, transforming input singing audio into a … | 58 | 2719 | active |
| yuqinie98/PatchTST Official PyTorch implementation of PatchTST, an ICLR 2023 Transformer model for long-term time series forecasting based on patching and cha… | 32 | 2719 | stable |
| mljs/ml ml.js is an umbrella library bundling the mljs organization's machine learning tools for JavaScript, including clustering, classification, … | 23 | 2719 | active |
| lengstrom/fast-style-transfer A TensorFlow implementation of fast neural style transfer that applies the style of famous paintings to photos and videos in real time. It … | 32 | 10957 | maintenance |
| secretflow/secretflow SecretFlow is a unified Python framework for privacy-preserving data analysis and machine learning. It layers cryptographic devices (MPC, H… | 62 | 2713 | active |
| bmild/nerf The official TensorFlow implementation of NeRF (Neural Radiance Fields), the ECCV 2020 paper representing scenes as neural radiance fields … | 36 | 10944 | maintenance |
| openai/privacy-filter OpenAI Privacy Filter is a small bidirectional token-classification model (1.5B total, 50M active parameters) for detecting and masking per… | 46 | 2711 | active |
| IDEA-Research/T-Rex T-Rex is the official Python API client for T-Rex2, a generic open-set object detection model that combines text and visual prompts to dete… | 45 | 2711 | active |
| yuweihao/MambaOut MambaOut is a PyTorch implementation of Gated CNN models from the CVPR 2025 paper 'MambaOut: Do We Really Need Mamba for Vision?', which qu… | 16 | 2710 | stable |
| KimMeen/Time-LLM Time-LLM is the official PyTorch implementation of an ICLR 2024 paper that reprograms frozen large language models (Llama, GPT-2, BERT) for… | 44 | 2707 | active |
| TMElyralab/MusePose MusePose is a diffusion-based, pose-guided image-to-video generation framework for creating virtual human videos, where a character in a re… | 25 | 2706 | active |
| magic-research/magic-animate MagicAnimate is the official implementation of a CVPR 2024 diffusion-based human image animation framework that animates a reference image … | 41 | 10889 | maintenance |
| dreamzero0/dreamzero DreamZero is NVIDIA's World Action Model (WAM) that jointly predicts future video and actions from a pretrained video diffusion backbone, e… | 48 | 2700 | active |
| ZHZisZZ/dllm dLLM is a Python library that unifies training, inference, and evaluation of diffusion language models such as LLaDA and Dream. It builds o… | 56 | 2699 | active |
| HiLab-git/SSL4MIS A benchmark and code collection of semi-supervised learning methods for medical image segmentation, re-implementing approaches like Mean Te… | 40 | 2699 | active |
| princeton-vl/DROID-SLAM DROID-SLAM is a deep learning-based visual SLAM system for monocular, stereo, and RGB-D cameras that estimates camera trajectory and dense … | 37 | 2699 | active |
| roboflow/maestro maestro is a Python library from Roboflow that streamlines fine-tuning of multimodal vision-language models such as Florence-2, PaliGemma 2… | 61 | 2696 | active |
| naiveHobo/InvoiceNet InvoiceNet is a deep neural network application with a GUI for extracting structured information from invoice documents in PDF, JPG, and PN… | 32 | 2696 | active |
| baaivision/EVA EVA is a family of large-scale vision foundation models from BAAI, including masked image models (EVA-01/02) and scaled CLIP models (EVA-CL… | 23 | 2694 | active |
| SkyReels SkyReels V1 is an open-source human-centric video foundation model with Text-to-Video and Image-to-Video variants, fine-tuned from HunyuanV… | 22 | 2694 | active |
| tryolabs/norfair Norfair is a lightweight, customizable Python library for real-time multi-object tracking that works with any detector outputting (x, y) co… | 28 | 2689 | stable |
| JIA-Lab-research/LISA LISA (Large Language Instructed Segmentation Assistant) is a multimodal large language model that performs reasoning-based image segmentati… | 29 | 2687 | active |
| openai/DALL-E The official PyTorch package for the discrete VAE (dVAE) component of OpenAI's DALL·E model. It does not include the transformer that gener… | 10 | 10820 | maintenance |
| MrGiovanni/UNetPlusPlus Official implementation of UNet++, a nested U-Net architecture for medical image segmentation, in both Keras and PyTorch. It redesigns skip… | 73 | 2685 | stable |
| InfiniteYou InfiniteYou (InfU) is a research framework from ByteDance for identity-preserved text-to-image generation built on Diffusion Transformers l… | 34 | 2685 | active |
| DDSP-SVC DDSP-SVC is an open-source singing voice conversion system built on Differentiable Digital Signal Processing, designed as a free AI voice c… | 65 | 2682 | active |
| PowerHouseMan/ComfyUI-AdvancedLivePortrait A ComfyUI custom node implementing LivePortrait for fast facial expression editing and animation with real-time preview. It can edit expres… | 24 | 2681 | active |
| Tencent-Hunyuan/HY-World-2.0 HY-World 2.0 is Tencent Hunyuan's open-source multi-modal world model framework that reconstructs, generates, and simulates 3D worlds from … | 55 | 2680 | active |
| LAM LHM is a PyTorch-based large reconstruction model that reconstructs high-fidelity animatable 3D human avatars from a single image in second… | 49 | 2679 | active |
| google-deepmind/mctx Mctx is a JAX-native Python library implementing Monte Carlo tree search algorithms such as AlphaZero, MuZero, and Gumbel MuZero. It suppor… | 81 | 2678 | active |
| 3D Slicer 3D Slicer is a free, open-source desktop platform for visualization, processing, segmentation, registration, and analysis of medical and bi… | 68 | 2676 | stable |
| stochasticai/xTuring xTuring is a Python library for fine-tuning, evaluating, and running open-source large language models such as LLaMA, GPT-J, GPT-2, Qwen, a… | 65 | 2675 | active |
| georgia-tech-db/evadb EvaDB is a Python database system for building AI-powered applications, exposing a SQL API that queries structured data, videos, and unstru… | 10 | 2674 | active |
| StableSR StableSR is a Python research library that leverages pre-trained Stable Diffusion priors for real-world blind image super-resolution. It pr… | 21 | 2672 | stable |
| AMAP-ML/SkillClaw SkillClaw is a Python-based framework that lets AI agent skills evolve collectively from real interactions across sessions, agents, devices… | 55 | 2669 | active |
| Tencent/MimicMotion MimicMotion is a diffusion-based framework from Tencent for generating high-quality human motion videos guided by pose sequences, featuring… | 44 | 2668 | active |
| open-gigaai/giga-brain-0 GigaBrain-0/0.7 is an open-source vision-language-action (VLA) model family for generalist embodied agents, powered by world models and a t… | 60 | 2659 | active |
| Nutlope/roomGPT RoomGPT is an open-source Next.js web application that lets users upload a photo of a room and generate redesigned variations using the Con… | 31 | 10681 | maintenance |
| phillipi/pix2pix The original Torch (Lua) implementation of pix2pix, a conditional GAN for image-to-image translation tasks such as synthesizing photos from… | 32 | 10659 | maintenance |
| FeatherTalk An ultralight talking-head (digital human) model that animates a person's face from audio input and runs in real time on mobile devices. It… | 61 | 2648 | active |
| Matchering Matchering 2.0 is an open-source Python library, containerized web app, and ComfyUI node for audio matching and mastering. It takes a targe… | 60 | 2648 | active |
| swz30/Restormer Restormer is an efficient Transformer architecture for high-resolution image restoration, published as a CVPR 2022 Oral paper. It provides … | 41 | 2648 | stable |
| SeldonIO/alibi Alibi is a Python library providing algorithms for explaining and interpreting machine learning models, including black-box, white-box, loc… | 40 | 2646 | active |
| plexe-ai/plexe Plexe is a Python library and CLI that builds machine learning models from natural language descriptions using a multi-agent AI workflow. Y… | 56 | 2645 | active |
| facebookresearch/ParlAI ParlAI is a Python framework from Facebook AI Research for sharing, training, and evaluating dialogue models across many openly available d… | 10 | 10618 | maintenance |
| robosuite robosuite is a modular simulation framework powered by the MuJoCo physics engine for robot learning, offering standardized benchmark enviro… | 65 | 2643 | active |
| ultralytics/yolov3 Ultralytics' PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection. It provides training, validation… | 66 | 10613 | maintenance |
| AudioLDM AudioLDM 2 is a Python library and CLI for generating audio, music, and speech from text prompts using latent diffusion models. It includes… | 29 | 2642 | active |
| EMI-Group/evox EvoX is a distributed GPU-accelerated framework for evolutionary computation, compatible with PyTorch and built on JAX. It provides 50+ evo… | 97 | 2640 | active |
| AudioMuse-AI AudioMuse-AI is a self-hosted application that performs sonic analysis on your music library to generate intelligent, AI-driven playlists w… | 80 | 2629 | active |
| OmniSVG/OmniSVG OmniSVG is a family of end-to-end multimodal SVG generation models built on pre-trained Vision-Language Models, released with inference cod… | 48 | 2626 | active |
| lucidrains/audiolm-pytorch A PyTorch implementation of AudioLM, Google Research's language modeling approach to audio generation, including a MIT-licensed SoundStream… | 34 | 2625 | active |
| meta-pytorch/torchrec TorchRec is a PyTorch domain library for building recommendation systems at scale. It provides distributed sharding of large embedding tabl… | 95 | 2616 | active |
| google-coral/coralnpu Coral NPU is an open-source neural processing unit (NPU) hardware IP core from Google Research, built on the 32-bit RISC-V ISA with matrix,… | 72 | 2604 | active |
| apache/hamilton Apache Hamilton is a lightweight Python library for defining directed acyclic graphs (DAGs) of data transformations as plain, testable Pyth… | 89 | 2603 | active |
| kayba-ai/agentic-context-engine Agentic Context Engine (ACE) is a Python library that lets LLM-based agents learn from experience by maintaining and evolving an evolving c… | 82 | 2602 | active |
| luca-medeiros/lang-segment-anything A Python library combining Meta's Segment Anything Model 2 with GroundingDINO to generate segmentation masks for objects in images specifie… | 35 | 2600 | active |
| crowsonkb/k-diffusion A PyTorch library implementing Karras et al. (2022) diffusion models with enhancements like improved sampling algorithms and transformer-ba… | 49 | 2599 | active |
| CodeGeeX CodeGeeX is a family of open multilingual code generation large language models (13B and successors CodeGeeX2/CodeGeeX4) pre-trained on 20+… | 24 | 2597 | active |
| atong01/conditional-flow-matching TorchCFM is a PyTorch library implementing Conditional Flow Matching (CFM), a simulation-free training objective for continuous normalizing… | 72 | 2596 | active |
| asteroid-team/asteroid Asteroid is a PyTorch-based audio source separation toolkit for researchers, providing modular building blocks (filterbanks, encoders, mask… | 66 | 2593 | active |
| median-research-group/LibMTL LibMTL is an open-source PyTorch library for Multi-Task Learning (MTL). It provides implementations of many MTL architectures and gradient-… | 38 | 2593 | active |
| facebookresearch/demucs Demucs is a state-of-the-art music source separation model from Meta AI that splits songs into stems like drums, bass, and vocals using a h… | 10 | 10373 | maintenance |
| Google Colaboratory The official Python libraries shipped inside Google Colaboratory, Google's free hosted Jupyter notebook environment for machine learning ed… | 77 | 2591 | active |
| kijai/ComfyUI-HunyuanVideoWrapper A set of custom ComfyUI nodes that wrap Tencent's HunyuanVideo text-to-video and image-to-video diffusion model for use inside ComfyUI work… | 35 | 2591 | active |
| DevXDojo/MrRSS MrRSS is a free, cross-platform desktop RSS/Atom feed reader built with Go and Wails (Vue.js frontend) that integrates AI features like aut… | 87 | 2588 | active |
| vita-epfl/Stable-Video-Infinity Stable Video Infinity (SVI) is a research codebase for infinite-length video generation using video diffusion transformers with an error-re… | 60 | 2579 | active |