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 |
|---|---|---|---|
| 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 |
| houseofsecrets/SdPaint A Python desktop application that provides a painting canvas where each stroke is sent to a Stable Diffusion (automatic1111) API and the ge… | 21 | 1617 | active |
| IDEA-Research/Rex-Omni Rex-Omni is a 3B-parameter multimodal large language model that unifies object detection, OCR, pointing, keypoint detection, and visual pro… | 45 | 1616 | 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 |
| 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 |
| OpenGVLab/LLaMA-Adapter Official implementation of LLaMA-Adapter and its V2 successor, a parameter-efficient fine-tuning method that adapts LLaMA models to follow … | 21 | 5913 | maintenance |
| MoonshotAI/Kimi-Linear Kimi Linear is a hybrid linear attention architecture (Kimi Delta Attention, based on Gated DeltaNet) released by Moonshot AI with 48B-para… | 37 | 1607 | active |
| 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 |
| AlibabaResearch/DAMO-ConvAI The official codebase for Alibaba DAMO Academy's Conversational AI research, containing implementations of models from their published pape… | 75 | 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 |
| Tencent-Hunyuan/HY-WorldPlay HY-WorldPlay (HY-World 1.5) is Tencent Hunyuan's open-source framework for interactive 3D world modeling, generating explorable 3D scenes f… | 52 | 1604 | active |
| 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 |
| dfm/emcee emcee is a stable, well-tested Python implementation of the affine-invariant ensemble sampler for Markov chain Monte Carlo (MCMC) proposed … | 67 | 1603 | stable |
| 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 |
| Drexubery/ViewCrafter ViewCrafter is a research codebase that uses video diffusion models to synthesize high-fidelity novel views of scenes from a single or spar… | 45 | 1602 | active |
| THUDM/WebGLM WebGLM is an efficient web-enhanced question answering system (KDD 2023) that combines a large language model with web search retrieval and… | 32 | 1602 | active |
| 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 |
| vibevoice-community/VibeVoice VibeVoice is a community-maintained fork of Microsoft's long-form conversational text-to-speech model, generating expressive multi-speaker … | 59 | 1601 | active |
| xLLM-AI/xllm xLLM is a high-performance C++ inference engine for LLM, VLM, DiT and recommendation models, optimized for heterogeneous AI accelerators su… | 80 | 1600 | 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 |
| XPixelGroup/HAT HAT (Hybrid Attention Transformer) is a PyTorch implementation of a state-of-the-art transformer model for image super-resolution and resto… | 32 | 1598 | stable |
| X-LANCE/AniTalker AniTalker is the official PyTorch implementation of an ACM MM 2024 paper that animates a single static portrait into a vivid talking-face v… | 25 | 1598 | active |
| 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 |
| pipecat-ai/smart-turn Smart Turn is an open-source, audio-native turn detection model that decides when a voice agent should respond to human speech, using proso… | 46 | 1597 | 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 |
| 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 |
| capitalone/DataProfiler DataProfiler is a Python library that loads CSV, AVRO, Parquet, JSON, text, or URL data into a pandas-compatible DataFrame and profiles it … | 91 | 1591 | 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 |
| Tencent/DepthCrafter DepthCrafter is a diffusion-based video depth estimation model from Tencent AI Lab that generates temporally consistent long depth sequence… | 34 | 1590 | active |
| FoundationVision/Infinity Infinity is a bitwise autoregressive text-to-image generation model (CVPR 2025 Oral) with released training and inference code, checkpoints… | 53 | 1589 | active |
| DeepGraphLearning/torchdrug TorchDrug is a PyTorch-based machine learning library for drug discovery, covering graph neural networks, deep generative models, and reinf… | 23 | 1589 | active |
| bashtage/arch A Python library for financial econometrics providing ARCH/GARCH volatility models, unit root tests, cointegration analysis, bootstrapping,… | 69 | 1588 | stable |
| stanfordnlp/pyreft pyreft is Stanford NLP's Python library for Representation Finetuning (ReFT), which adapts frozen language models by learning task-specific… | 52 | 1587 | active |
| meta-pytorch/torchtune Torchtune is a PyTorch-native library for authoring, post-training, and experimenting with large language models. It provides hackable trai… | 67 | 5819 | maintenance |
| gcorso/DiffDock DiffDock is a deep learning implementation of a diffusion generative model for molecular docking, predicting how small molecule ligands bin… | 28 | 1585 | active |
| CodeReclaimers/neat-python A pure-Python implementation of NEAT (NeuroEvolution of Augmenting Topologies), an algorithm for evolving neural network topologies and wei… | 73 | 1584 | active |
| shibing624/similarity A Java toolkit for computing text similarity at word, phrase, sentence, and paragraph levels, offering algorithms like Cilin-based similari… | 61 | 1583 | active |
| RWKV/rwkv.cpp A C++ port of the RWKV language model to the ggml tensor library, providing FP32, FP16, and quantized INT4/INT5/INT8 inference focused on C… | 34 | 1583 | active |
| Xilinx/brevitas Brevitas is a PyTorch library for neural network quantization supporting both post-training quantization (PTQ) and quantization-aware train… | 93 | 1582 | active |
| microsoft/MMdnn MMdnn is a Microsoft toolkit for converting, visualizing, and diagnosing deep learning models across frameworks such as TensorFlow, PyTorch… | 35 | 5803 | maintenance |
| google-ai-edge/model-explorer Model Explorer is a model graph visualizer and debugger from Google AI Edge that renders neural network graphs hierarchically with expandab… | 92 | 1581 | active |
| Equim-chan/Mortal Mortal is a free and open-source AI for Japanese (riichi) mahjong powered by deep reinforcement learning, written in Rust with a Python int… | 48 | 1580 | active |
| DiffEqML/torchdyn Torchdyn is a PyTorch library dedicated to numerical deep learning, providing tools for neural differential equations, implicit models, and… | 23 | 1580 | active |
| tensorflow/model-optimization The TensorFlow Model Optimization Toolkit (tfmot) is a Python library providing tools to optimize machine learning models for deployment, i… | 79 | 1579 | stable |
| ChipWhisperer ChipWhisperer is an open-source toolchain for hardware security research, providing capture hardware designs, FPGA/USB firmware, and a Pyth… | 95 | 1578 | active |
| menyifang/MIMO MIMO is the official PyTorch implementation of a CVPR 2025 paper on controllable character video synthesis using spatially decomposed model… | 32 | 1578 | active |
| google/meridian Meridian is Google's open-source marketing mix modeling (MMM) framework built on Bayesian causal inference, letting advertisers run in-hous… | 71 | 1576 | active |
| evo-design/evo Evo is a 7-billion-parameter DNA foundation model built on the StripedHyena architecture, trained on ~300 billion tokens of prokaryotic who… | 65 | 1576 | active |
| google/GNM GNM is an open ecosystem of parametric statistical human models and perception stacks from Google, starting with GNM Head, a high-fidelity … | 60 | 1576 | active |
| photosynthesis-team/piq PyTorch Image Quality (PIQ) is a collection of measures and metrics for image quality assessment in image-to-image tasks, written in pure P… | 23 | 1576 | stable |
| Memento-Teams/Memento-Skills Memento-Skills is a generalist, continually-learnable LLM agent framework where reusable skills stored as markdown files serve as persisten… | 71 | 1575 | active |
| BloodAxe/pytorch-toolbelt A Python library of PyTorch extensions providing building blocks for fast R&D prototyping, including encoder-decoder architectures, special… | 40 | 1575 | active |
| Flux Zygote.jl is a source-to-source automatic differentiation library for Julia that hooks into the Julia compiler to generate gradient (backwa… | 97 | 1574 | active |
| ArztSamuel/Applying_EANNs A 2D Unity simulation where cars learn to navigate courses using a feedforward neural network trained by a modified genetic algorithm. It s… | 39 | 1573 | stable |
| ARISE-Initiative/robomimic robomimic is a modular Python framework for robot learning from demonstration, providing standardized demonstration datasets and offline le… | 65 | 1572 | active |
| RightNow-AI/autokernel AutoKernel is an open-source autoresearch pipeline that takes any PyTorch model, profiles it to find GPU kernel bottlenecks, extracts them … | 44 | 1571 | active |
| ZFTurbo/Music-Source-Separation-Training A Python training framework for music source separation models, supporting many architectures such as MDX23C, Demucs, Band Split RoFormer, … | 87 | 1569 | active |
| sdv-dev/CTGAN CTGAN is a Python library of deep learning based synthetic data generators for single-table tabular data, implementing the CTGAN conditiona… | 76 | 1569 | active |
| Enemyx-net/VibeVoice-ComfyUI A ComfyUI custom node integration for Microsoft's VibeVoice text-to-speech model, providing single and multi-speaker voice synthesis with v… | 48 | 1567 | active |
| K-Dense-AI/karpathy Karpathy is an open-source agentic Machine Learning Engineer that trains state-of-the-art ML models using the Claude Agent SDK and Google A… | 58 | 1566 | active |
| Chinese-Text-Classification-Pytorch A PyTorch-based collection of ready-to-run Chinese text classification models including TextCNN, TextRNN, FastText, TextRCNN, BiLSTM with a… | 32 | 5728 | maintenance |
| google-deepmind/bsuite bsuite (Behaviour Suite for Reinforcement Learning) is a collection of carefully-designed experiments from DeepMind that investigate core c… | 79 | 1564 | stable |
| QuantaAlpha/QuantaAlpha QuantaAlpha is an LLM-driven framework for mining quantitative alpha factors using a trajectory-based self-evolving paradigm. Users describ… | 52 | 1563 | active |
| yeemachine/kalidokit KalidoKit is a TypeScript library that converts 3D landmark outputs from Mediapipe/Tensorflow.js face, pose, and hand tracking models into … | 45 | 5720 | maintenance |
| SuiteSparse SuiteSparse is a collection of sparse matrix algorithm libraries written in C/C++, including factorization packages (UMFPACK, CHOLMOD, SPQR… | 96 | 1561 | stable |
| baaivision/Emu3.5 Emu3.5 is BAAI's native multimodal foundation model that jointly predicts next states across vision and language, trained on 10T+ interleav… | 40 | 1561 | active |
| mosaicml/streaming StreamingDataset is a Python library from MosaicML for fast, accurate streaming of training data from cloud object storage (S3, GCS, Azure,… | 66 | 1560 | active |
| kritiksoman/GIMP-ML GIMP-ML is a set of Python plugins that bring computer vision and deep learning models into the GNU Image Manipulation Program (GIMP). It p… | 23 | 1560 | active |
| ByteDance-Seed/Triton-distributed Triton-distributed is a distributed compiler built on OpenAI Triton for computation-communication overlapping on multi-GPU systems. It lets… | 61 | 1559 | active |
| ComfyUI-BlenderAI-node A Blender addon that integrates ComfyUI into Blender by converting ComfyUI nodes into Blender nodes, enabling AI image generation, material… | 42 | 1559 | active |
| hustvl/MapTR MapTR is an end-to-end transformer-based framework for online vectorized HD map construction from camera imagery in autonomous driving. It … | 23 | 1558 | active |
| decoderesearch/SAELens SAELens is a Python library for training sparse autoencoders (SAEs) on language model activations and analyzing them for mechanistic interp… | 90 | 1555 | active |
| QwenAudio/Fun-ASR Fun-ASR is a family of open-source LLM-based end-to-end speech recognition models from Tongyi Lab, covering Chinese, dialects, accents, and… | 79 | 1553 | active |
| tianrun-chen/SAM-Adapter-PyTorch A PyTorch library that adapts Meta AI's Segment Anything Model (SAM, SAM2, SAM3) to underperforming downstream segmentation tasks using lig… | 64 | 1553 | active |
| AIDE ML AIDE ML is an open-source Python package implementing an LLM-driven agent that uses agentic tree search to autonomously write, debug, and i… | 60 | 1553 | active |
| ZeyueT/AudioX AudioX is a unified multimodal framework for anything-to-audio generation, producing audio and music conditioned on text, video, image, or … | 49 | 1552 | active |
| nubank/fklearn fklearn is a Python machine learning library from Nubank that applies functional programming principles to build, validate, and deploy mode… | 83 | 1550 | active |
| ShineChen1024/MagicClothing Official PyTorch implementation of Magic Clothing, a diffusion-based model for controllable garment-driven image synthesis (virtual try-on)… | 26 | 1548 | active |
| yuantianyuan01/FastWAM Official PyTorch codebase for Fast-WAM, a World Action Model for robot manipulation that skips test-time future video imagination and gener… | 56 | 1547 | active |
| hustvl/LightningDiT LightningDiT is a research codebase for latent diffusion models implementing VA-VAE and LightningDiT, achieving FID 1.35 on ImageNet-256 wi… | 44 | 1547 | active |
| 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 |