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 |
|---|---|---|---|
| py-why/causal-learn causal-learn is a Python library for causal discovery, implementing classical and state-of-the-art algorithms to recover causal structure f… | 87 | 1689 | active |
| intelligent-machine-learning/dlrover DLRover is an automatic distributed deep learning system that manages training of large AI models on Kubernetes and Ray clusters. It provid… | 80 | 1689 | active |
| pytorch-tabular/pytorch_tabular PyTorch Tabular is a Python library providing a unified, high-level interface for deep learning models on tabular data, built on PyTorch an… | 73 | 1686 | active |
| Anemll/Anemll ANEMLL is an open-source Python library and toolchain for converting and running large language models on the Apple Neural Engine (ANE) via… | 70 | 1686 | active |
| ckiplab/ckiptagger CkipTagger is a Python library implementing neural CKIP-style Chinese NLP tools for word segmentation, part-of-speech tagging, and named en… | 33 | 1686 | stable |
| takuseno/d3rlpy d3rlpy is a Python library for offline and online deep reinforcement learning built on PyTorch, offering state-of-the-art algorithms throug… | 45 | 1684 | active |
| kijai/ComfyUI-FramePackWrapper A ComfyUI custom node wrapper for FramePack, enabling next-frame video generation from images using HunyuanVideo-based diffusion models. It… | 44 | 1683 | active |
| bytedance/Sa2VA Sa2VA is a family of research models and codebases from ByteDance that combine SAM-2 with multimodal LLMs for pixel-level grounded understa… | 67 | 1682 | active |
| explosion/sense2vec A Python library implementing sense2vec, contextually-keyed word vectors that learn embeddings for multi-word phrases and part-of-speech-di… | 52 | 1681 | stable |
| google/vizier Open Source Vizier is a Python-based service and library for black-box and hyperparameter optimization, based on Google's internal Vizier t… | 79 | 1680 | active |
| turboderp-org/exllamav3 ExLlamaV3 is a Python library for fast quantization and inference of large language models on consumer-class GPUs, featuring the EXL3 quant… | 83 | 1678 | active |
| MgArcher/Text_select_captcha A PyTorch-based deep learning system that recognizes click-based (text-select) CAPTCHAs by detecting and ordering Chinese character positio… | 65 | 1678 | active |
| maziarraissi/PINNs The original reference implementation of Physics-Informed Neural Networks (PINNs), which train neural networks to solve and discover nonlin… | 58 | 6217 | maintenance |
| microsoft/Mage Mage is a family of lightweight 4B-parameter multimodal models from Microsoft, including Mage-VL for image and video understanding and Mage… | 55 | 1677 | active |
| opendatalab/labelU LabelU is an open-source multimodal data annotation platform supporting images, video, and audio with tools like bounding boxes, segmentati… | 89 | 1676 | active |
| gensyn-ai/rl-swarm RL Swarm is an open-source, permissionless peer-to-peer framework for running reinforcement learning training swarms over the internet, bui… | 49 | 1676 | active |
| elixir-nx/bumblebee Bumblebee is an Elixir library providing pre-trained neural network models built on Axon, with integration for downloading models from Hugg… | 94 | 1674 | active |
| deepsense-ai/ragbits Ragbits is a Python framework of modular building blocks for rapidly developing generative AI applications, covering LLM interaction, promp… | 68 | 1674 | active |
| koaning/drawdata A Python library providing notebook widgets for drawing datasets interactively, built on anywidget. Drawn data can be exported as lists, pa… | 67 | 1674 | active |
| PRML/PRMLT A self-contained MATLAB toolbox implementing the machine learning algorithms from Bishop's textbook 'Pattern Recognition and Machine Learni… | 23 | 6201 | maintenance |
| sunlabuiuc/PyHealth PyHealth is an open-source Python toolkit for clinical deep learning, unifying healthcare datasets (EHRs, physiological signals, medical im… | 95 | 1673 | active |
| intel/auto-round AutoRound is Intel's advanced quantization toolkit for LLMs and vision-language models, using sign-gradient descent to achieve high accurac… | 94 | 1672 | active |
| meyda/meyda Meyda is a JavaScript audio feature extraction library that computes standard signal features like RMS, spectral centroid, and spectral fla… | 23 | 1672 | stable |
| Tencent/AngelSlim AngelSlim is a Python toolkit from Tencent for compressing large language models and related architectures (VLMs, diffusion, audio models) … | 70 | 1669 | active |
| mit-han-lab/torchquantum TorchQuantum is a PyTorch-based framework for quantum computing simulation, supporting statevector and pulse-level simulation on GPUs with … | 60 | 1669 | active |
| tkarras/progressive_growing_of_gans Official TensorFlow implementation of the ICLR 2018 NVIDIA paper 'Progressive Growing of GANs', which trains generators and discriminators … | 32 | 6176 | maintenance |
| apple/ml-mobileclip Official implementation of Apple's MobileCLIP and MobileCLIP2, fast image-text contrastive (CLIP-style) models optimized for mobile latency… | 68 | 1664 | active |
| IPAdapter plus A ComfyUI custom node implementation of IPAdapter models for image-to-image conditioning in Stable Diffusion workflows. It transfers the su… | 32 | 6142 | maintenance |
| thtrieu/darkflow Darkflow is a Python library that translates Darknet's YOLO neural network definitions to TensorFlow, enabling real-time object detection a… | 32 | 6139 | maintenance |
| Data-Centric-AI-Community/fg-data-synthetic A Python library for generating synthetic tabular and time-series data using state-of-the-art generative models such as GANs (including CTG… | 87 | 1659 | active |
| googlecolab/google-colab-cli A Python-based command-line interface for Google Colab that lets users provision CPU, GPU, and TPU runtimes, execute local scripts and note… | 60 | 1656 | active |
| Robbyant/lingbot-depth LingBot-Depth is a PyTorch-based model and toolkit for masked depth modeling that transforms incomplete, noisy depth sensor data into metri… | 53 | 1655 | active |
| XueZeyue/DanceGRPO Official implementation of DanceGRPO, a framework applying Group Relative Policy Optimization (GRPO) to fine-tune visual generation models … | 37 | 1654 | active |
| skrub-data/skrub skrub is a Python library that facilitates machine learning with dataframes, providing preprocessing, encoding, and wrangling tools for tab… | 92 | 1653 | active |
| mhamilton723/FeatUp FeatUp is a model-agnostic framework that upsamples the spatial resolution of deep neural network features by 16-32x without changing their… | 17 | 1653 | active |
| huggingface/aisheets Hugging Face AI Sheets is an open-source no-code web application for building, enriching, and transforming datasets using AI models. It can… | 64 | 1652 | active |
| GoMLX GoMLX is an accelerated machine learning and math framework for Go, comparable to PyTorch/JAX/TensorFlow. It offers differentiable operator… | 96 | 1651 | active |
| meta-llama/synthetic-data-kit A Python CLI tool from Meta for generating high-quality synthetic datasets to fine-tune LLMs. It follows a four-stage pipeline (ingest, cre… | 39 | 1650 | active |
| kotaro-kinoshita/yomitoku YomiToku is an AI-powered document image analysis engine specialized for Japanese, providing full-text OCR, layout analysis, table structur… | 88 | 1649 | active |
| InterDigitalInc/CompressAI CompressAI is a PyTorch library and evaluation platform for end-to-end learned data compression research. It provides custom layers, entrop… | 76 | 1648 | active |
| opendilab/LightZero LightZero is a lightweight, open-source Python toolkit that unifies Monte Carlo Tree Search (MCTS) with deep reinforcement learning, implem… | 66 | 1647 | active |
| bowang-lab/scGPT scGPT is a foundation model for single-cell multi-omics built with a generative transformer, providing pretrained checkpoints and a Python … | 64 | 1646 | active |
| bytedance/DreamO DreamO is the official PyTorch implementation of a unified framework for image customization, built on FLUX diffusion models. It supports t… | 33 | 1646 | active |
| JIA-Lab-research/ControlNeXt ControlNeXt is the official implementation of a controllable generation method for images and videos, built on Stable Diffusion XL, Stable … | 24 | 1646 | active |
| facebookincubator/submitit Submitit is a lightweight Python library for submitting Python functions as jobs to a Slurm cluster, wrapping submission and providing acce… | 47 | 1642 | active |
| Lucchetto/SuperImage SuperImage is an Android application that upscales low-resolution images using a Real-ESRGAN neural network running on-device via the MNN d… | 10 | 1641 | active |
| AnswerDotAI/rerankers A lightweight, dependency-free Python library providing a unified API for using all common reranking and cross-encoder models. It lets deve… | 39 | 1640 | active |
| pixeltable/pixeltable Pixeltable is a Python library providing declarative, incremental data infrastructure for multimodal AI applications, unifying storage of i… | 94 | 1639 | active |
| Higgsfield Higgsfield is an open-source GPU orchestration and machine learning framework for fault-tolerant, distributed training of very large models… | 67 | 6047 | maintenance |
| yoshitomo-matsubara/torchdistill torchdistill is a modular, configuration-driven PyTorch framework for knowledge distillation and general deep learning experiments, requiri… | 83 | 1633 | active |
| jeffffffli/HybrIK HybrIK is the official PyTorch implementation of a hybrid analytical-neural inverse kinematics method for 3D human pose and shape estimatio… | 23 | 1632 | stable |
| facebookresearch/DomainBed DomainBed is a PyTorch suite containing benchmark datasets and algorithms for domain generalization research. It implements many algorithms… | 10 | 1632 | active |
| mlcommons/inference Reference implementations of the MLPerf Inference benchmark suite, the industry-standard benchmarks for measuring how fast ML systems run m… | 79 | 1631 | active |
| Snorkel Snorkel is a Python library for programmatically building and managing training data using weak supervision, letting users write labeling f… | 65 | 6010 | maintenance |
| ZhangGe6/onnx-modifier onnx-modifier is a web-based visual editor for ONNX neural network models, built on Netron for graph visualization and Flask as the backend… | 69 | 1629 | active |
| paperswithcode/ai-deadlines A web application that displays countdown timers to submission deadlines for top-tier AI, machine learning, computer vision, NLP, and robot… | 32 | 6007 | maintenance |
| scikit-learn-contrib/boruta_py A Python implementation of the Boruta all-relevant feature selection method, built on scikit-learn. It identifies all features carrying pre… | 42 | 1628 | active |
| csinva/imodels A Python package providing scikit-learn-compatible implementations of interpretable machine-learning models such as rule lists, RuleFit, an… | 98 | 1627 | active |
| andrew-waters/orchard Orchard is a native Swift macOS GUI for Apple's container tooling, serving as a Docker Desktop replacement for managing containers, Linux V… | 85 | 1627 | active |
| KEV0143/Adaptive-forecasting-of-electricity-consumption-24-168-720-h-with-load-regime-conditioning An open-source tool for adaptive forecasting of hourly electricity consumption over 24, 168, and 720 hour horizons, using load regime condi… | 56 | 1627 | active |
| Meta-Harness Meta-Harness is a Python framework for automated end-to-end search over task-specific model harnesses — the code surrounding a fixed base L… | 57 | 1625 | active |
| pq-yang/MatAnyone MatAnyone is a CVPR 2025 human video matting framework that extracts alpha mattes of target people from video using consistent memory propa… | 51 | 1625 | active |
| pavlin-policar/openTSNE openTSNE is a modular Python library implementing t-Distributed Stochastic Neighbor Embedding (t-SNE) for visualizing high-dimensional data… | 65 | 1624 | stable |
| NVIDIA/RULER RULER is a benchmark framework from NVIDIA that generates synthetic test examples to measure the true effective context size of long-contex… | 64 | 1619 | active |
| aiwaves-cn/agents Agents 2.0 is a Python framework for data-centric, self-evolving autonomous language agents based on symbolic learning. It adapts neural-ne… | 29 | 5964 | maintenance |
| waybarrios/vllm-mlx vllm-mlx is a vLLM-style LLM inference server for Apple Silicon Macs built on native MLX, exposing both OpenAI and Anthropic compatible API… | 82 | 1618 | active |
| Forsy Agent Apprenticeship is an open ecosystem and CLI tool where AI agents complete real-world tasks through iterative workflow loops, are eval… | 57 | 1617 | active |
| PKU-Alignment/safe-rlhf Beaver is a modular open-source framework from Peking University for training large language models with SFT, RLHF, and Safe RLHF (constrai… | 50 | 1617 | active |
| open-gigaai/giga-world-0 GigaWorld-0 is a unified world model framework that acts as a data engine for Vision-Language-Action (VLA) learning in embodied AI. It comb… | 38 | 1616 | active |
| NVlabs/sionna Sionna is an open-source, GPU-accelerated, differentiable Python library from NVIDIA for research on communication systems. It comprises Si… | 91 | 1615 | active |
| pypose/pypose PyPose is a PyTorch-based Python library for differentiable robotics on manifolds, combining deep perceptual models with physics-based opti… | 89 | 1615 | active |
| Soprano Soprano is an ultra-lightweight 80M-parameter text-to-speech model and Python library for fast, expressive, high-fidelity speech synthesis … | 41 | 1614 | active |
| PaddlePaddle/PaddleSlim PaddleSlim is an open-source library built on PaddlePaddle for deep learning model compression and architecture search. It provides low-bit… | 46 | 1612 | active |
| gorgonia/gorgonia Gorgonia is a Go library for machine learning that lets you define and evaluate mathematical equations over multidimensional arrays using a… | 23 | 5927 | maintenance |
| JustGlowing/minisom MiniSom is a minimalistic, NumPy-based Python implementation of Self Organizing Maps (SOM), a type of neural network for unsupervised learn… | 78 | 1609 | stable |
| Chainer Chainer is a Python-based deep learning framework that pioneered the define-by-run approach with dynamic computational graphs and automatic… | 23 | 5919 | maintenance |
| dmlc/gluon-cv GluonCV is a deep learning toolkit providing state-of-the-art computer vision model implementations with 170+ pre-trained models. It suppor… | 23 | 5914 | maintenance |
| autonomousvision/transfuser Official PyTorch implementation of TransFuser, a transformer-based multi-modal sensor fusion model for end-to-end autonomous driving, publi… | 75 | 1606 | stable |
| Babyhamsta/Aimmy Aimmy is a universal AI-based aim alignment mechanism (aim assist) for gamers with impairments, built in C# using YOLOv8 models run via ONN… | 76 | 1605 | active |
| Godot RL Agents Godot RL Agents is an open-source Python package that bridges games built in the Godot Engine with reinforcement learning algorithms, enabl… | 60 | 1605 | active |
| Apache OpenNLP Apache OpenNLP is a machine learning based Java toolkit for processing natural language text, supporting tasks like tokenization, sentence … | 97 | 1604 | stable |
| ml4a/ml4a ml4a is a Python library and collection of Jupyter notebooks for making art with machine learning. It wraps popular deep learning models li… | 32 | 1604 | active |
| meituan/YOLOv6 YOLOv6 is a single-stage object detection framework implemented in PyTorch, designed for industrial applications with a family of pretraine… | 23 | 5892 | maintenance |
| spotify/voyager Voyager is an in-memory approximate nearest-neighbor search library implementing the HNSW algorithm, with bindings for Python and Java (and… | 64 | 1602 | stable |
| google/uncertainty-baselines A library of high-quality, minimal-dependency implementations of standard and state-of-the-art uncertainty and robustness methods for deep … | 77 | 1601 | active |
| time-series-foundation-models/lag-llama Lag-Llama is the first open-source foundation model for probabilistic time series forecasting, built on a transformer architecture. It prov… | 34 | 1600 | active |
| SalesforceAIResearch/uni2ts Uni2TS is a PyTorch library for unified pre-training, fine-tuning, inference, and evaluation of universal time series forecasting transform… | 54 | 1599 | active |
| alibaba/Pai-Megatron-Patch Pai-Megatron-Patch is an open-source deep learning training toolkit from Alibaba Cloud for large-scale training and inference of LLMs and V… | 50 | 1599 | active |
| Tencent-Hunyuan/HunyuanWorld-Voyager HunyuanWorld-Voyager is a video diffusion framework from Tencent Hunyuan that generates world-consistent RGBD video and 3D point-cloud sequ… | 49 | 1599 | active |
| bilibili/ailab Bilibili's AI lab repository, best known for Real-CUGAN, a deep learning model for anime image super-resolution (upscaling). It provides pr… | 23 | 5877 | maintenance |
| SonyResearch/micro_diffusion Official implementation of Sony Research's micro-budget approach to training large-scale text-to-image diffusion transformer models from sc… | 24 | 1598 | active |
| scikit-learn-contrib/MAPIE MAPIE is a scikit-learn-compatible Python library for quantifying uncertainty in machine learning models via conformal prediction. It compu… | 90 | 1596 | active |
| databricks/megablocks MegaBlocks is a lightweight Python library for efficient training of mixture-of-experts (MoE) models, built around its dropless-MoE (dMoE) … | 58 | 1596 | active |
| pytorch/FBGEMM FBGEMM is a collection of highly optimized low-precision matrix multiplication and convolution kernels for server-side deep learning infere… | 91 | 1595 | active |
| samim23/polymath Polymath is a Python CLI tool that uses machine learning to convert any music library into a searchable music production sample-library. It… | 31 | 1595 | active |
| zju3dv/EasyVolcap EasyVolcap is a PyTorch-based library for accelerating neural volumetric video research, covering volumetric video capture, reconstruction,… | 69 | 1594 | active |
| real-stanford/universal_manipulation_interface Universal Manipulation Interface (UMI) is a data collection and policy learning framework that transfers in-the-wild human demonstrations i… | 60 | 1594 | active |
| byjlw/video-analyzer A Python CLI tool that analyzes videos by extracting key frames, transcribing audio with Whisper, and describing content using vision LLMs … | 54 | 1593 | active |
| facebookresearch/fast3r Fast3R is the official PyTorch implementation of a CVPR 2025 model from Meta FAIR that reconstructs 3D scenes and estimates camera poses fr… | 10 | 1592 | active |
| microsoft/Semi-supervised-learning USB (Unified Semi-supervised learning Benchmark) is a PyTorch-based codebase from Microsoft for semi-supervised learning across computer vi… | 10 | 1591 | active |