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 |
|---|---|---|---|
| KomputeProject/kompute Kompute is a general-purpose GPU compute framework built on Vulkan that works across vendor GPUs (AMD, NVIDIA, Qualcomm, etc.) with both C+… | 63 | 2577 | active |
| graphistry/pygraphistry PyGraphistry is a Python library for loading, shaping, and visually exploring large graphs with GPU-accelerated rendering and analytics via… | 100 | 2574 | active |
| scverse/scanpy Scanpy is a scalable Python toolkit for analyzing single-cell gene expression data, built alongside the anndata data structure. It provides… | 91 | 2574 | stable |
| Tencent-Hunyuan/HY-Motion-1.0 HY-Motion 1.0 is a series of text-to-3D human motion generation models from Tencent Hunyuan, based on Diffusion Transformer and Flow Matchi… | 55 | 2574 | active |
| advimman/lama LaMa is a PyTorch-based image inpainting model that fills large missing regions in images using fast Fourier convolutions, generalizing wel… | 32 | 10291 | maintenance |
| pykeio/ort ort is a safe, fast Rust binding for Microsoft's ONNX Runtime, providing hardware-accelerated machine learning inference and training for O… | 91 | 2569 | active |
| ymcui/Chinese-BERT-wwm A collection of Chinese pre-trained language models (BERT-wwm, BERT-wwm-ext, RoBERTa-wwm-ext, RBT3, etc.) built with Whole Word Masking, re… | 63 | 10231 | maintenance |
| LTH14/JiT A PyTorch/GPU re-implementation of JiT (Just image Transformer), a minimalist pixel-space diffusion model for high-resolution image generat… | 39 | 2553 | active |
| DeepDetect DeepDetect is an open-source deep learning runtime, CLI, and REST server written in C++ for training and inference across images, text, tab… | 95 | 2551 | active |
| mozilla/TTS A deep learning library for advanced text-to-speech generation, built on PyTorch with models like Tacotron2, Glow-TTS, and various vocoders… | 23 | 10173 | maintenance |
| SeldonIO/alibi-detect Alibi Detect is a Python library providing algorithms for outlier, adversarial, and drift detection across tabular data, text, images, and … | 49 | 2548 | active |
| yifan123/flow_grpo Flow-GRPO is the official PyTorch implementation of a NeurIPS 2025 paper that trains flow matching models (e.g., SD3.5, FLUX.1, Qwen-Image,… | 52 | 2543 | active |
| pydata/numexpr NumExpr is a fast numerical expression evaluator for NumPy arrays that compiles expressions like '3*a+4*b' into op-codes executed by an int… | 93 | 2542 | stable |
| SafeAILab/EAGLE EAGLE is the official implementation of the EAGLE family of speculative decoding algorithms (EAGLE-1/2/3) for accelerating LLM text generat… | 54 | 2542 | active |
| learning-at-home/hivemind Hivemind is a PyTorch library for decentralized deep learning across the Internet, enabling training of large models on hundreds of volunte… | 53 | 2541 | active |
| X-PLUG/mPLUG-Owl mPLUG-Owl is a family of open-source multimodal large language models that combine visual encoders with LLMs (built on LLaMA) for image and… | 33 | 2539 | active |
| Roboto Origin A fully open-source DIY humanoid robot project (ROBOTO_ORIGIN) that can walk and run, including all hardware designs, electronics, training… | 60 | 2537 | active |
| VITA-MLLM/VITA VITA is an open-source interactive omni multimodal large language model (VITA-1.5) that supports real-time vision and speech interaction, s… | 26 | 2536 | active |
| chatdoc-com/OCRFlux OCRFlux is a Python toolkit built on a 3B-parameter vision-language model that converts PDFs and images into clean Markdown, handling compl… | 50 | 2532 | active |
| LyCORIS LyCORIS is a Python library implementing parameter-efficient fine-tuning algorithms (LoRA/LoCon, LoHa, LoKr, IA3, DyLoRA, and more) for Sta… | 99 | 2530 | active |
| rpautrat/SuperPoint A TensorFlow (with PyTorch conversion) implementation of the SuperPoint self-supervised interest point detector and descriptor network. It … | 37 | 2528 | stable |
| mflux-community/mflux MFLUX is a native MLX implementation of state-of-the-art generative image and video models (Flux, Qwen-Image, Z-Image, and others), ported … | 91 | 2526 | active |
| FacePlugin Face Recognition SDK An open-source face recognition SDK by Faceplugin providing face detection, landmark extraction, feature embedding generation, and face tem… | 65 | 2526 | active |
| Farama-Foundation/Minigrid Minigrid is a Python library of lightweight, easily configurable discrete grid-world environments for reinforcement learning research, foll… | 80 | 2523 | active |
| bilibili/Index-anisora Index-AniSora is Bilibili's open-source anime video generation model, capable of creating video shots in diverse anime styles from images, … | 60 | 2523 | active |
| amap-cvlab/ABot-World ABot-World is an open-source interactive world model that generates infinite, explorable video worlds from a single image, running on a sin… | 57 | 2513 | active |
| oegedijk/explainerdashboard A Python library for quickly building interactive Explainable AI dashboards that explain the inner workings of scikit-learn compatible mach… | 68 | 2512 | active |
| HiDream-ai/HiDream-I1 HiDream-I1 is an open-source 17B-parameter text-to-image generative foundation model based on a Sparse Diffusion Transformer, with full and… | 31 | 2510 | active |
| luanfujun/deep-photo-styletransfer Reference implementation of the CVPR 2017 paper 'Deep Photo Style Transfer', performing photorealistic image style transfer using Torch wit… | 32 | 9982 | maintenance |
| Epic Awesome Gamer A Python library that solves hCaptcha challenges using multimodal large language models and ONNX vision models (YOLO, CLIP, ResNet) integra… | 80 | 2508 | active |
| google-deepmind/sonnet Sonnet is a Python library built on TensorFlow 2 that provides simple, composable abstractions (snt.Module) for constructing neural network… | 66 | 9973 | maintenance |
| scikit-learn-contrib/category_encoders A scikit-learn-contrib library providing a collection of sklearn-compatible transformers for encoding categorical variables into numeric fo… | 91 | 2506 | active |
| yformer/EfficientSAM EfficientSAM is an efficient image segmentation model that leverages masked image pretraining to provide a lightweight alternative to Meta'… | 27 | 2498 | active |
| sthalles/SimCLR A PyTorch reference implementation of SimCLR, a self-supervised contrastive learning framework for learning visual representations from unl… | 23 | 2494 | stable |
| google/tunix Tunix is a lightweight JAX-based library for post-training large language models, supporting supervised fine-tuning, preference optimizatio… | 81 | 2491 | active |
| ipazc/mtcnn A Python library implementing the MTCNN (Multitask Cascaded Convolutional Networks) algorithm for face detection and facial landmark alignm… | 23 | 2490 | stable |
| GaParmar/img2img-turbo A research library implementing one-step image-to-image translation models (CycleGAN-Turbo and pix2pix-turbo) built on SD-Turbo diffusion m… | 38 | 2488 | active |
| ppogg/YOLOv5-Lite YOLOv5-Lite is a lightweight object detection model family evolved from YOLOv5, with models as small as ~900KB (int8) that run 10-15+ FPS o… | 23 | 2488 | active |
| visgl/luma.gl luma.gl is a modular TypeScript toolkit for high-performance GPU rendering and compute on the web, providing a portable device API over Web… | 98 | 2485 | active |
| wgsxm/PartCrafter PartCrafter is a structured 3D generative model that jointly generates multiple semantically meaningful 3D mesh parts and objects from a si… | 50 | 2485 | active |
| thuml/Autoformer Autoformer is the official PyTorch implementation of the NeurIPS 2021 paper 'Autoformer: Decomposition Transformers with Auto-Correlation f… | 32 | 2485 | stable |
| homuler/MediaPipeUnityPlugin A Unity native plugin that ports the MediaPipe C++ API to C#, enabling MediaPipe graphs and solutions to run inside Unity applications. It … | 56 | 2484 | active |
| google/XNNPACK XNNPACK is a highly optimized C library of floating-point neural network inference operators for ARM, x86, WebAssembly, and RISC-V platform… | 77 | 2481 | active |
| alexjc/neural-doodle A Python implementation of Semantic Style Transfer (Champandard, 2016) based on the Neural Patches algorithm. It turns rough doodles into r… | 10 | 9851 | maintenance |
| chemprop/chemprop Chemprop is a Python package implementing message passing neural networks (MPNNs) for predicting molecular and reaction properties. It prov… | 88 | 2480 | active |
| Lightning-AI/torchmetrics TorchMetrics is a PyTorch-native library of machine learning metrics that work in distributed and scalable training pipelines. It provides … | 87 | 2477 | stable |
| roboflow/inference Roboflow Inference is a Python library and self-hostable inference server for deploying computer vision models on any computer or edge devi… | 91 | 2474 | active |
| sanderland/katrain KaTrain is a cross-platform desktop application for playing and analyzing the game of Go (Baduk/Weiqi) with AI feedback powered by KataGo. … | 89 | 2474 | active |
| jadore801120/attention-is-all-you-need-pytorch A PyTorch implementation of the Transformer model from the paper 'Attention Is All You Need', including training and translation scripts fo… | 32 | 9816 | maintenance |
| TrashUwU/PokeAssistant A Discord assistant bot for the Pokétwo game that automatically identifies Pokémon from spawn images using AI, pings roles or users when le… | 10 | 2470 | active |
| inference-labs-inc/subnet-2 Subnet 2 on the Bittensor network implements a Proof-of-Inference system that uses zero-knowledge machine learning (zk-ML) to cryptographic… | 86 | 2462 | active |
| Farama-Foundation/Arcade-Learning-Environment The Arcade Learning Environment (ALE) is a framework built on the Stella Atari 2600 emulator that lets researchers and hobbyists develop an… | 92 | 2458 | stable |
| sqliteai/warp WARP is a dependency-free, embeddable C inference engine that runs trillion-parameter mixture-of-experts models like Kimi K3 by streaming a… | 77 | 2457 | active |
| Alibaba-Quark/LiveAvatar LiveAvatar is an open-source implementation of an ECCV 2026 paper for streaming, real-time, infinite-length audio-driven avatar video gener… | 58 | 2456 | active |
| metarank/metarank Metarank is an open-source, low-code Learning-to-Rank service that personalizes search results, listings, and recommendations using real-ti… | 85 | 2454 | active |
| google-parfait/tensorflow-federated TensorFlow Federated (TFF) is an open-source Python framework for machine learning and other computations on decentralized data. It provide… | 67 | 2454 | active |
| tracel-ai/cubecl CubeCL is a Rust language extension, JIT compiler, and runtime set for writing high-performance GPU compute kernels in plain Rust. A single… | 88 | 2453 | active |
| facebookresearch/pifuhd PIFuHD is a PyTorch implementation of a CVPR 2020 research model that reconstructs high-resolution 3D human body meshes from a single 2D im… | 10 | 9724 | maintenance |
| Unakar/Logic-RL Logic-RL is a research framework that reproduces R1-Zero-style LLM reasoning via rule-based reinforcement learning (GRPO) on logic puzzles,… | 23 | 2452 | active |
| AI-Hypercomputer/maxtext MaxText is a high-performance, scalable open-source LLM training library written in pure Python/JAX, targeting Google Cloud TPUs and GPUs. … | 98 | 2446 | active |
| TorchIO-project/torchio TorchIO is a Python library for loading, augmenting, and processing 3D medical images (MRI, CT) within PyTorch deep learning pipelines. It … | 91 | 2446 | active |
| skfolio/skfolio skfolio is a Python library for portfolio optimization, factor model construction, and risk management built on top of scikit-learn. It pro… | 88 | 2446 | active |
| BeyondMimic Motion Tracking BeyondMimic's official motion tracking training code, a humanoid control framework built on Isaac Lab that trains sim-to-real-ready whole-b… | 10 | 2444 | active |
| hitsz-ids/synthetic-data-generator SDG (Synthetic Data Generator) is a Python framework for generating high-quality synthetic structured tabular data using deep learning, GAN… | 75 | 2440 | active |
| iliane5/meridian Meridian is an open-source AI-powered news intelligence system that scrapes hundreds of RSS sources, clusters and analyzes articles with LL… | 27 | 2437 | active |
| NovaSky-AI/SkyRL SkyRL is a modular full-stack reinforcement learning library for post-training large language models, combining a training framework (skyrl… | 83 | 2436 | active |
| shenweichen/DeepMatch DeepMatch is a Python library of deep matching models for recommendations and advertising, built on TensorFlow/Keras. It lets users train m… | 65 | 2435 | active |
| resemble-ai/resemble-enhance Resemble Enhance is an AI-powered Python tool that improves speech quality through denoising and enhancement, using a denoiser module and a… | 18 | 2434 | active |
| Statistical Rethinking Course An R package accompanying McElreath's Statistical Rethinking book and course on Bayesian data analysis. It provides tools for quadratic app… | 58 | 2427 | active |
| wolny/pytorch-3dunet A PyTorch implementation of 3D U-Net and its variants (residual, squeeze-and-excitation) for volumetric semantic segmentation, with 2D U-Ne… | 57 | 2423 | active |
| HuCaoFighting/Swin-Unet Official PyTorch implementation of Swin-Unet, a U-shaped pure Transformer model for medical image segmentation, published at ECCV 2022 Medi… | 38 | 2420 | stable |
| tflearn/tflearn TFLearn is a modular deep learning library providing a higher-level, Keras-like API on top of TensorFlow for building and training neural n… | 23 | 9570 | maintenance |
| Smorodov/Multitarget-tracker A C++ library for multiple object tracking that combines detectors (YOLO, D-FINE, RF-DETR, MobileNet-SSD) with tracking algorithms based on… | 67 | 2418 | active |
| X-PLUG/mPLUG-DocOwl mPLUG-DocOwl is a family of multimodal large language models from Alibaba for OCR-free document understanding, including DocOwl1.5 and DocO… | 37 | 2413 | active |
| Jittor/JittorLLMs JittorLLMs is a large language model inference library built on the Jittor deep learning framework, supporting models like ChatGLM, LLaMA/L… | 31 | 2406 | active |
| fal-ai-community/video-starter-kit A Next.js-based starter kit for building AI-powered video applications in the browser, integrating fal.ai video models (Minimax, Hunyuan, L… | 30 | 2403 | active |
| Olive Olive is Microsoft's AI model optimization toolkit for the ONNX Runtime, automating finetuning, conversion, quantization, and compression o… | 89 | 2402 | active |
| OpenGVLab/InternVideo InternVideo is a series of open-source video foundation models for multimodal video understanding, spanning generative and discriminative l… | 69 | 2393 | active |
| zai-org/GLM-V GLM-V is the open-source repository for Zhipu AI's GLM-4.6V, GLM-4.5V, and GLM-4.1V-Thinking vision-language models, which perform versatil… | 63 | 2393 | active |
| akfamily/akquant AKQuant is a high-performance quantitative research and backtesting framework with a Rust core engine and Python interface for strategy dev… | 76 | 2392 | active |
| NVlabs/ProtoMotions ProtoMotions is a GPU-accelerated simulation and reinforcement learning framework for training physically simulated digital humans and huma… | 66 | 2392 | active |
| sjwhitworth/golearn GoLearn is a 'batteries included' machine learning library for Go with a scikit-learn-style Fit/Predict interface. It provides data loading… | 32 | 9436 | maintenance |
| Hedgehog-Computing/hedgehog-lab Hedgehog Lab is an open-source scientific computing environment that runs entirely in the browser, offering JavaScript-based matrix operati… | 23 | 2389 | active |
| AminHP/gym-anytrading gym-anytrading is a collection of OpenAI Gym environments for developing and testing reinforcement learning-based trading algorithms in FOR… | 23 | 2386 | stable |
| google/neural-tangents Neural Tangents is a Python library built on JAX for defining, training, and evaluating neural networks of both finite and infinite width. … | 10 | 2385 | stable |
| TencentARC/Pixal3D Pixal3D is a research codebase for generating high-fidelity 3D assets from a single image using a pixel-aligned generation paradigm that ba… | 56 | 2383 | active |
| facebookresearch/perception_models Meta's Perception Models repository hosting state-of-the-art image, video, and audio encoders (Perception Encoder, PE) and a multimodal lan… | 51 | 2382 | active |
| jik876/hifi-gan The official PyTorch implementation of HiFi-GAN, a generative adversarial network that converts mel-spectrograms into high-fidelity 22.05 k… | 32 | 2377 | stable |
| Turbo1123/roubao Roubao is an open-source AI phone automation assistant for Android, built natively in Kotlin and powered by vision-language models. It runs… | 47 | 2376 | active |
| apple/axlearn AXLearn is a Python deep learning library built on JAX and XLA for developing and training large-scale models, with an object-oriented conf… | 69 | 2375 | active |
| XiaomiMiMo/MiMo Xiaomi's MiMo is a 7B-parameter reasoning language model trained from pretraining through posttraining with reinforcement learning, release… | 28 | 2373 | active |
| FMInference/FlexLLMGen FlexLLMGen is a high-throughput generation engine for running large language models on a single GPU with limited memory. It uses IO-efficie… | 10 | 9346 | maintenance |
| MouseLand/cellpose Cellpose is a generalist deep learning algorithm for cellular and nucleus segmentation in microscopy images, with human-in-the-loop capabil… | 80 | 2371 | active |
| fastino-ai/GLiNER2 GLiNER2 is a Python library for schema-based information extraction and text classification, supporting named entity recognition, structure… | 85 | 2369 | active |
| Tongyi-MAI/MAI-UI Qwen-UI-Agent (MAI-UI) is a foundation GUI agent model from Alibaba's Tongyi-MAI team that unifies mobile, desktop, browser, and deep-resea… | 57 | 2369 | active |
| neuropsychology/NeuroKit NeuroKit2 is a Python library for neurophysiological and biosignal processing, offering easy access to advanced routines for signals like E… | 76 | 2366 | active |
| TransWithAI/Faster-Whisper-TransWithAI-ChickenRice A high-performance audio/video transcription and translation application built on Faster Whisper, optimized for Japanese-to-Chinese transla… | 71 | 2366 | active |
| PYNQ PYNQ is an open-source Python framework from AMD/Xilinx for designing embedded systems on Zynq and other adaptive computing platforms (FPGA… | 93 | 2364 | active |
| meizhong986/WhisperJAV WhisperJAV is a local, privacy-preserving subtitle generator for Japanese adult videos, combining Qwen3-ASR, Whisper, TEN-VAD/FireRedVAD vo… | 95 | 2362 | active |
| EasyRec EasyRec is a TensorFlow-based framework from Alibaba for building large-scale deep learning recommendation models covering matching, rankin… | 59 | 2361 | active |