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
BishopFox/eyeballer Eyeballer is a convolutional neural network tool that classifies screenshots of web hosts taken during large-scope penetration tests. It la…	51	1293	active
Xiangyue-Zhang/auto-deep-researcher-24x7 An open-source Python framework where LLM agents autonomously run deep learning experiments 24/7, covering hypothesis formation, code imple…	50	1293	active
xinychen/transdim transdim is a Python/Jupyter Notebook project providing machine learning models for transportation data imputation and spatiotemporal time …	59	1292	active
0xSero/ai-data-extraction A Python toolkit of extraction scripts that pulls complete chat, agent, and code-context history from AI coding assistants like Cursor, Cla…	60	1291	active
locuslab/TCN PyTorch implementation of Temporal Convolutional Networks (TCN) with benchmarks from the paper 'An Empirical Evaluation of Generic Convolut…	32	4569	maintenance
plaidml/plaidml PlaidML is a portable tensor compiler that enables deep learning on hardware (especially GPUs and embedded devices) not well supported by m…	10	4566	maintenance
RoyalVane/CLAN Official PyTorch implementation of CLAN, a CVPR 2019 (oral) / TPAMI 2022 method for unsupervised domain adaptation in semantic segmentation…	32	1289	stable
ELLA ELLA is an Efficient Large Language Model Adapter that equips text-to-image diffusion models with LLM-based text understanding via a Timest…	26	1289	active
e3nn/e3nn e3nn is a Python/PyTorch library for building E(3)-equivariant neural networks that respect 3D rotation, translation, and mirror symmetries…	64	1288	active
city-super/Scaffold-GS Scaffold-GS is a research implementation of a structured 3D Gaussian splatting method that uses anchor points on a sparse voxel grid to dis…	27	1286	active
flutter-ml/google_ml_kit_flutter A set of Flutter plugins wrapping Google's ML Kit on-device machine learning APIs for Android and iOS. It provides vision APIs (barcode, fa…	75	1285	active
DreamTechAI/Direct3D-S2 Direct3D-S2 is a research framework for high-resolution 3D shape generation from images, built on sparse volumetric representations and a n…	26	1285	active
nicodv/kmodes A Python library implementing the k-modes and k-prototypes clustering algorithms for categorical and mixed numerical/categorical data. It f…	23	1285	stable
kubeai-project/kubeai KubeAI is a Kubernetes operator for serving machine learning models in production, supporting LLMs via vLLM and Ollama, vector embeddings, …	95	1284	active
X-Square-Robot/wall-x Wall-X is the open-source training and inference stack for X Square Robot's WALL series of embodied foundation models (VLAs) for general-pu…	61	1283	active
poetiq-ai/poetiq-arc-agi-solver A Python research codebase that reproduces Poetiq's record-breaking, leaderboard-topping submissions to the ARC-AGI-1 and ARC-AGI-2 abstrac…	39	1283	active
BachiLi/diffvg diffvg is a differentiable rasterizer for 2D vector graphics that bridges the raster and vector domains via backpropagation. It computes pi…	38	1283	active
Nixtla/mlforecast mlforecast is a Python framework for time series forecasting using any machine learning model with fit/predict methods, providing efficient…	94	1282	active
dcharatan/pixelsplat pixelSplat is a PyTorch implementation of a feed-forward model that reconstructs 3D radiance fields parameterized by 3D Gaussian primitives…	27	1282	stable
opendatalab/LabelLLM LabelLLM is an open-source data annotation platform designed to streamline the labeling workflows needed for LLM development. It offers con…	62	1281	active
Tianxiaomo/pytorch-YOLOv4 A minimal PyTorch implementation of YOLOv4 (and YOLOv4-tiny) supporting inference and training, with tools to convert Darknet weights to Py…	32	4520	maintenance
suragnair/alpha-zero-general A clean, flexible implementation of the AlphaZero self-play reinforcement learning algorithm that can be adapted to any two-player turn-bas…	32	4520	maintenance
AaronJackson/vrn Research code for the ICCV 2017 paper 'Large Pose 3D Face Reconstruction from a Single Image via Direct Volumetric CNN Regression'. It uses…	32	4518	maintenance
stair-lab/kg-gen kg-gen is a Python library that extracts knowledge graphs from arbitrary plain text or conversation messages using LLMs, with model routing…	57	1278	active
DeepEyes DeepEyes is a research project that trains multimodal vision-language models to 'think with images' using end-to-end reinforcement learning…	40	1278	active
SimonBlanke/Gradient-Free-Optimizers A Python library for gradient-free optimization of black-box functions, offering 23 algorithms (hill climbing, Bayesian optimization, parti…	91	1277	active
Wiktextract A Python package and CLI tool that parses Wiktionary XML dump files and extracts structured dictionary data (glosses, translations, pronunc…	75	1277	active
ModelCloud/GPTQModel GPTQModel is a production-ready Python toolkit for quantizing (compressing) large language models using GPTQ, AWQ, and related methods, wit…	92	1276	active
nianticlabs/monodepth2 Monodepth2 is the reference PyTorch implementation of the ICCV 2019 paper 'Digging into Self-Supervised Monocular Depth Prediction'. It tra…	32	4504	maintenance
Renumics/spotlight Renumics Spotlight is an open-source tool for interactively exploring unstructured datasets (images, audio, text, video, time-series, meshe…	89	1275	active
JoyCaption JoyCaption is an open, free, and uncensored image captioning Visual Language Model (VLM) with released weights and training scripts. It gen…	50	1275	active
google-research/simclr Google Research's official implementation of SimCLR and SimCLRv2, a framework for contrastive learning of visual representations, with 65 p…	10	4503	maintenance
MilaNLProc/contextualized-topic-models A Python library implementing Contextualized Topic Models (CTM), which combine pre-trained contextual embeddings like BERT with neural topi…	43	1274	active
lpiccinelli-eth/UniDepth UniDepth is a Python library and research codebase for universal monocular metric depth estimation from single images, based on CVPR 2024 a…	32	1274	active
Dex Retargeting A Python library of retargeting optimizers that translate human hand motion (from video or pose datasets) into robot dexterous hand joint c…	32	1274	active
Ma-Lab-Berkeley/CRATE CRATE is the official PyTorch implementation of the Coding RAte reduction TransformEr, a family of 'white-box' transformer architectures de…	30	1274	active
magenta/magenta-studio Magenta Studio is a collection of five MIDI music generation plugins (Continue, Generate, Interpolate, Groove, Drumify) for Ableton Live, b…	59	1272	active
luo3300612/Visualizer A lightweight Python library that extracts attention maps and other local variables from deep inside PyTorch models for visualization. It w…	32	1272	stable
NVlabs/stylegan2-ada-pytorch Official PyTorch implementation of StyleGAN2-ADA, a generative adversarial network with adaptive discriminator augmentation for training wi…	32	4487	maintenance
aiming-lab/Agent0 Agent0 Series is a research framework for training self-evolving LLM agents from zero external data via tool-integrated reasoning and co-ev…	55	1271	active
stardist/stardist StarDist is a Python library for object detection and instance segmentation in 2D and 3D microscopy images using star-convex shapes, built …	54	1271	stable
withoutbg/withoutbg-python A Python SDK (pip install withoutbg) for removing image backgrounds, offering a free local open-weights ONNX model and an optional paid clo…	79	1270	active
leoxiaobin/deep-high-resolution-net.pytorch Official PyTorch implementation of HRNet (Deep High-Resolution Representation Learning for Human Pose Estimation, CVPR 2019), which maintai…	32	4478	maintenance
ICT-FinD-Lab/alphagen AlphaGen is a Python research library that automatically generates formulaic alpha (predictive) stock factors using reinforcement learning,…	67	1268	active
amaiya/ktrain ktrain is a lightweight Python wrapper around TensorFlow Keras that provides pre-canned, low-code models for text, vision, graph, and tabul…	23	1268	active
NVIDIA-NeMo/Gym NeMo Gym is a Python library from NVIDIA for evaluating and improving LLM models and agents using environments. It provides infrastructure …	81	1267	active
tensorflow/model-analysis TensorFlow Model Analysis (TFMA) is a Python library for evaluating TensorFlow models on large datasets in a distributed manner. It compute…	77	1267	active
meta-pytorch/data TorchData is a PyTorch library providing scalable, performant data loading utilities, including StatefulDataLoader, a drop-in replacement f…	72	1267	active
Bjarten/early-stopping-pytorch A small PyTorch utility package providing an EarlyStopping class that monitors validation loss during training and stops when it stops impr…	34	1267	stable
OpenNRE OpenNRE is an open-source Python toolkit for neural relation extraction, extracting relation triples between entities from plain text. It u…	32	4467	maintenance
TencentCloudADP/youtu-graphrag Youtu-GraphRAG is a Python framework for graph-based retrieval-augmented generation that unifies graph schema construction, community detec…	52	1266	active
fastgs/FastGS FastGS is a general acceleration framework for 3D Gaussian Splatting that trains scenes in roughly 100 seconds using multi-view consistent …	46	1265	active
JonathonLuiten/TrackEval TrackEval is a Python library for evaluating multi-object tracking (MOT) algorithms, implementing metrics such as HOTA, CLEARMOT, IDF1, VAC…	32	1265	stable
bryandlee/animegan2-pytorch A PyTorch implementation of AnimeGANv2, a GAN-based image-to-image style transfer model that converts photos into anime-style images. It pr…	32	4457	maintenance
MyoHub/myosuite MyoSuite is a collection of musculoskeletal environments and tasks simulated with the MuJoCo physics engine and wrapped in the OpenAI gym A…	95	1263	active
mjpost/sacrebleu SacreBLEU is a Python library and CLI tool for computing shareable, comparable, and reproducible BLEU, chrF, and TER scores for machine tra…	70	1263	active
cvg/depthsplat DepthSplat is a PyTorch research library implementing a CVPR 2025 model that connects Gaussian splatting with single/multi-view depth estim…	52	1262	active
meta-pytorch/attention-gym Attention Gym is a collection of tools, examples, and reference implementations for working with PyTorch's FlexAttention API. It provides a…	91	1260	active
PyDMD/PyDMD PyDMD is a Python package implementing Dynamic Mode Decomposition (DMD) and its many variants (mrDMD, HODMD, etc.) for extracting spatiotem…	56	1260	active
Roblox/cube Cube is Roblox's open-source family of foundation models for 3D intelligence, including text-to-3D shape generation and part-controllable m…	55	1256	active
Maia Chess Maia is a collection of human-like neural network chess engines trained on millions of human games, targeting skill levels from ELO 1100 to…	57	1255	active
unum-cloud/UForm UForm is a compact multimodal AI library providing tiny image-text embedding models (64-768 dimensions, Matryoshka-style) and small generat…	50	1254	active
Equim-chan/mjai-reviewer A Rust CLI tool that reviews riichi mahjong game logs using mjai-compatible AI engines such as Mortal and akochan. It fetches logs from Ten…	23	1254	active
automl/SMAC3 SMAC3 is a Python library for Bayesian Optimization used to tune hyperparameters of machine learning algorithms and configure arbitrary alg…	89	1253	active
dmlc/dlpack DLPack is a minimal, stable open in-memory tensor structure defined as a single C ABI header, enabling zero-copy tensor exchange between de…	72	1253	stable
VainF/pytorch-msssim A PyTorch library providing fast, differentiable SSIM and MS-SSIM image quality metrics using separable Gaussian filtering for speed. It ca…	23	1253	stable
facebookresearch/llm-transparency-tool An interactive toolkit from Meta Research for analyzing the internal workings of Transformer-based language models. It visualizes contribut…	10	1253	active
sentinel-hub/eo-learn eo-learn is a collection of open-source Python packages for accessing and processing spatio-temporal satellite imagery, built around modula…	65	1251	active
Stock Indicators A C# NuGet library (FacioQuo.Stock.Indicators, formerly Skender.Stock.Indicators) that transforms historical OHLCV price bars from equities…	95	1248	active
wannesm/dtaidistance A Python library for computing time series distances, most notably Dynamic Time Warping (DTW), with a fast C implementation exposed via Cyt…	84	1248	stable
facebookresearch/home-robot HomeRobot is an open-source robotics stack from Meta AI for mobile manipulation tasks on low-cost hardware like the Hello Robot Stretch. It…	23	1247	active
LTH14/fractalgen A PyTorch implementation of Fractal Generative Models (FractalGen), enabling pixel-by-pixel high-resolution image generation. It includes p…	21	1247	active
abewley/sort SORT is a barebones Python implementation of a simple online and realtime multiple object tracking algorithm for 2D video sequences, based …	32	4386	maintenance
Acly/comfyui-inpaint-nodes A set of custom nodes for ComfyUI that improve image inpainting and outpainting workflows. It integrates the Fooocus inpaint model for SDXL…	70	1246	active
marian42/mesh_to_sdf A Python library that computes approximate signed distance fields (SDFs) for arbitrary triangle meshes, including non-watertight, self-inte…	32	1246	stable
IDSIA/sacred Sacred is a Python library for configuring, organizing, logging, and reproducing machine learning and scientific experiments. It provides c…	41	4381	maintenance
ZHKKKe/MODNet MODNet is a deep learning model for real-time portrait matting (background removal) that requires only an RGB image as input, with no trima…	32	4376	maintenance
HJYao00/Mulberry Mulberry is a research implementation of an o1-like multimodal large language model (MLLM) that performs step-by-step reasoning and reflect…	46	1244	active
FenTechSolutions/CausalDiscoveryToolbox A Python library for causal inference in graphs and pairwise settings, implementing many algorithms for graph structure recovery from obser…	49	1242	active
mrousavy/react-native-fast-tflite A high-performance TensorFlow Lite library for React Native built on Nitro Modules, using the low-level C/C++ TFLite core API with zero-cop…	80	1240	active
apchenstu/TensoRF TensoRF is a PyTorch implementation of the ECCV 2022 paper 'TensoRF: Tensorial Radiance Fields', which models and reconstructs radiance fie…	41	1239	stable
sh-lee-prml/HierSpeechpp Official PyTorch implementation of HierSpeech++, a fast zero-shot speech synthesizer for text-to-speech and voice conversion based on hiera…	29	1237	active
pytorch/serve TorchServe is a flexible, production-ready model server for serving, optimizing, and scaling PyTorch models over HTTP with support for CPU,…	10	4344	maintenance
ElectricAlexis/NotaGen NotaGen is a symbolic music generation model that produces high-quality classical sheet music using LLM-style training paradigms: pre-train…	29	1236	active
SciML/NeuralPDE.jl NeuralPDE.jl is a Julia library of physics-informed neural network (PINN) solvers for ordinary, stochastic, and partial differential equati…	99	1235	active
open-mmlab/playground OpenMMLab Playground is a central hub collecting and showcasing community projects that extend OpenMMLab libraries with Segment Anything Mo…	31	1235	active
InternScience/GraphGen GraphGen is a Python framework for knowledge-graph-guided synthetic data generation for LLM training. It builds fine-grained knowledge grap…	53	1232	active
chelsea0x3b/cudarc cudarc is a safe, minimal Rust wrapper around the NVIDIA CUDA toolkit, exposing the CUDA driver API plus libraries such as NVRTC, cuBLAS/cu…	97	1231	active
qualcomm/ai-hub-models Qualcomm AI Hub Models is a curated collection of 300+ state-of-the-art machine learning models (vision, audio, speech, generative AI) pre-…	90	1230	active
OpenTSLM/OpenTSLM OpenTSLM is a family of Time-Series Language Models that integrate time series as a native modality into pretrained LLMs (Llama, Gemma), en…	58	1230	active
python-adaptive/adaptive Adaptive is an open-source Python library for parallel active learning of mathematical functions. It intelligently selects the most informa…	87	1229	active
sgl-project/SpecForge SpecForge is a Python framework from the SGLang team for training speculative decoding models such as EAGLE/EAGLE3 draft heads. Trained mod…	64	1228	active
MotrixLab/SMPLer-X Official code for SMPLer-X, a family of foundation models for expressive human pose and shape estimation (EHPS) that unifies body, hand, an…	56	1228	stable
NVIDIA/BigVGAN BigVGAN is NVIDIA's official PyTorch implementation of a universal neural vocoder (ICLR 2023) that generates high-fidelity raw audio wavefo…	23	1228	stable
datawhalechina/torch-rechub Torch-RecHub is a lightweight PyTorch framework for building recommendation system models with 30+ out-of-the-box algorithms covering ranki…	96	1227	active
langchain-ai/openevals OpenEvals is a library of readymade evaluators for testing LLM applications, including LLM-as-judge evaluators and reference-based/referenc…	89	1227	active
zkonduit/ezkl EZKL is a Rust-based library and command-line tool that converts deep learning models and arbitrary computational graphs (exported as ONNX)…	69	1227	active
GuitarML/NeuralPi NeuralPi is a DIY Raspberry Pi 4-based guitar pedal that uses neural networks to emulate real amplifiers and distortion/overdrive pedals in…	23	1227	active
alibaba/x-deeplearning X-DeepLearning (XDL) is an industrial deep learning framework from Alibaba optimized for high-dimension sparse data scenarios such as adver…	23	4301	maintenance
run-house/kubetorch Kubetorch is a Python library that lets you distribute and run ML workloads (training, inference, data processing) on Kubernetes directly f…	76	1225	active