Open-source machine-learning software
4007 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.
4007 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| meta-pytorch/botorch BoTorch is a Python library for Bayesian Optimization research built on top of PyTorch, part of the PyTorch ecosystem. It provides modular,… | 89 | 3632 | active |
| ploomber/ploomber Ploomber is a Python framework for building maintainable data pipelines from scripts and Jupyter notebooks, with iterative local developmen… | 10 | 3621 | active |
| facebookresearch/detr DETR is Facebook Research's PyTorch implementation of Detection Transformer, an end-to-end object detection model that replaces hand-crafte… | 10 | 15352 | maintenance |
| nv-tlabs/kimodo Kimodo is NVIDIA's official implementation of a kinematic motion diffusion model trained on 700 hours of motion capture data to generate hi… | 58 | 3615 | active |
| edtechre/pybroker PyBroker is a Python framework for developing algorithmic trading strategies with a focus on machine learning. It provides a fast NumPy/Num… | 92 | 3611 | active |
| Albumentations Albumentations is a fast, flexible Python image augmentation library for computer vision, supporting images, masks, bounding boxes, keypoin… | 10 | 15309 | maintenance |
| OpenArm OpenArm is a fully open-source 7DOF humanoid robot arm (hardware, firmware, and software) designed for physical AI research and deployment … | 67 | 3605 | active |
| Qwen3-ASR Qwen3-ASR is a family of open-source speech recognition models from Alibaba's Qwen team, supporting ASR and language identification across … | 52 | 3605 | active |
| pytorch/rl TorchRL is a PyTorch-native library of composable primitives for building reinforcement learning systems, including environments, policies,… | 99 | 3603 | active |
| truera/trulens TruLens is a Python library for evaluating and tracking LLM-based applications and AI agents. It provides OpenTelemetry-native tracing of e… | 99 | 3601 | active |
| RDKit RDKit is an open-source cheminformatics toolkit written in C++ with Python, Java, C#, and JavaScript wrappers. It provides core data struct… | 95 | 3597 | stable |
| RUC-NLPIR/FlashRAG FlashRAG is a modular Python toolkit for reproducing and developing Retrieval-Augmented Generation (RAG) research, bundling 36 pre-processe… | 88 | 3590 | active |
| NVIDIA/TransformerEngine Transformer Engine is an NVIDIA library for accelerating Transformer model training and inference on NVIDIA GPUs using low-precision format… | 99 | 3576 | active |
| sdv-dev/SDV SDV (Synthetic Data Vault) is a Python library for generating synthetic tabular data using machine learning models ranging from GaussianCop… | 98 | 3572 | stable |
| microsoft/hummingbird Hummingbird is a Microsoft library that compiles trained traditional ML models (decision trees, random forests, XGBoost, LightGBM) into ten… | 65 | 3554 | active |
| Vision Transformer Google Research's official Jax/Flax codebase for training large-scale vision models such as Vision Transformer, SigLIP, MLP-Mixer, and LiT … | 39 | 3551 | active |
| pomegranate Pomegranate is a Python library for fast and flexible probabilistic modeling, built on PyTorch. It supports distributions, mixture models, … | 23 | 3546 | active |
| MNE-Python MNE-Python is an open-source Python library for exploring, visualizing, and analyzing human neurophysiological data such as MEG, EEG, sEEG,… | 93 | 3540 | stable |
| fnnx-ai/scikit-llm Scikit-LLM is a Python library that integrates large language models like GPT into scikit-learn workflows as estimators. It enables tasks s… | 72 | 3533 | active |
| Farama-Foundation/PettingZoo PettingZoo is a Python library providing a standard API for multi-agent reinforcement learning environments, analogous to Gymnasium but for… | 87 | 3532 | active |
| Anttwo/SuGaR SuGaR is the official PyTorch implementation of a CVPR 2024 method that extracts accurate, editable meshes from 3D Gaussian Splatting recon… | 27 | 3525 | active |
| google-deepmind/alphafold Open-source implementation of the AlphaFold 2 inference pipeline for predicting protein structures from amino acid sequences, including Alp… | 54 | 14887 | maintenance |
| helblazer811/ManimML ManimML is a Python library built on the Manim Community Library for creating animations and visualizations of machine learning concepts li… | 23 | 3517 | active |
| visionml/pytracking PyTracking is a PyTorch-based framework for visual object tracking and video object segmentation, providing official implementations of tra… | 23 | 3516 | active |
| google-ai-edge/LiteRT LiteRT is Google's successor to TensorFlow Lite, an on-device runtime for high-performance ML and GenAI inference on edge platforms. It pro… | 86 | 3514 | active |
| ob-f/OpenBot OpenBot is an open-source project that turns Android smartphones into the brains of low-cost robots, paired with a ~$50 electric vehicle bo… | 64 | 3505 | active |
| huggingface/optimum Optimum is a Hugging Face library that extends Transformers, Diffusers, timm, and Sentence Transformers with hardware-specific optimization… | 91 | 3503 | active |
| imgaug imgaug is a Python library for augmenting images in machine learning experiments, converting a small set of input images into a much larger… | 23 | 14744 | maintenance |
| Mozilla Common Voice Mozilla Common Voice is a web platform for crowdsourcing voice donations to build public-domain speech datasets for training voice recognit… | 99 | 3493 | active |
| vietanhdev/anylabeling AnyLabeling is a desktop image annotation tool that combines LabelImg/Labelme-style manual labeling with AI-assisted auto-labeling. It runs… | 92 | 3491 | active |
| borisdayma/dalle-mini DALL·E Mini is a Python library and model that generates images from a text prompt, available via pip and hosted on Hugging Face Model Hub.… | 23 | 14724 | maintenance |
| facebookresearch/ijepa Official PyTorch implementation of I-JEPA, a self-supervised learning method that predicts latent representations of image regions from oth… | 10 | 3488 | active |
| POSTECH-CVLab/PyTorch-StudioGAN PyTorch-StudioGAN is a PyTorch library providing unified implementations of representative GAN architectures (BigGAN, StyleGAN2/3, etc.) fo… | 23 | 3483 | stable |
| Docta-ai/docta Docta is a Python library for data-centric AI that diagnoses and fixes issues in datasets such as label errors, noise, and misannotations. … | 66 | 3479 | active |
| PennyLaneAI/pennylane PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. It lets users build an… | 88 | 3476 | stable |
| TextAttack TextAttack is a Python framework for generating adversarial examples against NLP models, as well as for data augmentation and model trainin… | 80 | 3471 | active |
| lakehq/sail Sail is an open-source, Rust-native multimodal compute engine that serves as a drop-in replacement for Apache Spark, unifying batch process… | 94 | 3465 | active |
| shenweichen/DeepCTR-Torch DeepCTR-Torch is a PyTorch library providing easy-to-use, modular, and extendable implementations of deep-learning-based CTR (click-through… | 71 | 3460 | active |
| aiming-lab/MetaClaw MetaClaw is a continual meta-learning framework that lets an LLM agent evolve from real conversations, combining skill synthesis from failu… | 63 | 3460 | active |
| PaddlePaddle/PARL PARL is a flexible, high-performance reinforcement learning framework built on PaddlePaddle, providing Model/Algorithm/Agent abstractions a… | 38 | 3457 | active |
| MTEB MTEB (Massive Text Embedding Benchmark) is a Python library and CLI for benchmarking embedding models across 1000+ tasks, languages, and mo… | 95 | 3455 | active |
| modelscope/evalscope EvalScope is a Python framework from the ModelScope community for evaluating large language models, vision-language models, embedding model… | 91 | 3454 | active |
| XinJingHao/DRL-Pytorch A unified PyTorch implementation collection of popular deep reinforcement learning algorithms including DQN variants, PPO, DDPG, TD3, SAC, … | 40 | 3447 | active |
| aqlaboratory/openfold OpenFold is a faithful, trainable PyTorch reproduction of DeepMind's AlphaFold 2 for protein structure prediction. It is memory-efficient a… | 45 | 3435 | active |
| alibaba/ROLL ROLL is an open-source reinforcement learning library from Alibaba for training large language models at scale, supporting algorithms like … | 82 | 3429 | active |
| NVlabs/stylegan The official TensorFlow implementation of StyleGAN, NVIDIA's style-based generator architecture for generative adversarial networks from th… | 32 | 14414 | maintenance |
| AlexanderKoch-Koch/low_cost_robot An open-source project with files and Python control code for building a ~$250 low-cost robot arm using Dynamixel XL430 and XL330 servo mot… | 26 | 3423 | active |
| RKNN-Toolkit2 RKNN-Toolkit2 is Rockchip's SDK for converting trained neural network models into RKNN format and deploying them on Rockchip NPU chips like… | 33 | 3419 | active |
| microsoft/nni NNI (Neural Network Intelligence) is an open-source AutoML toolkit from Microsoft that automates hyperparameter tuning, neural architecture… | 10 | 14367 | maintenance |
| davidsandberg/facenet A TensorFlow implementation of the FaceNet face recognizer that generates 128-dimensional face embeddings, including face detection via MTC… | 32 | 14357 | maintenance |
| superlinked/sie SIE (Superlinked Inference Engine) is an open-source, self-hosted inference server and production cluster that serves 100+ open models (emb… | 95 | 3414 | active |
| IQA-PyTorch A pure Python/PyTorch toolbox for image quality assessment (IQA) providing GPU-accelerated reimplementations of many full-reference and no-… | 82 | 3408 | active |
| HKUDS/VideoRAG VideoRAG is a retrieval-augmented generation framework for chatting with and understanding extremely long-context videos, using a dual-chan… | 50 | 3401 | active |
| NovaSky-AI/SkyThought SkyThought is the open-source repository behind Sky-T1, a family of reasoning language models trained for under $450, including training sc… | 22 | 3401 | active |
| NVIDIA/cuda-python NVIDIA's official Python package for accessing the CUDA platform, providing Pythonic APIs and low-level bindings to CUDA C APIs, parallel a… | 95 | 3400 | active |
| opengeos/geoai GeoAI is a Python package that integrates artificial intelligence with geospatial data analysis, built on PyTorch, Transformers, and segmen… | 89 | 3390 | active |
| huggingface/datatrove DataTrove is a Python library from Hugging Face for processing, filtering, and deduplicating large-scale text data through customizable pip… | 87 | 3390 | active |
| magenta/ddsp DDSP is a Python library of differentiable digital signal processing components (synthesizers, filters, waveshapers) that can be embedded i… | 68 | 3380 | active |
| CompVis/latent-diffusion The official research code and pretrained model zoo for Latent Diffusion Models (LDM), the paper behind Stable Diffusion, enabling high-res… | 32 | 14168 | maintenance |
| WongKinYiu/yolov7 Official PyTorch implementation of the YOLOv7 paper, a state-of-the-art real-time object detector with trainable bag-of-freebies techniques… | 23 | 14153 | maintenance |
| PhysicsNeMo NVIDIA PhysicsNeMo is an open-source Python deep-learning framework for building, training, fine-tuning, and inferring physics AI models us… | 93 | 3373 | active |
| libAudioFlux/audioFlux audioFlux is a C-based library with Python bindings for audio and music analysis and feature extraction. It supports dozens of time-frequen… | 50 | 3372 | active |
| VainF/Torch-Pruning Torch-Pruning is a PyTorch framework for structural neural network pruning based on the DepGraph algorithm from CVPR 2023. It automatically… | 44 | 3359 | active |
| shankarpandala/lazypredict Lazy Predict is a Python library that trains dozens of machine learning models with minimal code to quickly identify which algorithms perfo… | 86 | 3352 | active |
| Lagrange-Labs/deep-prove DeepProve is a Rust framework for generating zero-knowledge proofs of neural network inference, including end-to-end proving of full LLM fo… | 57 | 3352 | active |
| pgmpy pgmpy is a Python library for causal and probabilistic reasoning with graphical models such as Bayesian Networks, Dynamic Bayesian Networks… | 78 | 3338 | stable |
| Bytez-com/docs Bytez is a serverless model inference platform offering a unified API with a single key to run 175k+ open and closed-source AI models (LLMs… | 53 | 3329 | active |
| vladmandic/human Human is a JavaScript/TypeScript library built on TensorFlow.js that combines multiple ML models for 3D face detection and recognition, bod… | 44 | 3324 | active |
| LLM-as-a-Verifier A Python framework that uses LLM token logprobs to produce fine-grained verification scores for agent trajectories without additional train… | 59 | 3322 | active |
| microsoft/LoRA loralib is the official PyTorch implementation of LoRA (Low-Rank Adaptation), which fine-tunes large language models by injecting trainable… | 23 | 13868 | maintenance |
| 2D Gaussian Splatting Official implementation of 2D Gaussian Splatting (2DGS), a SIGGRAPH 2024 method that represents scenes as 2D oriented Gaussian disks for ge… | 66 | 3312 | stable |
| ikawrakow/ik_llama.cpp A fork of llama.cpp offering state-of-the-art quantization types and improved CPU/CUDA inference performance for large language models. It … | 60 | 3304 | active |
| mljar/mljar-supervised MLJAR AutoML is a Python package for automated machine learning on tabular data, providing feature engineering, hyperparameter tuning, mode… | 85 | 3294 | active |
| facebookresearch/tribev2 TRIBE v2 is a multimodal deep learning model from Meta AI that predicts fMRI brain responses to naturalistic video, audio, and text stimuli… | 51 | 3293 | active |
| PrismML-Eng/Bonsai-demo A demo repository for running PrismML's Bonsai family of 1-bit and ternary-quantized language models locally via llama.cpp and MLX. It prov… | 59 | 3286 | active |
| MAIF/shapash Shapash is a Python library that makes machine learning models interpretable and understandable through clear visualizations, a webapp, and… | 95 | 3267 | active |
| cocktailpeanut/fluxgym FluxGym is a simple web UI for training FLUX LoRA models with low VRAM support (12GB/16GB/20GB). It combines the AI-Toolkit Gradio frontend… | 62 | 3257 | active |
| LeelaChessZero/lc0 Lc0 is an open-source, UCI-compliant chess engine that plays chess using neural networks trained via AlphaZero-style self-play reinforcemen… | 74 | 3248 | active |
| determined-ai/determined Determined is an open-source deep learning platform that combines distributed training, hyperparameter tuning, experiment tracking, and GPU… | 36 | 3245 | active |
| Nerogar/OneTrainer OneTrainer is a GUI and CLI application for fine-tuning diffusion image models, supporting full fine-tuning, LoRA, and embeddings across ma… | 75 | 3243 | active |
| onnx/onnx-tensorrt A C++ parser library and backend that converts ONNX models into TensorRT engines for high-performance GPU inference. It is maintained by NV… | 91 | 3238 | active |
| Jittor/jittor Jittor is a high-performance deep learning framework from Tsinghua University based on just-in-time (JIT) compilation and meta-operators, w… | 68 | 3237 | active |
| Pointcept Pointcept is a PyTorch-based research codebase for point cloud perception, providing implementations of state-of-the-art 3D scene understan… | 74 | 3233 | active |
| VMamba VMamba is a PyTorch implementation of a visual state space model (SSM) vision backbone based on Mamba, featuring 2D Selective Scan (SS2D) f… | 18 | 3230 | active |
| facebookresearch/dinov2 PyTorch implementation and pretrained models for DINOv2, a self-supervised vision transformer method from Meta AI that learns robust visual… | 75 | 13422 | maintenance |
| qdrant/fastembed FastEmbed is a lightweight Python library for generating text embeddings using ONNX Runtime instead of PyTorch, requiring no GPU and minima… | 90 | 3216 | active |
| Captcha Solution A deep learning training tool for image CAPTCHA recognition built on TensorFlow, using CNN/ResNet/DenseNet backbones with GRU/LSTM recurren… | 52 | 3215 | active |
| Djdefrag/QualityScaler QualityScaler is a Windows GUI application that uses AI deep-learning models to upscale, enhance, and de-noise images and videos. It is wri… | 89 | 3207 | active |
| webdataset/webdataset A Python library providing a high-performance sequential I/O system based on tar-shard files for large-scale deep learning training, with s… | 49 | 3204 | stable |
| Arm NN Arm's Compute Library is a C++ collection of over 100 low-level machine learning and computer vision functions optimized for Arm Cortex-A/N… | 98 | 3197 | active |
| imbue-bit/AlphaGPT AlphaGPT is an open-source automated factor factory based on deep reinforcement learning for quantitative finance. It mines and generates a… | 52 | 3187 | active |
| Miles Miles is an open-source, enterprise-grade reinforcement learning framework for large-scale LLM and VLM post-training, forked from and co-ev… | 84 | 3185 | active |
| tslearn-team/tslearn tslearn is a Python machine learning library dedicated to time series analysis, built on numpy and compatible with scikit-learn. It provide… | 84 | 3183 | stable |
| tekaratzas/RustGPT A transformer-based large language model implemented entirely in pure Rust with no external ML frameworks, using only ndarray for matrix op… | 35 | 3175 | active |
| stemrollerapp/stemroller StemRoller is a free open-source desktop app that separates vocals, drums, bass, and other stems from any song with a single click, using F… | 77 | 3173 | active |
| parrt/dtreeviz dtreeviz is a Python library for visualizing decision trees and interpreting tree-based machine learning models. It supports scikit-learn, … | 63 | 3157 | active |
| scikit-learn-contrib/hdbscan A high-performance Python implementation of the HDBSCAN hierarchical density-based clustering algorithm, part of the scikit-learn-contrib e… | 88 | 3154 | stable |
| modelscope/3D-Speaker 3D-Speaker is an open-source Python toolkit for single- and multi-modal speaker verification, speaker recognition, and speaker diarization,… | 53 | 3148 | active |
| dexmal/dexbotic Dexbotic is an open-source PyTorch-based toolbox for developing Vision-Language-Action (VLA) models for embodied intelligence. It unifies p… | 70 | 3144 | active |
| z-x-yang/Segment-and-Track-Anything An open-source pipeline (SAM-Track) that segments and tracks arbitrary objects in videos using the Segment Anything Model for key-frame seg… | 58 | 3140 | active |