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 |
|---|---|---|---|
| lorey/mlscraper mlscraper is a Python library that automatically extracts structured data from HTML pages using machine learning. Instead of writing CSS se… | 23 | 1385 | active |
| soda-inria/tabicl TabICLv2 is an open-source tabular foundation model that performs classification and regression via a single forward pass through a pre-tra… | 83 | 1384 | active |
| Zefan-Cai/KVCache-Factory KVCache-Factory is a unified Python framework implementing many KV cache compression, retrieval, merging, and quantization methods for long… | 65 | 1384 | active |
| yanx27/Pointnet_Pointnet2_pytorch A pure PyTorch implementation of the PointNet and PointNet++ deep learning architectures for point cloud processing. It includes training a… | 32 | 4958 | maintenance |
| hustvl/VAD VAD is an end-to-end autonomous driving framework that models the driving scene as a fully vectorized representation of agents and map elem… | 57 | 1383 | active |
| autonomousvision/unimatch UniMatch is a PyTorch research library implementing a unified transformer-based model for optical flow, stereo matching, and depth estimati… | 32 | 1383 | stable |
| ali-vilab/TeaCache TeaCache is a training-free caching approach that accelerates inference for video diffusion models by estimating output differences across … | 30 | 1383 | active |
| keyu-tian/SparK SparK is the official PyTorch implementation of an ICLR 2023 Spotlight paper that applies BERT/MAE-style masked image modeling to convoluti… | 23 | 1382 | stable |
| ReActor ComfyUI-ReActor is a fast and simple face swap extension node for ComfyUI, based on the ReActor face-swapping engine. It includes a nudity … | 79 | 1381 | active |
| OpenGVLab/DragGAN An unofficial full-featured Python implementation of DragGAN, the interactive point-based image manipulation method built on StyleGAN gener… | 30 | 4944 | maintenance |
| AlmondGod/tinyworlds A minimal Python implementation of DeepMind's Genie autoregressive world model, including a video tokenizer, action tokenizer, and dynamics… | 51 | 1380 | active |
| QwenAudio/ThinkSound ThinkSound is a PyTorch implementation of a NeurIPS 2025 framework that generates and edits audio from video, text, or audio inputs using C… | 49 | 1380 | active |
| hudson-and-thames/mlfinlab MlFinLab is a Python library implementing financial machine learning techniques from data structure generation through labeling, feature en… | 32 | 4941 | maintenance |
| nomadkaraoke/python-audio-separator A Python package and CLI that separates audio files into stems (vocals, instrumental, drums, bass, etc.) using pre-trained models from Ulti… | 90 | 1378 | active |
| unitreerobotics/unitree_rl_lab A collection of reinforcement learning environments for Unitree robots (Go2, H1, G1) built on NVIDIA IsaacLab. It enables training locomoti… | 53 | 1378 | active |
| FireRedTeam/FireRed-Image-Edit FireRed-Image-Edit is an open-source image editing foundation model built on diffusion models, released as PyTorch model weights with infer… | 46 | 1378 | active |
| Neuromancer NeuroMANCER is a PyTorch-based differentiable programming library for solving parametric constrained optimization problems, physics-informe… | 69 | 1377 | active |
| FujiwaraChoki/supoclip SupoClip is an open-source, AI-powered video clipping tool that turns long videos like podcasts and streams into vertical 9:16 short clips … | 63 | 1377 | active |
| ServiceNow/BrowserGym BrowserGym is a Gym-style environment library for web task automation, providing a unified interface to web agent benchmarks like MiniWoB, … | 78 | 1376 | active |
| erwincoumans/tiny-differentiable-simulator Tiny Differentiable Simulator (TDS) is a header-only C++ and CUDA physics library for rigid-body dynamics with zero dependencies, supportin… | 67 | 1376 | active |
| giacomelli/GeneticSharp GeneticSharp is a fast, extensible, multithreading C# genetic algorithm library for .NET 6, .NET Standard, and .NET Framework applications.… | 42 | 1376 | stable |
| siyuanliii/masa Official PyTorch implementation of MASA (CVPR 2024 Highlight), a universal instance appearance model that learns to match any objects acros… | 31 | 1376 | active |
| qubvel/segmentation_models A Python library providing neural network architectures for image segmentation (Unet, FPN, Linknet, PSPNet) built on Keras and TensorFlow K… | 23 | 4923 | maintenance |
| minimaxir/textgenrnn A Python 3 library built on Keras/TensorFlow for easily training char-rnn style neural networks that generate text from any dataset in a fe… | 23 | 4918 | maintenance |
| Goal-Driven Autonomous Exploration (DRL Robot Navigation) A ROS Gazebo simulation project that trains a mobile robot to navigate to random goals while avoiding obstacles using a TD3 deep reinforcem… | 54 | 1374 | active |
| MACE MACE is a Python library implementing fast and accurate machine learning interatomic potentials using higher-order equivariant message pass… | 86 | 1373 | active |
| macanv/BERT-BiLSTM-CRF-NER A TensorFlow implementation of named entity recognition that fine-tunes Google BERT with a BiLSTM-CRF model, primarily targeting Chinese te… | 32 | 4907 | maintenance |
| Meituan-AutoML/MobileVLM MobileVLM is a family of compact vision language models (1.4B-3B parameters) designed to run efficiently on mobile devices, combining small… | 17 | 1371 | active |
| OpenPrompt OpenPrompt is a PyTorch-based open-source framework for prompt-learning, providing a standard, flexible pipeline of templates and verbalize… | 23 | 4903 | maintenance |
| bytedance/flux Flux is a GPU kernel library from ByteDance that overlaps computation with communication for tensor and expert parallelism in dense and MoE… | 30 | 1369 | active |
| k2-fsa/k2 k2 is a C++/CUDA library with Python bindings that implements differentiable Finite State Automaton (FSA) and Finite State Transducer (FST)… | 60 | 1368 | active |
| OpenPPL/ppl.nn PPLNN is a high-performance deep-learning inference engine written in C++ that runs ONNX models on x86 CPUs and NVIDIA GPUs, with a dedicat… | 32 | 1368 | active |
| mega-sam/mega-sam MegaSaM is a research codebase implementing a deep visual SLAM system that estimates camera parameters and consistent depth maps from casua… | 45 | 1367 | active |
| NVIDIA-AI-IOT/torch2trt torch2trt is a Python library that converts PyTorch models to TensorRT engines using the TensorRT Python API, with a simple single-function… | 23 | 4885 | maintenance |
| google/visualblocks Visual Blocks for ML is a Google visual programming framework that provides a no-code node graph editor for building ML pipelines from drag… | 67 | 1366 | active |
| PKU-VCL-3DV/SLAM3R SLAM3R is a real-time dense 3D scene reconstruction system that regresses 3D points from monocular RGB video using feed-forward neural netw… | 39 | 1366 | active |
| MoonInTheRiver/DiffSinger Official PyTorch implementation of DiffSinger, an AAAI 2022 paper on singing voice synthesis and text-to-speech using a shallow diffusion m… | 61 | 4882 | maintenance |
| kengz/SLM-Lab SLM Lab is a modular deep reinforcement learning framework built in PyTorch, offering ready-to-use algorithms like PPO, SAC, DQN, and A2C v… | 91 | 1365 | active |
| ImprintLab/MedSegDiff MedSegDiff is a diffusion probabilistic model framework for segmenting and reconstructing organs and tissues from medical images, with a tr… | 47 | 1364 | active |
| bytedance/UNO UNO is a research framework from ByteDance for subject-driven image generation with diffusion transformers, supporting both single- and mul… | 35 | 1364 | active |
| Sense-X/Co-DETR Co-DETR is a PyTorch implementation of DETRs with Collaborative Hybrid Assignments Training, an ICCV 2023 object detection and instance seg… | 32 | 1364 | stable |
| FreedomIntelligence/HuatuoGPT-o1 HuatuoGPT-o1 is a medical large language model project for complex medical reasoning, with open-sourced models (7B-72B), fine-tuning data, … | 22 | 1363 | active |
| huggingface/finetrainers finetrainers is a Hugging Face library for scalable, memory-optimized training (fine-tuning) of diffusion models, including LoRA training o… | 68 | 1362 | active |
| natasha/natasha Natasha is a Python library that solves basic NLP tasks for the Russian language, including tokenization, sentence segmentation, morphology… | 77 | 1361 | active |
| Swift Transformers A Swift Package providing a transformers-like API for Swift apps, including fast tokenization, chat templating, and reliable model download… | 97 | 1360 | active |
| shivammehta25/Matcha-TTS Matcha-TTS is a PyTorch-based text-to-speech system that uses conditional flow matching for fast, non-autoregressive speech synthesis. It s… | 60 | 1360 | active |
| MoonshotAI/Kimi-Dev Kimi-Dev-72B is an open-source coding LLM from Moonshot AI specialized in software engineering tasks like issue resolution and bug fixing, … | 35 | 1359 | active |
| rockingdingo/deepnlp DeepNLP is a deep learning NLP pipeline implemented on TensorFlow, distributed as a Python package, which has evolved into the DeepNLP AI S… | 32 | 1358 | active |
| jhc13/taggui TagGUI is a cross-platform desktop application for quickly adding and editing image tags and captions, aimed at creators of image datasets … | 47 | 1357 | active |
| yinguobing/head-pose-estimation A Python library for realtime human head pose estimation using ONNX Runtime and OpenCV. It combines face detection (SCRFD), 68-point facial… | 23 | 1357 | stable |
| ant-research/CoDeF CoDeF is the official PyTorch implementation of Content Deformation Fields, a video representation combining a canonical content field and … | 29 | 4843 | maintenance |
| blei-lab/edward Edward is a Python library for probabilistic modeling, inference, and criticism built on TensorFlow. It supports deep generative models, va… | 23 | 4843 | maintenance |
| huridocs/pdf-document-layout-analysis A Dockerized microservice by HURIDOCS that performs PDF document layout analysis, OCR, and element segmentation/classification (texts, titl… | 91 | 1356 | active |
| uxlfoundation/scikit-learn-intelex Intel's Extension for Scikit-learn is a free AI accelerator that speeds up existing scikit-learn workflows on CPUs and GPUs, claiming up to… | 87 | 1356 | active |
| ZJU-REAL/ClawGUI ClawGUI is a unified Python framework for GUI agents covering the full lifecycle: online reinforcement learning training (ClawGUI-RL with G… | 64 | 1356 | active |
| wyhuai/DDNM DDNM is a Python research codebase implementing the Denoising Diffusion Null-Space Model for zero-shot image restoration, published as an I… | 32 | 1356 | stable |
| llm-attacks/llm-attacks Official research code for 'Universal and Transferable Adversarial Attacks on Aligned Language Models', implementing the GCG algorithm for … | 29 | 4839 | maintenance |
| mmorise/World WORLD is a C++ library for high-quality speech analysis, manipulation, and synthesis based on a vocoder design. It estimates F0 (via DIO/Ha… | 75 | 1355 | stable |
| gempy-project/gempy GemPy is an open-source Python library for implicit 3D structural geological modeling, generating complex models of layers, faults, folds, … | 99 | 1354 | active |
| Softcatala/whisper-ctranslate2 A command-line transcription and translation tool compatible with OpenAI's Whisper CLI, built on CTranslate2 and faster-whisper for up to 4… | 57 | 1354 | active |
| JoySafety/JoySafety JoySafety is an open-source large language model safety framework from JD.com, written in Java, providing prompt injection detection, conte… | 42 | 1354 | active |
| lxtGH/OMG-Seg Official research codebase for OMG-Seg (CVPR 2024) and OMG-LLaVA (NeurIPS 2024), unified models for image-level, object-level, and pixel-le… | 68 | 1353 | active |
| ClownsharkBatwing/RES4LYF RES4LYF is a ComfyUI custom node collection providing advanced diffusion samplers (RES samplers) that achieve high-quality image generation… | 68 | 1352 | active |
| sjvasquez/handwriting-synthesis A Python implementation of Alex Graves' handwriting synthesis experiments using recurrent neural networks, generating realistic handwritten… | 32 | 4824 | maintenance |
| owlbarn/owl Owl is an OCaml library for scientific and engineering computing, providing n-dimensional arrays, linear algebra, statistics, optimization,… | 62 | 1351 | active |
| studio-dots-ai/dots.tts dots.tts is a 2B-parameter fully continuous, end-to-end autoregressive text-to-speech system, distributed as a Python library with pretrain… | 77 | 1350 | active |
| bytedance/Lance Lance is a 3B-parameter native unified multimodal model from ByteDance for image and video understanding, generation, and editing, trained … | 52 | 1350 | active |
| christophschuhmann/improved-aesthetic-predictor A CLIP+MLP neural network that predicts how much people on average like an image, trained on AVA dataset ratings. It is widely used for fil… | 32 | 1349 | stable |
| MegEngine/MegEngine MegEngine is a fast, scalable deep learning framework with automatic differentiation, developed in C++ with Python bindings. It unifies tra… | 23 | 4809 | maintenance |
| airbnb/aerosolve Aerosolve is a machine learning library from Airbnb built for human-friendly, interpretable modeling on the JVM. It provides a thrift-based… | 42 | 4807 | maintenance |
| FIND3 FIND (Framework for Internal Navigation and Discovery) is a self-hosted server that determines indoor position using WiFi fingerprints from… | 23 | 4806 | maintenance |
| FIND3 FIND3 is a self-hosted framework for high-precision indoor positioning ('indoor GPS') that uses WiFi, Bluetooth, and magnetic field fingerp… | 23 | 4806 | maintenance |
| segment-any-text/wtpsplit wtpsplit is a Python toolkit for segmenting text into sentences or other semantic units using the SaT and WtP deep learning models. It prov… | 93 | 1347 | active |
| xtreme1-io/xtreme1 Xtreme1 is an open-source, self-hosted data labeling and annotation platform for multimodal training data, supporting images, 3D LiDAR poin… | 97 | 1345 | active |
| 3dg1luk43/ha_washdata A Home Assistant custom integration that monitors smart-plug-connected appliances like washing machines, dryers, and dishwashers. It learns… | 82 | 1345 | active |
| muzishen/IMAGDressing IMAGDressing-v1 is a diffusion-based framework for customizable virtual dressing that generates human images with fixed garments and contro… | 41 | 1345 | active |
| peterdsharpe/AeroSandbox AeroSandbox is a Python library for designing and optimizing aircraft and other engineered systems using automatic differentiation. It prov… | 75 | 1344 | active |
| VAST-AI-Research/TripoSplat TripoSplat is an inference-only Python library from TripoAI that converts a single 2D image into high-quality 3D Gaussian splats with a var… | 54 | 1344 | active |
| hao-ai-lab/LookaheadDecoding A Python library implementing Lookahead Decoding, an exact parallel decoding algorithm that accelerates LLM inference without a draft model… | 28 | 1343 | active |
| claritylab/lucida Lucida is an open-source speech and vision based intelligent personal assistant inspired by Sirius. It orchestrates modular back-end micros… | 32 | 4784 | maintenance |
| inference-labs-inc/dsperse DSperse is a proving-system-agnostic tool for verifiable AI that decomposes ONNX neural network models into circuit-compatible segments and… | 76 | 1342 | active |
| rstudio/tensorflow An R package that provides full access to the TensorFlow API from R via reticulate, bridging R users to TensorFlow's Python implementation.… | 56 | 1340 | active |
| HiDream-ai/HiDream-O1-Image HiDream-O1-Image is an open-weights 8B image generation foundation model built on a Pixel-level Unified Transformer (UiT) that natively enc… | 50 | 1340 | active |
| alibaba/graph-learn Graph-Learn (formerly AliGraph) is a distributed framework for developing and applying large-scale graph neural networks, with a training l… | 32 | 1340 | active |
| open-edge-platform/geti Geti is an open-source, end-to-end Vision AI application from Intel that takes users from raw images to deployed computer vision models, ru… | 92 | 1338 | active |
| jonathan-laurent/AlphaZero.jl A generic, simple, and fast Julia implementation of DeepMind's AlphaZero algorithm for training game-playing agents via self-play and MCTS.… | 69 | 1338 | active |
| BytedTsinghua-SIA/CUDA-Agent CUDA-Agent is a large-scale agentic reinforcement learning system from ByteDance Seed and Tsinghua that trains LLMs to generate high-perfor… | 53 | 1338 | active |
| senguptaumd/Background-Matting Official research code for 'Background Matting: The World is Your Green Screen' (CVPR 2020), a deep network that extracts per-pixel alpha m… | 32 | 4764 | maintenance |
| PKU-YuanGroup/MagicTime MagicTime is a metamorphic time-lapse video generation pipeline built on diffusion-based text-to-video models, with a MagicAdapter, dynamic… | 56 | 1337 | active |
| wenqsun/DimensionX DimensionX is a research framework that generates photorealistic 3D and 4D scenes from a single image using controllable video diffusion mo… | 40 | 1336 | active |
| CarperAI/trlx trlX is a distributed training framework for fine-tuning large language models with reinforcement learning from human feedback (RLHF), supp… | 23 | 4756 | maintenance |
| lean-dojo/LeanCopilot Lean Copilot integrates large language models natively into the Lean 4 theorem prover for proof automation. It provides tactic suggestion, … | 91 | 1335 | active |
| awslabs/dgl-ke DGL-KE is a high-performance Python package built on Deep Graph Library (DGL) for training, evaluating, and inferring knowledge graph embed… | 60 | 1335 | active |
| pykalman/pykalman pykalman is a Python library implementing the Kalman Filter, Kalman Smoother, Unscented Kalman Filter, and EM algorithm for state estimatio… | 68 | 1334 | active |
| cvzone/cvzone CVZone is a Python computer vision helper library that wraps OpenCV and MediaPipe to simplify image processing and AI functions like hand t… | 32 | 1334 | active |
| IrisRainbowNeko/genshin_auto_fish A Genshin Impact auto-fishing AI built from a YOLOX object detection model (fish and rod landing point localization) and a DQN reinforcemen… | 23 | 4748 | maintenance |
| mapillary/inplace_abn A PyTorch extension library implementing In-Place Activated BatchNorm (InPlace-ABN), which redefines BN plus nonlinear activation as a sing… | 67 | 1333 | stable |
| braindecode/braindecode Braindecode is an open-source Python toolbox built on PyTorch for decoding raw electrophysiological brain signals such as EEG, ECoG, and ME… | 97 | 1332 | active |
| bytedance/Bernini Bernini is a unified framework for video generation and editing that combines an MLLM-based semantic planner with a DiT-based renderer perf… | 54 | 1330 | active |
| Yas Scanner Yas is a fast screen-scanning tool that uses a custom-trained SVTR OCR model to read Genshin Impact and Honkai: Star Rail artifact stats di… | 37 | 1330 | active |