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 |
|---|---|---|---|
| NirAharon/BoT-SORT BoT-SORT is a state-of-the-art multi-object tracker that combines motion and appearance information with camera motion compensation and an … | 32 | 1547 | active |
| combust/mleap MLeap is a serialization format (Bundle.ML) and portable execution engine for machine learning pipelines, implemented in Scala with Python … | 98 | 1546 | active |
| tensorflow/gnn TensorFlow GNN is a Python library for building Graph Neural Networks on TensorFlow, including a GraphTensor type for heterogeneous graphs,… | 66 | 1546 | active |
| kijai/ComfyUI-CogVideoXWrapper A ComfyUI custom node wrapper for CogVideoX and related video generation models (including Fun variants, CogVideoX 1.5, and Go-with-the-Flo… | 36 | 1546 | active |
| RLHFlow/RLHF-Reward-Modeling A collection of training recipes for reward models used in RLHF, covering Bradley-Terry reward models, pairwise preference models, ArmoRM, … | 30 | 1546 | active |
| baichuan-inc/Baichuan-7B Baichuan-7B is an open-source, commercially usable 7-billion-parameter pretrained language model built on the Transformer architecture, tra… | 30 | 5648 | maintenance |
| UCCL UCCL is a high-performance GPU communication library written in C++ that provides collectives (as a drop-in NCCL/RCCL replacement), P2P tra… | 69 | 1545 | active |
| lucidrains/soundstorm-pytorch A PyTorch implementation of SoundStorm, Google DeepMind's efficient parallel audio generation model that applies MaskGiT-style masked gener… | 32 | 1545 | active |
| alibaba/FederatedScope FederatedScope is a comprehensive federated learning platform built on PyTorch with an event-driven architecture. It provides convenient us… | 23 | 1545 | active |
| facebookresearch/mmf MMF is a modular PyTorch framework for vision and language multimodal research from Facebook AI Research. It ships reference implementation… | 67 | 5637 | maintenance |
| ROMP ROMP is a PyTorch-based library and pip-installable API (simple-romp) for real-time monocular multi-person 3D human mesh recovery, implemen… | 23 | 1543 | stable |
| baijiuyang/collision-avoidance2 A computational modeling framework for pedestrian moving-obstacle avoidance behavior, implementing Fajen-style steering and Cohen-style avo… | 69 | 1541 | active |
| sobri909/LocoKit LocoKit is a Swift framework for iOS that combines Core Location and Core Motion recording with machine learning based activity type detect… | 25 | 1541 | active |
| lucidrains/DALLE-pytorch A PyTorch implementation/replication of OpenAI's DALL-E, a text-to-image transformer, including a discrete VAE and optional CLIP for rankin… | 23 | 5627 | maintenance |
| TransparentLC/realesrgan-gui A cross-platform graphical interface for the Real-ESRGAN AI image upscaler (with Real-CUGAN support), built in Python with tkinter. It wrap… | 55 | 1540 | active |
| SALMONN SALMONN is a family of open-source multi-modal large language models from ByteDance and Tsinghua that unify speech, audio, music, and video… | 70 | 1539 | active |
| KruxAI/ragbuilder RagBuilder is a Python toolkit that automatically builds an optimal, production-ready Retrieval-Augmented Generation (RAG) pipeline for you… | 65 | 1539 | active |
| Shawn-Shan/fawkes Fawkes is a privacy protection tool from University of Chicago researchers that adds imperceptible adversarial perturbations to photos to p… | 23 | 5614 | maintenance |
| ATH-MaaS/Marco-o1 Marco-o1 is an open large reasoning model from Alibaba International Digital Commerce, designed for o1-like chain-of-thought reasoning acro… | 58 | 1537 | active |
| skforecast Skforecast is a Python library for time series forecasting that turns any scikit-learn compatible estimator (LightGBM, XGBoost, CatBoost, K… | 92 | 1536 | active |
| Bumblebee Optimus is a Python library for agile data preparation that provides a unified API over pandas, Dask, cuDF, Dask-cuDF, Vaex, and PySpark. I… | 23 | 1535 | active |
| vanloctech/youwee Youwee is a cross-platform desktop GUI for yt-dlp built with Tauri, supporting video downloads from YouTube, TikTok, Instagram, and 1800+ s… | 80 | 1533 | active |
| SERL HIL-SERL is a Python library suite for training reinforcement learning policies for precise robotic manipulation using human demonstrations… | 41 | 1533 | active |
| allenzren/open-pi-zero An open-source re-implementation of the pi0 vision-language-action (VLA) model from Physical Intelligence, built on a pre-trained PaliGemma… | 24 | 1533 | active |
| Leela Zero Leela Zero is an open-source Go engine that reimplements AlphaGo Zero, combining Monte Carlo Tree Search with a deep residual convolutional… | 23 | 5591 | maintenance |
| azuwis/pianotrans A simple GUI wrapper and packaging for ByteDance's Piano Transcription with Pedals, a PyTorch system that converts piano audio recordings i… | 65 | 1532 | active |
| ComfyDeploy An open-source, Vercel-like deployment platform for ComfyUI workflows, letting teams share workflows, manage machines (on-premise or server… | 37 | 1532 | active |
| leggedrobotics/ocs2 OCS2 is a C++ toolbox for formulating and solving nonlinear optimal control problems, with an emphasis on real-time Model Predictive Contro… | 66 | 1529 | active |
| interpretml/DiCE DiCE is a Python library that generates diverse counterfactual explanations for any machine learning model, showing feature-perturbed versi… | 33 | 1529 | active |
| luciddreamer-cvlab/LucidDreamer LucidDreamer is the official implementation of a research method that generates 3D Gaussian Splatting scenes from text prompts, published i… | 66 | 1528 | active |
| ATH-MaaS/Ovis Ovis is an open-source Multimodal Large Language Model (MLLM) architecture that structurally aligns visual and textual embeddings, with rel… | 62 | 1528 | active |
| kyutai-labs/hibiki Hibiki is a decoder-only model for streaming (simultaneous) speech-to-speech translation, built on the multistream Moshi architecture. It p… | 63 | 1527 | active |
| WenmuZhou/PytorchOCR A PyTorch-based OCR toolkit that ports PaddleOCR models to PyTorch, supporting common text detection and recognition algorithms like the PP… | 56 | 1527 | active |
| cchen156/Learning-to-See-in-the-Dark TensorFlow implementation of 'Learning to See in the Dark' (CVPR 2018), a deep learning model that brightens very dark, short-exposure RAW … | 47 | 5567 | maintenance |
| NVIDIA/DLSS NVIDIA DLSS is the public SDK repository for NVIDIA's RTX Deep Learning Super Sampling, a neural network that boosts game frame rates and g… | 90 | 1526 | active |
| Soul-AILab/SoulX-FlashTalk SoulX-FlashTalk is a 14B audio-driven talking avatar model that streams infinite real-time video from a reference image and audio, achievin… | 55 | 1526 | active |
| Tencent/TFace TFace is a research platform from Tencent Youtu Lab for trusty face analysis, covering face recognition, face security (anti-spoofing), fac… | 54 | 1524 | active |
| meiqua/shape_based_matching A C++ library implementing Halcon-style shape-based matching (equivalent to LINE-MOD) using gradient orientation templates for robust 2D ob… | 32 | 1524 | active |
| geomstats/geomstats Geomstats is an open-source Python package for computations, statistics, and machine learning on manifolds with geometric structures. It pr… | 67 | 1523 | active |
| facebookexperimental/Robyn Robyn is Meta Marketing Science's open-source, semi-automated Marketing Mix Modeling (MMM) package available in R and Python. It uses ridge… | 48 | 1523 | active |
| keras-rl/keras-rl A Python library implementing state-of-the-art deep reinforcement learning algorithms (DQN, DDPG, SARSA, and more) that integrates seamless… | 23 | 5549 | maintenance |
| anishathalye/neural-style A Python command-line tool implementing the neural style transfer algorithm (Gatys et al.) in TensorFlow, applying the style of one image t… | 63 | 5547 | maintenance |
| sail-sg/envpool EnvPool is a C++-based batched environment pool with pybind11 bindings and a thread pool for high-performance parallel RL environment execu… | 96 | 1522 | active |
| hkchengrex/Tracking-Anything-with-DEVA DEVA is a decoupled video segmentation framework that combines task-specific image-level segmentation models with a universal bi-directiona… | 24 | 1522 | stable |
| o19s/elasticsearch-learning-to-rank An Elasticsearch plugin that integrates Learning to Rank (machine-learned relevance) into Elasticsearch. It stores feature query templates,… | 84 | 1521 | active |
| Yuliang-Liu/MonkeyOCRv2 MonkeyOCRv2 is a document-native vision encoder and visual-text foundation model for Document AI, pretrained on the 113M-image MonkeyDoc v2… | 56 | 1517 | active |
| Phantom-video/Phantom Phantom is a subject-consistent video generation model from ByteDance that preserves reference subject identity via cross-modal alignment. … | 36 | 1517 | active |
| NVlabs/describe-anything Describe Anything Model (DAM) is a vision-language model that generates detailed descriptions of user-specified regions in images and video… | 29 | 1517 | active |
| mattmireles/gemma-tuner-multimodal A Python tool for LoRA fine-tuning of Gemma 4 and 3n models on text, images, and audio using Apple Silicon's Metal Performance Shaders. It … | 60 | 1516 | active |
| mbzuai-oryx/Video-ChatGPT Video-ChatGPT is a video conversation model that combines large language models with a pretrained visual encoder adapted for spatiotemporal… | 72 | 1513 | active |
| hustvl/DiffusionDrive DiffusionDrive is a truncated diffusion model for real-time end-to-end autonomous driving, released as the official PyTorch implementation … | 41 | 1513 | active |
| k2-fsa/icefall Icefall is a collection of speech recognition (ASR) and TTS training recipes built on the k2 and lhotse libraries, implemented in Python wi… | 67 | 1512 | active |
| open-thought/reasoning-gym Reasoning Gym is a Python library of procedural dataset generators and algorithmically verifiable reasoning environments for training LLMs … | 60 | 1512 | active |
| microsoft/ai-dev-gallery AI Dev Gallery is a Windows application from Microsoft that lets developers explore over 25 interactive samples powered by local AI models … | 67 | 1510 | active |
| imoneoi/openchat OpenChat is a library of open-source large language models fine-tuned with C-RLFT, an offline reinforcement learning strategy that learns f… | 21 | 5485 | maintenance |
| sepandhaghighi/pycm PyCM is a Python library for computing multi-class confusion matrices and a wide range of per-class and overall evaluation statistics. It a… | 77 | 1507 | stable |
| charlesq34/pointnet Reference implementation of PointNet, a neural network architecture that directly consumes unordered 3D point clouds for classification and… | 32 | 5482 | maintenance |
| wb14123/seq2seq-couplet A deep learning project that generates Chinese couplets (对联) using a seq2seq model built with TensorFlow. It includes training scripts, a w… | 32 | 5480 | maintenance |
| czczup/ViT-Adapter Official PyTorch implementation of ViT-Adapter, an ICLR 2023 Spotlight paper introducing a pre-training-free adapter that lets plain Vision… | 30 | 1506 | stable |
| graphdeco-inria/diff-gaussian-rasterization A CUDA-based differentiable rasterization engine for 3D Gaussian Splatting, used in the SIGGRAPH 2023 paper '3D Gaussian Splatting for Real… | 29 | 1506 | stable |
| Flashlight wav2letter++ is Facebook AI Research's end-to-end automatic speech recognition (ASR) toolkit written in C++. It has been consolidated into … | 67 | 5476 | maintenance |
| piddnad/DDColor DDColor is the official PyTorch implementation of an ICCV 2023 paper on photo-realistic automatic image colorization using dual decoders an… | 56 | 1502 | active |
| TencentARC/MotionCtrl MotionCtrl is the official implementation of a SIGGRAPH 2024 paper providing a unified and flexible motion controller for video generation … | 27 | 1500 | active |
| CUT3R/CUT3R CUT3R is the official PyTorch implementation of 'Continuous 3D Perception Model with Persistent State' (CVPR 2025 Oral), a stateful recurre… | 35 | 1499 | active |
| open-mmlab/mmengine MMEngine is the foundational training engine library for OpenMMLab projects, providing a unified training loop, config system, registry, ho… | 68 | 1496 | active |
| NVlabs/Fast-FoundationStereo Fast-FoundationStereo is NVIDIA's official PyTorch implementation of a real-time zero-shot stereo matching model family, accepted to CVPR 2… | 59 | 1496 | active |
| pygod-team/pygod PyGOD is a Python library for graph outlier detection (anomaly detection) built on PyTorch and PyTorch Geometric. It provides 10+ graph-bas… | 23 | 1495 | active |
| acados/acados acados is a modular C library providing fast embedded solvers for nonlinear optimal control problems, designed for real-time applications l… | 91 | 1493 | active |
| ucb-bar/gemmini Gemmini is Berkeley's open-source generator for parameterizable systolic-array DNN hardware accelerators, written in Chisel (Scala) and int… | 66 | 1493 | active |
| Aceinna/gnss-ins-sim A Python library for simulating GNSS/INS integrated navigation systems. It generates reference trajectories and synthetic IMU, GPS, odomete… | 23 | 1493 | active |
| ibab/tensorflow-wavenet A TensorFlow implementation of DeepMind's WaveNet generative neural network architecture for raw audio waveform generation. It provides tra… | 32 | 5422 | maintenance |
| bert4keras A lightweight, clean reimplementation of BERT and other transformer models (RoBERTa, ALBERT, T5, GPT, ELECTRA, NEZHA) for Keras/tf.keras. I… | 23 | 5409 | maintenance |
| jonaswinkler/paperless-ng Paperless-ng is a self-hosted document management system that scans, OCRs, indexes, and archives physical documents with full-text search a… | 10 | 5408 | maintenance |
| google-deepmind/graph_nets DeepMind's library for building graph networks (graph neural networks) in TensorFlow and Sonnet, based on the 'Relational inductive biases,… | 32 | 5407 | maintenance |
| ModelOriented/DALEX DALEX (moDel Agnostic Language for Exploration and eXplanation) is a library for exploring, explaining, and visualizing the behavior of pre… | 61 | 1489 | active |
| MaxHalford/prince Prince is a Python library for multivariate exploratory data analysis, implementing PCA, CA, MCA, MFA, FAMD, GPA, and PGA with a scikit-lea… | 77 | 1486 | stable |
| reczoo/FuxiCTR FuxiCTR is an open-source Python library for click-through rate (CTR) prediction built on PyTorch and TensorFlow. It offers a configurable,… | 75 | 1481 | active |
| fan-ziqi/rl_sar A C++ framework for simulation verification and physical deployment of reinforcement learning policies for robots, supporting quadruped, wh… | 73 | 1481 | active |
| SciSharp/NumSharp NumSharp is a .NET library providing NumPy-shaped N-dimensional arrays with broadcasting, slicing views, dtype-aware math, and runtime-gene… | 92 | 1480 | active |
| alxndrTL/mamba.py A simple, readable pure-PyTorch (plus MLX) implementation of the Mamba state-space model architecture with a parallel scan for efficient tr… | 50 | 1480 | active |
| autonomousvision/mip-splatting Mip-Splatting is a research implementation of alias-free 3D Gaussian Splatting, introducing a 3D smoothing filter and 2D Mip filter to elim… | 27 | 1480 | active |
| LLM Vision LLM Vision is a Home Assistant integration (installed via HACS) that uses multimodal large language models to analyze images, videos, live … | 88 | 1479 | active |
| sb-ai-lab/LightAutoML LightAutoML (LAMA) is a Python framework for automatic machine learning model creation (AutoML) supporting tabular, time series, image, and… | 73 | 1479 | active |
| salesforce/CodeTF CodeTF is a Python transformer library for code large language models, providing unified interfaces for training, fine-tuning, and inferenc… | 10 | 1478 | active |
| Nutlope/napkins Napkins.dev is an open-source web application that turns screenshots or wireframes of website designs into working React + Tailwind code us… | 60 | 1477 | active |
| sczhou/Upscale-A-Video Upscale-A-Video is a diffusion-based model for real-world video super-resolution that takes low-resolution videos and text prompts as input… | 27 | 1477 | active |
| MiroShark/MiroShark MiroShark is a universal swarm-intelligence engine that turns any document or scenario into a simulated world: it builds a Neo4j knowledge … | 60 | 1476 | active |
| mit-han-lab/torchsparse TorchSparse is a high-performance PyTorch library for sparse convolution on 3D point clouds, with optimized GPU kernels for both training a… | 23 | 1476 | active |
| rzru/nightingale Nightingale is an open-source karaoke application that turns any song in your music library into a karaoke track using neural networks. It … | 79 | 1475 | active |
| Lightning-AI/lightning-thunder Lightning Thunder is a source-to-source deep learning compiler for PyTorch that optimizes models for training and inference. It provides a … | 70 | 1475 | active |
| Om-Alve/smolGPT A minimal pure-PyTorch implementation for training small GPT-style LLMs from scratch, featuring flash attention, RMSNorm, SwiGLU, RoPE, and… | 21 | 1474 | active |
| hojonathanho/diffusion The official reference implementation of Denoising Diffusion Probabilistic Models (DDPM) from the 2020 paper by Jonathan Ho et al., written… | 32 | 5341 | maintenance |
| NVIDIA-ISAAC-ROS/isaac_ros_visual_slam Isaac ROS Visual SLAM is a ROS 2 package providing GPU-accelerated visual simultaneous localization and mapping (VSLAM) using stereo visual… | 92 | 1473 | active |
| refinery An open-source tool for scaling, assessing, and maintaining natural language training data, treating datasets like software artifacts. It p… | 23 | 1473 | active |
| LocoMuJoCo LocoMuJoCo is an imitation learning benchmark for whole-body locomotion control built on MuJoCo, featuring humanoid, quadruped, and musculo… | 70 | 1468 | active |
| jax-md/jax-md JAX MD is a Python library for molecular dynamics simulations built on JAX, making them hardware accelerated on CPU, GPU, and TPU and end-t… | 80 | 1465 | active |
| facebookresearch/MobileLLM Meta's training code for MobileLLM, a family of sub-billion parameter language models optimized for on-device use, published at ICML 2024. … | 56 | 1465 | active |
| FeiYull/TensorRT-Alpha A C++/CUDA library providing TensorRT-accelerated deployment for 30+ popular computer vision models including YOLOv3-v8, YOLOv8-Pose/Seg/Cl… | 100 | 1464 | active |
| do-mpc/do-mpc do-mpc is an open-source Python toolbox for robust model predictive control (MPC) and moving horizon estimation (MHE) of nonlinear systems.… | 91 | 1464 | active |
| Topdu/OpenOCR OpenOCR is an open-source Python toolkit for general OCR research and applications, covering text detection and recognition, formula and ta… | 58 | 1464 | active |