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
TradeMaster-NTU/TradeMaster TradeMaster is an open-source Python platform for quantitative trading powered by reinforcement learning, covering the full pipeline of alg…	30	3087	active
DamRsn/NeuralNote NeuralNote is an audio plugin (VST3/AU/Standalone) that performs audio-to-MIDI transcription using deep learning, based on Spotify's basic-…	86	3084	active
guillaume-be/rust-bert A Rust-native library providing ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT-2, RoBERTa, BART, etc.), por…	57	3084	active
microsoft/MoGe MoGe is a deep learning model from Microsoft Research that recovers 3D geometry from a single open-domain image, predicting metric point ma…	68	3077	active
leggedrobotics/rsl_rl RSL-RL is a GPU-accelerated, lightweight reinforcement learning library for robotics research, featuring algorithms like PPO and Student-Te…	97	3075	active
sonos/tract Tract is Sonos' tiny, self-contained neural-network inference engine written in Rust. It loads ONNX, TensorFlow/TFLite, and NNEF models, op…	97	3075	active
yuyuyzl/EasyVtuber EasyVtuber is a Python-based VTubing application built on the Talking Head Anime model that turns a single anime character illustration int…	59	3071	active
kairos-agi/kairos Kairos is the official open-source implementation of a 4B-parameter native cross-embodiment world model that unifies video understanding, f…	59	3066	active
OpenWorm OpenWorm is an open-source project building a comprehensive bottom-up computational model of the Caenorhabditis elegans roundworm, integrat…	75	3063	active
Neural Amp Modeler Neural Amp Modeler (NAM) is a Python project that trains neural network models to emulate guitar amplifiers and exports them as .nam files.…	80	3060	active
RFDiffusion RFdiffusion is an open-source method for de novo protein structure generation using diffusion models, with or without conditional informati…	59	3058	active
viperrcrypto/Siftly Siftly is a self-hosted, local-first web application for organizing Twitter/X bookmarks into a searchable, categorized knowledge base. It r…	59	3057	active
kyutai-labs/delayed-streams-modeling Kyutai's repository of Speech-To-Text and Text-To-Speech models built on the Delayed Streams Modeling framework, with implementations in Py…	44	3041	active
facebookresearch/Pearl Pearl is a production-ready reinforcement learning agent library developed by Meta's Applied Reinforcement Learning team. It provides modul…	71	3033	active
wasserth/TotalSegmentator TotalSegmentator is a Python command-line tool that robustly segments over 100 anatomical structures in CT and MR images using deep learnin…	66	3033	active
deepseek-ai/DreamCraft3D Official PyTorch implementation of DreamCraft3D, an ICLR 2024 hierarchical 3D content generation method that turns a single 2D image into a…	32	3032	stable
tensorflow/agents TF-Agents is a TensorFlow-based Python library for building, training, and deploying reinforcement learning and contextual bandit algorithm…	57	3028	active
luminal-ai/luminal Luminal is a high-performance general-purpose ML inference compiler written in Rust that lowers models to a minimal 15-op dataflow IR and c…	77	3026	active
deepseek-ai/DualPipe DualPipe is a Python library implementing a bidirectional pipeline parallelism algorithm that overlaps forward and backward computation wit…	45	3017	active
Eladlev/AutoPrompt AutoPrompt is a Python framework for optimizing LLM prompts using Intent-based Prompt Calibration. It iteratively generates challenging edg…	39	3017	active
hgmzhn/manga-translator-ui A desktop GUI application built on manga-image-translator that automatically translates text in manga/comic images across Japanese, Korean,…	79	3016	active
ZQPei/deep_sort_pytorch A PyTorch implementation of the Deep SORT multi-object tracking algorithm, pairing YOLOv3/YOLOv5 (or Mask R-CNN) detectors with a CNN re-id…	32	3013	active
osmr/imgclsmob A research sandbox providing (re)implementations of numerous deep learning computer vision models for classification, segmentation, detecti…	23	3012	active
pytorch/TensorRT Torch-TensorRT is a compiler library that accelerates PyTorch model inference on NVIDIA GPUs using TensorRT. It supports just-in-time compi…	93	3009	active
thu-pacman/chitu Chitu (赤兔) is a high-performance inference framework for large language models, positioned as a production-grade LLM serving engine. It sup…	82	3009	active
leigest519/OpenGame OpenGame is an open-source agentic framework for end-to-end web game creation from a single prompt, built around a reusable 'Game Skill' sy…	56	3009	active
leigest519/ScreenCoder ScreenCoder is a UI-to-code generation system that converts screenshots or design mockups into clean, editable HTML/CSS using a modular mul…	61	3006	active
Doubiiu/DynamiCrafter DynamiCrafter is an open-source research model that animates open-domain still images into short videos using pre-trained video diffusion p…	27	3005	active
pytorch/ao TorchAO is a PyTorch-native library for model optimization through quantization and sparsity. It supports quantizing weights, gradients, op…	87	3001	active
MeiGen-AI/MultiTalk MultiTalk is an audio-driven framework for generating multi-person conversational videos from multi-stream audio, a reference image, and a …	53	3001	active
williamyang1991/Rerender_A_Video The official PyTorch implementation of 'Rerender A Video', a SIGGRAPH Asia 2023 zero-shot text-guided video-to-video translation framework.…	30	2997	stable
vLLM vllm-ascend is a community-maintained hardware plugin that enables vLLM to run large language model inference on Huawei Ascend NPUs. It imp…	83	2996	active
ulab-uiuc/LLMRouter LLMRouter is an open-source Python library that intelligently routes queries to the most suitable LLM based on task complexity, cost, and p…	62	2995	active
mazzzystar/Queryable Queryable is an open-source iOS app that runs Apple's MobileCLIP (formerly OpenAI's CLIP) entirely on-device to search your photo album wit…	59	2993	active
benedekrozemberczki/pytorch_geometric_temporal PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric providing spatiotemporal signal processing with …	64	2992	active
hosseinmoein/DataFrame A high-performance, templatized C++ library providing a heterogeneous DataFrame container for in-memory data exploration, transformation, a…	95	2990	active
openai/shap-e Shap-E is OpenAI's official release of a generative model that produces 3D implicit functions conditioned on text prompts or images. It inc…	30	12263	maintenance
Biohub/esm A Python library and model release providing ESMC protein language models, ESMFold2 structure prediction, and the ESM Atlas of 6.8 billion …	88	2987	active
CLIP Interrogator A Python library that combines OpenAI's CLIP and Salesforce's BLIP to reverse-engineer text prompts from images, optimized for use with tex…	23	2986	stable
sovrasov/flops-counter.pytorch A Python library (ptflops) that computes the theoretical number of multiply-add operations (FLOPs) and parameter counts for neural network …	36	2983	active
bethgelab/foolbox Foolbox is a Python library for generating adversarial examples that fool deep neural networks, with state-of-the-art gradient-based and de…	44	2979	active
patrick-kidger/equinox Equinox is a Python library providing neural networks and scientific computing utilities for JAX, using PyTorch-like class-based syntax whe…	88	2976	stable
anyoptimization/pymoo pymoo is an open-source Python framework for single-, multi-, and many-objective optimization, offering state-of-the-art algorithms like NS…	75	2970	stable
LIME Lime (Local Interpretable Model-agnostic Explanations) is a Python library that explains the predictions of any machine learning classifier…	23	12170	maintenance
0xemmkty/QuantMuse A Python-based quantitative trading system combining factor analysis, ML models, and LLM-powered market analysis with real-time market data…	33	2964	active
zju3dv/LoFTR LoFTR is a detector-free local image feature matching method using Transformers, released with PyTorch inference and training code plus pre…	32	2963	stable
sunsmarterjie/yolov12 YOLOv12 is a PyTorch implementation of attention-centric real-time object detectors, published at NeurIPS 2025. It provides detection model…	65	2961	active
mit-nlp/MITIE MITIE is a C++ library and toolset for information extraction, providing named entity recognition and binary relation detection built on dl…	39	2960	active
karpathy/char-rnn char-rnn is a Torch/Lua implementation of multi-layer recurrent neural networks (RNN, LSTM, GRU) for character-level language modeling. It …	32	12120	maintenance
li-plus/chatglm.cpp A pure C++ implementation of ChatGLM-6B, ChatGLM2-6B, ChatGLM3, and GLM-4(V) based on ggml, enabling memory-efficient real-time chatting on…	21	2957	active
dreamquark-ai/tabnet A PyTorch implementation of the TabNet deep learning architecture for tabular data, providing attentive, interpretable models for classific…	23	2956	stable
iscyy/ultralyticsPro A PyTorch-based collection of improved YOLO-family object detection models (YOLOv5 through YOLOv13, RT-DETR) with pluggable modules for bac…	45	2954	active
TylerYep/torchinfo torchinfo is a Python library that provides Keras-style model summaries for PyTorch, showing layer types, input/output shapes, parameter co…	66	2953	active
ogkalu2/comic-translate An AI-powered application and browser extension that automatically translates comics, manga, manhwa, webtoons, and BDs across many language…	89	2949	active
jeeliz/jeelizFaceFilter A lightweight JavaScript/WebGL library for real-time face detection and tracking from a camera feed via WebRTC, designed for building augme…	43	2949	active
KichangKim/DeepDanbooru DeepDanbooru is a Python/TensorFlow system that estimates Danbooru-style tags for anime-style girl images using multi-label classification.…	60	2948	active
microsoft/table-transformer Table Transformer (TATR) is a deep learning object detection model from Microsoft for detecting, recognizing, and extracting tables from un…	10	2948	active
turboderp/exllama ExLlama is a standalone Python/C++/CUDA implementation of the Llama architecture optimized for running 4-bit GPTQ quantized weights on mode…	30	2947	active
bghira/SimpleTuner SimpleTuner is a Python fine-tuning toolkit for image, video, and audio diffusion models built on Hugging Face Diffusers. It provides a web…	93	2945	active
facebookresearch/omnilingual-asr An open-source multilingual speech recognition library from Meta AI supporting over 1,600 languages, including hundreds never previously co…	47	2941	active
sokrypton/ColabFold ColabFold is a set of tools and Google Colab notebooks that make protein structure prediction with AlphaFold2, AlphaFold3, ESMFold, and rel…	92	2940	active
Tencent-Hunyuan/HunyuanWorld-1.0 Tencent HunyuanWorld-1.0 is an open-source 3D world generation model that creates immersive, explorable, and interactive 3D worlds from tex…	50	2936	active
zjunlp/EasyEdit EasyEdit is an easy-to-use knowledge editing framework for large language models, supporting methods that modify model parameters or steer …	74	2929	active
Mitsuba 3 Mitsuba 3 is a research-oriented, retargetable rendering system for forward and inverse light transport simulation, written in C++17 on top…	90	2928	active
InternLM/InternLM-XComposer InternLM-XComposer is a family of large vision-language models from the InternLM team, including XComposer2.5 for long-context multimodal u…	35	2927	active
state-spaces/s4 Official implementations of Structured State Space (S4) sequence models and related variants like S4D, HiPPO, and SaShiMi, built in PyTorch…	32	2927	active
keras-team/keras-tuner KerasTuner is a hyperparameter optimization framework for Keras models with a define-by-run syntax for configuring search spaces. It ships …	46	2923	active
NVlabs/FoundationStereo FoundationStereo is NVIDIA's official PyTorch implementation of a foundation model for zero-shot stereo depth estimation, published as a CV…	44	2923	active
idootop/MagicMirror MagicMirror is a desktop application for instant AI face swapping in photos, built with Tauri. It runs entirely offline on standard hardwar…	34	2922	active
decoderesearch/circuit-tracer A Python library for circuit tracing and attribution graphs in language model interpretability, based on cross-layer MLP transcoders. It co…	77	2918	active
eric-mitchell/direct-preference-optimization A reference implementation of Direct Preference Optimization (DPO) for training language models from human preference data, built on Huggin…	29	2916	stable
elixir-nx/nx Nx is an Elixir library providing multi-dimensional arrays (tensors) and numerical definitions with pluggable backends. The monorepo also i…	98	2915	active
AllenNLP AllenNLP is an open-source NLP research library built on PyTorch for developing state-of-the-art deep learning models on linguistic tasks. …	10	11886	maintenance
Saiyan-World/goku Goku is a family of flow-based (rectified flow Transformer) foundation models for joint image and video generation, released by HKU and Byt…	21	2906	active
learnables/learn2learn learn2learn is a PyTorch library for meta-learning research, providing utilities for few-shot task creation, high-level wrappers for algori…	45	2894	active
RL Baselines3 Zoo RL Baselines3 Zoo is a training framework for reinforcement learning agents built on Stable Baselines3. It provides CLI scripts for trainin…	88	2892	active
explosion/thinc Thinc is a lightweight deep learning library from the makers of spaCy offering a type-checked, functional-programming API for composing neu…	74	2891	active
ideogram-oss/ideogram4 Ideogram 4 is an open-weight text-to-image foundation model trained from scratch, with inference code and weights released in Python. It fe…	51	2888	active
KellerJordan/Muon Muon is a PyTorch optimizer for the hidden layers of neural networks, based on orthogonalized momentum updates via Newton-Schulz iteration.…	56	2887	active
Y-Research-SBU/QuantHarness QuantHarness is a multi-agent LLM framework for high-frequency trading analysis, built on LangChain and LangGraph. Specialized agents compu…	59	2874	active
Trusted-AI/AIF360 AI Fairness 360 is an open-source Python and R library from IBM Research providing fairness metrics, explanations, and bias mitigation algo…	58	2872	active
openmv/openmv OpenMV is an open-source machine vision platform consisting of camera hardware firmware programmable in Python 3 (MicroPython). The firmwar…	93	2871	active
werner-duvaud/muzero-general A commented and documented PyTorch implementation of DeepMind's MuZero reinforcement learning algorithm, designed to be easily adapted to a…	32	2870	active
numz/ComfyUI-SeedVR2_VideoUpscaler The official ComfyUI integration of ByteDance's SeedVR2 model for high-quality video and image upscaling, provided as custom nodes. It can …	49	2866	active
intro-skipper/intro-skipper A Jellyfin server plugin that analyzes the audio of TV episodes to automatically detect intro and credit sequences. It provides media segme…	95	2859	active
physical-superintelligence-lab/Psi0 Psi-Zero (Ψ₀) is an open vision-language-action (VLA) foundation model for dexterous humanoid loco-manipulation, combining a Qwen3-VL backb…	59	2859	active
linto-ai/whisper-timestamped A Python library extending OpenAI's Whisper models to produce accurate word-level timestamps and confidence scores during multilingual spee…	77	2855	active
UX-Decoder/Semantic-SAM Official PyTorch implementation of Semantic-SAM, a universal image segmentation model that segments and recognizes anything at any desired …	30	2855	active
OpenGVLab/InternImage InternImage is a large-scale CNN-based vision foundation model that uses deformable convolutions as its core operator, released with pretra…	25	2854	stable
PythonOT/POT POT is an open-source Python library providing a large set of differentiable solvers for optimal transport problems, including exact and re…	85	2846	stable
huggingface/nanotron Nanotron is a minimalistic Python library from Hugging Face for pretraining large language models with 3D parallelism (data, tensor, and pi…	63	2843	active
TMElyralab/MuseV MuseV is a diffusion-based framework for generating high-fidelity virtual human videos of infinite length using a Visual Conditioned Parall…	26	2843	active
huggingface/setfit SetFit is a Python library for efficient, prompt-free few-shot fine-tuning of Sentence Transformers for text classification. It achieves hi…	88	2842	active
rednote-machine-learning/RedKnot RedKnot is a long-context LLM inference acceleration library built on SGLang, using head-classified KV reuse, offline KV storage with RoPE …	57	2840	active
dscripka/openWakeWord openWakeWord is an open-source Python library for detecting wake words (or phrases) in streaming audio, with pre-trained models for common …	46	2837	active
AdapterHub Adapters is a Python add-on library for HuggingFace Transformers that integrates 10+ parameter-efficient fine-tuning methods (bottleneck ad…	74	2831	active
Mininglamp-AI/Mano-P Mano-P is an open-source GUI-VLA (vision-language-action) agent model and SDK for edge devices, enabling purely vision-driven cross-platfor…	52	2829	active
VieNeu-TTS VieNeu-TTS is an on-device Vietnamese text-to-speech library with instant zero-shot voice cloning from short reference clips, supporting bi…	84	2828	active
datachain-ai/datachain DataChain is a Python library that turns unstructured files in S3, GCS, and Azure into typed, versioned datasets queryable at warehouse spe…	88	2827	active
facebook/Ax Ax is a Python platform for adaptive experimentation that uses machine learning (Bayesian and bandit optimization via BoTorch) to efficient…	93	2826	active