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
fluid-cloudnative/fluid Fluid is a Kubernetes-native distributed dataset orchestrator and accelerator for data-intensive applications such as big data and AI workl…	77	1986	active
davda54/sam An unofficial PyTorch implementation of Sharpness-Aware Minimization (SAM) and its adaptive variant ASAM, provided as an optimizer wrapper …	32	1986	stable
Netflix/void-model VOID (Video Object and Interaction Deletion) is a research model from Netflix that removes objects from videos along with the physical inte…	51	1983	active
patrikhuber/eos A lightweight, header-only 3D Morphable Face Model (3DMM) fitting library written in modern C++11/14, with Python bindings. It provides mod…	28	1981	active
HybridRobotics/berkeley-humanoid-lite Berkeley Humanoid Lite is the open-source codebase for a sub-$5,000 3D-printed humanoid robot platform from UC Berkeley. It includes Isaac …	51	1980	active
cloneofsimo/lora A Python library for applying Low-Rank Adaptation (LoRA) to quickly fine-tune text-to-image diffusion models like Stable Diffusion. It prod…	23	7557	maintenance
haoheliu/versatile_audio_super_resolution AudioSR is a Python library and CLI tool that performs versatile audio super-resolution, upsampling any audio (music, speech, sound effects…	42	1979	active
WassimTenachi/PhySO PhySO is a Python library for physical symbolic optimization that uses deep reinforcement learning to discover analytical physical laws fro…	49	1978	active
showlab/Show-o Show-o is a research repository implementing a unified transformer model that combines autoregressive and discrete diffusion modeling for m…	48	1978	active
Linzaer/Ultra-Light-Fast-Generic-Face-Detector-1MB An ultra-lightweight face detection model (~1MB FP32, ~300KB quantized) designed for edge computing devices, with slim and RFB variants tra…	32	7548	maintenance
adobe-research/custom-diffusion Custom Diffusion is a research codebase for efficiently fine-tuning text-to-image diffusion models like Stable Diffusion on a few example i…	66	1977	stable
2U1/Qwen-VL-Series-Finetune An open-source Python repository providing training scripts for fine-tuning Alibaba's Qwen-VL series of vision-language models (Qwen2-VL, Q…	65	1977	active
OpenMotionLab/MotionGPT MotionGPT is a unified motion-language model that treats 3D human motion as a foreign language by converting motion into discrete motion to…	30	1974	active
ROCm/FastFlowLM FastFlowLM (FLM) is an NPU-first LLM inference runtime purpose-built and deeply optimized for AMD Ryzen AI NPUs (XDNA2), offering an Ollama…	87	1967	active
meta-pytorch/opacus Opacus is a PyTorch library for training neural networks with differential privacy via DP-SGD, requiring minimal code changes through its P…	78	1967	active
mlc-ai/xgrammar XGrammar is an open-source library for efficient, flexible, and portable structured generation in LLM inference via constrained decoding. I…	91	1964	active
huang-sir1/radiology-skills An open-source collection of 27 Claude/Codex 'skills' (modular rule files) for medical imaging research, covering radiomics, deep learning,…	67	1962	active
NVlabs/RADIO Official PyTorch implementation of AM-RADIO and its successors (RADIOv2.5, C-RADIOv4), agglomerative vision foundation models distilled fro…	61	1961	active
autodiff/autodiff autodiff is a C++17 library for automatic differentiation that computes derivatives of functions efficiently using forward mode (dual numbe…	23	1960	active
CannyLab/tsne-cuda A CUDA-accelerated implementation of the FIt-SNE t-SNE algorithm with Python bindings, offering up to 1200x speedup over scikit-learn. It e…	89	1958	stable
alibaba/EasyCV EasyCV is an all-in-one PyTorch-based computer vision toolkit from Alibaba covering self-supervised learning, vision transformers, and majo…	28	1955	active
eriklindernoren/PyTorch-YOLOv3 A minimal PyTorch implementation of YOLOv3 supporting training, inference, and evaluation, with compatibility for YOLOv4 and YOLOv7 weights…	32	7439	maintenance
microsoft/DiskANN DiskANN3 is a composable Rust library for scalable, accurate vector indexing and nearest-neighbor search, designed to be embedded into data…	100	1952	active
openai/guided-diffusion OpenAI's codebase for guided diffusion models from the paper 'Diffusion Models Beat GANs on Image Synthesis', including classifier conditio…	32	7427	maintenance
kadirnar/whisper-plus A Python library wrapping OpenAI Whisper-family models (including distil-whisper and MLX variants) for fast speech-to-text transcription wi…	62	1950	active
AIStore AIStore (AIS) is a lightweight, distributed object storage stack built by NVIDIA specifically for AI workloads. It provides linear scalabil…	95	1947	active
kyegomez/BitNet A PyTorch implementation of the BitNet architecture from the paper 'BitNet: Scaling 1-bit Transformers for Large Language Models', providin…	73	1947	active
starik222/BooruDatasetTagManager A desktop tag editor for managing booru-style tagged image and video datasets used to train Stable Diffusion models such as LoRAs, embeddin…	70	1947	active
Graph Convolutional Networks A TensorFlow implementation of Graph Convolutional Networks (GCN) for semi-supervised node classification on graphs, accompanying the ICLR …	32	7406	maintenance
MLJ MLJ.jl is a machine learning framework for Julia providing a common interface to over 200 models, with meta-algorithms for model selection,…	91	1946	stable
tensorlayer/TensorLayer TensorLayer is a TensorFlow-based deep learning and reinforcement learning library offering customizable neural layers for researchers and …	23	7383	maintenance
c2g-dev/city2graph City2Graph is a Python library that converts geospatial data such as buildings, streets, GTFS transit feeds, and origin-destination matrice…	84	1940	active
tum-pbs/PhiFlow PhiFlow is an open-source Python simulation toolkit for solving partial differential equations with support for optimization and machine le…	63	1939	active
Audio-AGI/AudioSep AudioSep is the official implementation of the 'Separate Anything You Describe' foundation model for open-domain, language-queried audio so…	29	1938	active
llvm/torch-mlir Torch-MLIR is a compiler project providing first-class translation of PyTorch programs into the MLIR compiler ecosystem. It lets hardware v…	99	1935	active
PAIR-code/facets Facets is a pair of web-component visualizations (Overview and Dive) for understanding and analyzing machine learning datasets, embeddable …	10	7333	maintenance
mlco2/codecarbon CodeCarbon is a lightweight Python library and CLI that estimates and tracks CO2 emissions from local computing hardware (CPU, GPU, RAM) by…	96	1921	active
GPflow/GPflow GPflow is a Python library for building Gaussian process models on top of TensorFlow 2 and TensorFlow Probability. It implements modern Gau…	95	1921	active
Robbyant/lingbot-va LingBot-VA is an autoregressive diffusion framework that unifies video world modeling and robot policy learning in a single interleaved vid…	53	1918	active
LeapLabTHU/Absolute-Zero-Reasoner Official implementation of Absolute Zero Reasoner (AZR), a system that trains LLM reasoning via reinforced self-play with zero external dat…	33	1916	active
dynamicslab/pysindy PySINDy is a Python library for discovering governing equations of dynamical systems from measurement data using the Sparse Identification …	67	1913	active
Obico Obico Server is the self-hostable backend of the Obico smart 3D printing platform, providing AI-based print failure detection, webcam strea…	75	1912	active
FACEGOOD/FACEGOOD-Audio2Face FACEGOOD Audio2Face is an open-source deep learning framework that converts audio into facial blendshape weights for driving digital humans…	61	1912	active
visual-layer/fastdup fastdup is a free Python tool for rapidly analyzing image and video datasets to surface duplicates, outliers, broken, dark, bright, blurry,…	64	1911	active
diffgram/diffgram Diffgram is a self-hosted AI datastore for managing schemas, BLOBs, and predictions, with built-in human supervision (data labeling), data …	59	1911	active
flexflow/flexflow-train FlexFlow Train is a deep learning framework that accelerates distributed DNN training by automatically searching for efficient parallelizat…	67	1903	active
AutoViML/AutoViz AutoViz is a Python library that automatically generates visualizations for any dataset, of any size, with a single line of code. It also i…	32	1902	active
lucidrains/byol-pytorch A PyTorch library implementing the Bootstrap Your Own Latent (BYOL) self-supervised learning method from DeepMind. It wraps any image-based…	55	1900	active
ashvardanian/NumKong NumKong is a mixed-precision numeric and linear-algebra library with ~2,000 SIMD-accelerated kernels for dot products, distances, matrix mu…	93	1899	active
google-deepmind/penzai Penzai is a JAX research toolkit for building, editing, and visualizing neural networks as legible, functional pytree data structures. It i…	10	1899	active
vllm-project/vllm-metal A community-maintained hardware plugin that enables vLLM to run LLM inference on Apple Silicon Macs using MLX as the primary compute backen…	80	1898	active
laugh12321/TensorRT-YOLO A C++/Python deployment toolkit for running YOLO-family models (YOLOv3 through YOLO26) on NVIDIA GPUs using TensorRT, with custom plugins, …	71	1893	active
HyperTools HyperTools is a Python library for visualizing and exploring high-dimensional data via dimensionality reduction, built on matplotlib, sciki…	91	1891	active
ocrs Ocrs is a Rust library and CLI tool for optical character recognition that extracts text from images such as scanned documents, photos, and…	70	1891	active
trevorstephens/gplearn gplearn is a Python library implementing genetic programming with a scikit-learn compatible API, focused on symbolic regression. It provide…	69	1890	stable
stanfordmlgroup/ngboost NGBoost is a Python library implementing Natural Gradient Boosting for probabilistic prediction, built on top of scikit-learn. It outputs f…	86	1889	active
SqueezeAILab/LLMCompiler LLMCompiler is a Python framework from an ICML 2024 paper that orchestrates parallel function calling with LLMs. It uses a planner, task fe…	27	1889	active
INESCTEC/yake YAKE! is a lightweight unsupervised keyword extraction library that ranks the most important keywords of a single document using text stati…	57	1887	stable
d8ahazard/sd_dreambooth_extension A Stable Diffusion WebUI extension that adds DreamBooth fine-tuning capabilities, ported from Shivam Shrirao's optimized Diffusers implemen…	38	1884	active
guess-js/guess Guess.js is a collection of libraries and tools that use analytics data and machine learning to predict user navigation patterns on website…	77	7121	maintenance
nitrain/nitrain Nitrain is a framework-agnostic Python library for sampling, augmenting, and training AI models on medical imaging datasets, with support f…	23	1882	active
netease-youdao/BCEmbedding BCEmbedding is NetEase Youdao's open-source library of bilingual and crosslingual embedding and reranker models for English and Chinese, bu…	68	1881	active
timsainb/noisereduce A Python library for noise reduction in time-domain signals using spectral gating, supporting stationary and non-stationary noise reduction…	36	1881	active
qqwweee/keras-yolo3 A Keras (TensorFlow backend) implementation of YOLOv3 for object detection, including Darknet weight conversion, image/video detection scri…	32	7113	maintenance
nndeploy/nndeploy nndeploy is an easy-to-use, high-performance AI deployment framework written in C++ with Python bindings. It provides a visual drag-and-dro…	81	1880	active
microsoft/MarS MarS is a financial market simulation engine powered by a generative foundation model, developed by Microsoft. It provides tools for simula…	76	1880	active
nomic-ai/nomic The official Python client for Nomic Atlas, a platform for exploring, labeling, searching, and sharing massive unstructured datasets (text,…	56	1880	active
johannfaouzi/pyts pyts is a Python package for time series classification that provides preprocessing, transformation, and utility tools along with implement…	88	1879	stable
asavinov/intelligent-trading-bot A Python-based automated cryptocurrency trading bot that generates trade signals using machine learning and feature engineering. It separat…	74	1879	active
425776024/nlpcda nlpcda is a Python library for one-click Chinese NLP data augmentation. It provides techniques like random entity replacement, synonym subs…	24	1875	active
patrick-kidger/jaxtyping A Python library providing type annotations and runtime type-checking for the shape and dtype of arrays and tensors in JAX, PyTorch, NumPy,…	90	1873	active
descriptinc/descript-audio-codec Descript Audio Codec (.dac) is a high-fidelity neural audio codec based on improved RVQGAN that compresses audio at roughly 90x compression…	58	1873	stable
e-p-armstrong/augmentoolkit Augmentoolkit is a Python application that generates domain-expert fine-tuning datasets from uploaded documents and trains custom LLMs on t…	65	1872	active
dauparas/ProteinMPNN ProteinMPNN is a PyTorch-based tool that designs amino acid sequences for given protein backbone structures using a message-passing neural …	23	1872	stable
PRIME-RL/PRIME PRIME (Process Reinforcement through Implicit Rewards) is an open-source Python framework for online reinforcement learning with process re…	23	1872	active
dphnAI/sonar Sonar is a large-scale LLM inference engine for Hugging Face-compatible language and multimodal models, based on vLLM (formerly the Aphrodi…	92	1871	active
AkaliKong/MiniOneRec MiniOneRec is an open-source framework for generative recommendation built on large language models, covering the full pipeline of semantic…	59	1868	active
PRIME-RL/SimpleVLA-RL SimpleVLA-RL is an open-source reinforcement learning framework for training Vision-Language-Action (VLA) models for robotic manipulation, …	43	1864	active
RoboVerseOrg/RoboVerse RoboVerse is a unified platform, dataset, and benchmark for scalable and generalizable robot learning, providing tasks, robots, scenes, ass…	65	1863	active
laekov/fastmoe FastMoE is a PyTorch library providing efficient Mixture of Experts (MoE) layers with custom C/CUDA operators. It supports distributed expe…	23	1860	active
OpenNMT OpenNMT is an open-source ecosystem for neural machine translation and sequence learning, with PyTorch (OpenNMT-py) and TensorFlow (OpenNMT…	40	7018	maintenance
LuChengTHU/dpm-solver Official PyTorch implementation of DPM-Solver and DPM-Solver++, fast high-order ODE solvers for diffusion probabilistic model sampling that…	32	1859	stable
openreasoner/openr OpenR is an open-source Python framework that integrates search, reinforcement learning, and process supervision to improve chain-of-though…	23	1859	active
TorchSharp TorchSharp is a .NET library providing bindings to LibTorch, the library that powers PyTorch, with a focus on tensors and a PyTorch-like AP…	69	1858	active
facebookresearch/MetaCLIP Meta's research code and models for Meta CLIP, a reimplementation and scaling recipe for CLIP-style contrastive vision-language models, inc…	76	1854	active
SerpentAI Framework Serpent.AI is a Python framework for building game agents—AIs and bots that learn to play any video game you own—turning games into machine…	10	6989	maintenance
microsoft/mattergen MatterGen is Microsoft's official implementation of a generative diffusion model for designing inorganic crystalline materials across the p…	65	1852	active
Tele-AI/Telechat TeleChat is a family of open-source bilingual (Chinese-English) large language models (1B, 7B, 12B) developed by China Telecom's AI team, r…	64	1852	active
trevin-creator/autoresearch-mlx An Apple Silicon (MLX) port of Karpathy's autoresearch that runs autonomous AI research loops on a Mac without PyTorch or CUDA. A coding ag…	52	1852	active
probcomp/Gen.jl Gen.jl is a general-purpose probabilistic programming system embedded in Julia that lets users write generative models as probabilistic pro…	58	1851	active
bmabey/pyLDAvis pyLDAvis is a Python library for interactive visualization of fitted LDA topic models, ported from the R LDAvis package. It renders web-bas…	44	1851	active
Responsible AI Toolbox A Microsoft suite of Python libraries and Jupyter widgets providing dashboards for model assessment, error analysis, interpretability, fair…	68	1848	active
boundless-large-model/boundless-world-model Boundless-World-Model (BWM) is a physically consistent, action-conditioned video world model built on Wan2.2-TI2V-5B that acts as a low-cos…	57	1840	active
magenta/magenta-realtime Magenta RealTime 2 (MRT2) is an open-weights model for real-time, interactive music generation controllable via text, MIDI, and audio. It s…	70	1839	active
clovaai/donut Donut is an OCR-free end-to-end Transformer model for visual document understanding tasks such as document classification and information e…	23	6928	maintenance
Robbyant/lingbot-vla LingBot-VLA is a Vision-Language-Action foundation model for robot manipulation, pretrained on 20,000 hours of real-world dual-arm robot da…	51	1837	active
ytongbai/LVM LVM is a large vision model trained with sequential next-token prediction over 'visual sentences', using no linguistic data. It builds on O…	31	1836	active
cazala/synaptic Synaptic is an architecture-free neural network library for JavaScript that runs in both Node.js and the browser. It supports building and …	62	6912	maintenance
AgiBot X1 The inference module of the open-source AgiBot X1 humanoid robot software stack, providing model inference, platform drivers, and simulatio…	25	1835	active
ZFTurbo/Weighted-Boxes-Fusion A Python library implementing several methods for ensembling bounding boxes from multiple object detection models, including Non-maximum Su…	66	1831	stable