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 |
|---|---|---|---|
| google-deepmind/weathernext Google DeepMind's WeatherNext family of AI weather forecasting models, including WeatherNext 2, GraphCast, and GenCast, with code and pretr… | 82 | 7709 | active |
| meta-llama/llama-models Meta's official repository of Llama large language models with Python utilities for working with them, including model cards, licenses, and… | 53 | 7693 | active |
| alteryx/featuretools Featuretools is an open-source Python library for automated feature engineering, using Deep Feature Synthesis (DFS) to generate features fr… | 65 | 7687 | active |
| openai/gym Gym is a Python toolkit for developing and comparing reinforcement learning algorithms via a standard API between learning algorithms and e… | 10 | 37236 | maintenance |
| myshell-ai/MeloTTS MeloTTS is a high-quality multi-lingual text-to-speech library supporting English (multiple accents), Spanish, French, Chinese, Japanese, a… | 17 | 7644 | active |
| babysor/MockingBird MockingBird is a PyTorch-based AI voice cloning toolbox that can clone a voice from a 5-second sample and generate arbitrary speech in real… | 51 | 36895 | maintenance |
| lllyasviel/Omost Omost is a Python application that converts LLM coding capability into image composition by having pretrained LLMs (based on Llama3 and Phi… | 25 | 7595 | active |
| SkyReels SkyReels-V2 is an open-source infinite-length film/video generative model using an AutoRegressive Diffusion-Forcing architecture, released … | 45 | 7589 | active |
| HIPS/autograd Autograd is a Python library that efficiently computes derivatives of native Python and NumPy code, supporting reverse-mode (backpropagatio… | 82 | 7543 | stable |
| face-alignment A Python library built on PyTorch that detects 2D and 3D facial landmarks in images using the FAN deep learning face alignment network. It … | 64 | 7543 | active |
| algorithmicsuperintelligence/openevolve OpenEvolve is an open-source Python implementation of DeepMind's AlphaEvolve, an evolutionary coding agent that uses LLM ensembles to auton… | 86 | 7518 | active |
| SAM 3D SAM 3D Objects is a foundation model from Meta that reconstructs full 3D shape geometry, texture, and layout from a single image, with code… | 52 | 7516 | active |
| H2O H2O-3 is an open-source, distributed, in-memory machine learning platform implementing algorithms such as GLM, GBM/XGBoost, Random Forest, … | 75 | 7509 | stable |
| zai-org/GLM-OCR GLM-OCR is an open-source 0.9B-parameter multimodal OCR model built on the GLM-V encoder-decoder architecture for complex document understa… | 73 | 7504 | active |
| open-compass/opencompass OpenCompass is a Python-based evaluation platform and toolkit for benchmarking large language models across 100+ datasets, supporting model… | 93 | 7471 | active |
| EleutherAI/gpt-neox GPT-NeoX is EleutherAI's library for training large-scale autoregressive transformer language models on GPUs, built on NVIDIA's Megatron an… | 67 | 7471 | active |
| TheCraigHewitt/seomachine SEO Machine is a specialized Claude Code workspace of custom commands, AI agents, and Python analysis modules for researching, writing, and… | 57 | 7467 | active |
| facebookresearch/SlowFast PySlowFast is a PyTorch-based open-source video understanding codebase from Facebook AI Research (FAIR). It provides implementations of sta… | 61 | 7432 | active |
| LargeWorldModel/LWM Large World Model (LWM) is a family of open-source 7B-parameter multimodal autoregressive transformer models trained on long videos and boo… | 26 | 7425 | active |
| apple/ml-fastvlm Official implementation of FastVLM, a vision language model with an efficient hybrid vision encoder (FastViTHD) that reduces token count an… | 62 | 7423 | active |
| arcee-ai/mergekit mergekit is a Python toolkit for merging pre-trained large language models directly in weight space, supporting many merge methods (SLERP, … | 64 | 7389 | active |
| Final2x Final2x is a cross-platform desktop application for image super-resolution (upscaling) built with Electron, Vue3, and a PyTorch-based Pytho… | 91 | 7370 | active |
| imgly/background-removal-js An npm package (browser and Node.js variants) that removes image backgrounds using ONNX-based image segmentation/matting models running ent… | 40 | 7354 | active |
| vladmandic/sdnext SD.Next is an open-source, self-hosted WebUI server application for AI generative image and video creation built on Stable Diffusion and Di… | 77 | 7353 | active |
| google/flax Flax is a neural network library and ecosystem for JAX designed for flexibility, featuring the newer NNX API with first-class Python refere… | 99 | 7348 | active |
| zai-org/GLM-5 GLM-5 is a family of open-weight large language models from Z.ai (Zhipu AI), including the flagship GLM-5.2 with 1M-token context and the m… | 60 | 7329 | active |
| Feast Feast is an open-source feature store for machine learning that manages offline stores for historical training data and low-latency online … | 99 | 7327 | stable |
| naver/dust3r DUSt3R is the official PyTorch implementation of a CVPR 2024 model that performs dense, unconstrained stereo and multi-view 3D reconstructi… | 42 | 7317 | active |
| Paddle Lite Paddle Lite is a high-performance, lightweight deep learning inference engine from Baidu's PaddlePaddle ecosystem, designed for mobile, emb… | 55 | 7277 | active |
| InternLM/InternLM Official repository for the InternLM series of open-source large language models (InternLM, InternLM2, InternLM2.5, InternLM3), including p… | 38 | 7276 | active |
| mit-han-lab/streaming-llm StreamingLLM is a research framework from MIT Han Lab implementing the Attention Sinks method (ICLR 2024) for efficient streaming language … | 28 | 7267 | stable |
| liuliu/ccv ccv is a modern, minimalist computer vision library written in C/C++ with an application-driven set of state-of-the-art algorithms includin… | 77 | 7248 | active |
| kohya-ss/sd-scripts A collection of Python training, generation, and utility scripts for Stable Diffusion and other image generation models, most widely used f… | 96 | 7242 | active |
| Zyphra/Zonos Zonos-v0.1 is an open-weight text-to-speech model trained on over 200k hours of multilingual speech, with a Python library for inference. I… | 22 | 7236 | active |
| stanfordnlp/GloVe Stanford's C implementation of the GloVe model for training distributed word representations (word embeddings), plus downloadable pre-train… | 34 | 7230 | stable |
| AI Scientist An end-to-end autonomous AI research system that generates hypotheses, runs machine learning experiments, analyzes data, and writes complet… | 43 | 7222 | active |
| MineDojo/Voyager Voyager is an open-source LLM-powered embodied agent that plays Minecraft through GPT-4, using an automatic curriculum, an ever-growing ski… | 30 | 7219 | active |
| NVIDIA/warp NVIDIA Warp is a Python framework that JIT-compiles regular Python functions into efficient CPU or CUDA GPU kernels. It provides differenti… | 99 | 7215 | active |
| deepseek-ai/DeepSpec DeepSpec is a full-stack Python codebase from DeepSeek for training and evaluating draft models used in speculative decoding of large langu… | 51 | 7207 | active |
| BVLC/caffe Caffe is a fast, modular deep learning framework developed by Berkeley AI Research, focused on convolutional neural networks for vision, sp… | 23 | 34549 | maintenance |
| OpenPose OpenPose is a real-time multi-person keypoint detection library that jointly estimates human body, face, hand, and foot keypoints (135 tota… | 23 | 34493 | maintenance |
| tensortrade-org/tensortrade TensorTrade is an open-source Python framework for building, training, and evaluating reinforcement learning agents for algorithmic trading… | 58 | 7194 | active |
| flwrlabs/flower Flower (flwr) is an open-source Python framework for building federated and collaborative AI systems, supporting any ML framework such as P… | 99 | 7162 | active |
| ControlNet ControlNet is a neural network architecture that adds conditional control (edges, poses, depth, etc.) to pretrained text-to-image diffusion… | 32 | 34128 | maintenance |
| testerSunshine/12306 A Python-based ticket-sniping assistant for China Railway's 12306 booking system that automates login, CAPTCHA recognition, and ticket purc… | 10 | 34068 | maintenance |
| scikit-learn-contrib/imbalanced-learn imbalanced-learn is a Python library offering re-sampling techniques (over-sampling, under-sampling, and combinations) for datasets with st… | 76 | 7120 | stable |
| Vespa Vespa is an open-source, distributed AI search platform and serving engine that combines full-text search, vector/tensor search, and machin… | 93 | 7117 | stable |
| yangchris11/samurai SAMURAI is the official implementation of a zero-shot visual object tracker built on top of Segment Anything Model 2 (SAM 2), using a motio… | 24 | 7116 | active |
| albertan017/LLM4Decompile LLM4Decompile is an open-source series of large language models (1.3B to 33B) trained to decompile binary code back into readable, executab… | 53 | 7076 | active |
| threestudio-project/threestudio threestudio is a unified open-source framework for 3D content generation from text prompts, single images, and few-shot images by lifting 2… | 21 | 7069 | active |
| google/gemma.cpp A lightweight, standalone C++ inference engine for Google's Gemma foundation models (Gemma 2/3, PaliGemma 2), with a small ~2K LoC core and… | 73 | 7060 | active |
| DeepChem DeepChem is a Python library that democratizes deep learning for drug discovery, quantum chemistry, materials science, and biology. It prov… | 63 | 7050 | active |
| TheTom/turboquant_plus TurboQuant+ is a Python reference implementation of the TurboQuant KV cache compression method (ICLR 2026), using PolarQuant codebooks and … | 54 | 7036 | active |
| latentcat/qrbtf QRBTF is an AI and parametric QR code generator that creates stylized, scannable QR codes, available as a hosted web app at qrbtf.com with … | 27 | 7030 | active |
| deepseek-ai/DeepSeek-Coder-V2 DeepSeek-Coder-V2 is an open-weight Mixture-of-Experts code language model family released by DeepSeek AI, with weights available on Huggin… | 44 | 7012 | active |
| apple/corenet CoreNet is Apple's deep neural network training toolkit for training standard and novel small and large-scale models, including foundation … | 68 | 7008 | active |
| VAST-AI-Research/TripoSR TripoSR is an open-source model for fast feedforward 3D object reconstruction from a single image, developed by Tripo AI and Stability AI. … | 61 | 7005 | active |
| gepa-ai/gepa GEPA is a Python framework that optimizes textual system parameters such as prompts, code, and agent configurations using LLM-based reflect… | 81 | 6996 | active |
| deeppavlov/DeepPavlov DeepPavlov is an open-source Python NLP library built on PyTorch and Hugging Face transformers for developing, training, and deploying stat… | 35 | 6995 | active |
| sczhou/ProPainter ProPainter is a PyTorch-based video inpainting model from ICCV 2023 that combines dual-domain propagation with a mask-guided sparse video T… | 19 | 6964 | stable |
| InterpretML InterpretML is a Python library that unifies machine learning interpretability techniques, letting users train inherently interpretable gla… | 89 | 6961 | active |
| PaddlePaddle/models PaddlePaddle's officially maintained industry-grade model repository containing 600+ models across computer vision, NLP, speech, recommenda… | 23 | 6931 | active |
| eriklindernoren/ML-From-Scratch A Python library providing bare-bones NumPy implementations of fundamental machine learning models and algorithms, from linear regression t… | 32 | 32980 | maintenance |
| ClearML ClearML is an open-source Python SDK and platform for MLOps/LLMOps that provides auto-logged experiment tracking, data versioning, pipeline… | 96 | 6903 | stable |
| TencentQQGYLab/AppAgent AppAgent is an LLM-based multimodal agent framework that operates smartphone apps through human-like interactions such as tapping and swipi… | 29 | 6896 | active |
| SenseNova SenseNova-U is a series of open-weight unified multimodal models (e.g., SenseNova-U1.5-8B-MoT) built on the NEO-unify architecture that com… | 58 | 6884 | active |
| dramaclaw/dramaclaw DramaClaw is a self-hosted, general-purpose AIGC video engine that turns scripts into finished films through a single pipeline covering sto… | 81 | 6851 | active |
| postgresml/postgresml PostgresML is a PostgreSQL extension that brings machine learning and AI inference directly into the database, with GPU acceleration and pg… | 32 | 6823 | active |
| NicolasHug/Surprise Surprise is a Python scikit for building and analyzing recommender systems that work with explicit rating data. It provides ready-to-use pr… | 67 | 6817 | stable |
| facebookresearch/fairseq Fairseq is a PyTorch-based sequence modeling toolkit from Facebook AI Research for training custom models for translation, summarization, l… | 10 | 32211 | maintenance |
| FoundationVision/ByteTrack ByteTrack is a PyTorch-based multi-object tracking (MOT) library implementing the ECCV 2022 paper 'Multi-Object Tracking by Associating Eve… | 32 | 6740 | stable |
| kijai/ComfyUI-WanVideoWrapper ComfyUI custom nodes wrapping WanVideo (Wan2.1) and related video generation models. It provides a standalone sandbox for quickly implement… | 55 | 6737 | active |
| flashinfer-ai/flashinfer FlashInfer is a GPU kernel library and kernel generator for LLM inference, providing unified APIs for attention, GEMM, and MoE operations w… | 92 | 6728 | active |
| guofei9987/scikit-opt A Python library implementing heuristic and swarm intelligence optimization algorithms, including genetic algorithm, particle swarm optimiz… | 52 | 6720 | active |
| google-ai-edge/LiteRT-LM LiteRT-LM is Google's production-ready, high-performance open-source framework for running large language models on edge devices, built as … | 88 | 6702 | active |
| tencent-ailab/IP-Adapter IP-Adapter is a lightweight (22M parameter) adapter that adds image prompt capability to pretrained text-to-image diffusion models like Sta… | 29 | 6694 | stable |
| simplescaling/s1 s1 is an open-source research project implementing simple test-time scaling for large language models, including the s1K dataset of 1,000 c… | 30 | 6675 | active |
| linkedin/Liger-Kernel Liger Kernel is a collection of Triton kernels designed to make LLM training faster and more memory-efficient, offering drop-in replacement… | 91 | 6664 | active |
| yangjianxin1/Firefly Firefly is an open-source one-stop training tool for large language models, supporting pretraining, instruction fine-tuning (SFT), and DPO … | 21 | 6653 | active |
| TMElyralab/MuseTalk MuseTalk is a real-time, high-fidelity lip-sync model that modifies a face region in video according to input audio via latent space inpain… | 42 | 6651 | active |
| Yuliang-Liu/MonkeyOCR MonkeyOCR is a lightweight large multimodal model (LMM) for document parsing that uses a Structure-Recognition-Relation triplet paradigm to… | 58 | 6650 | active |
| ml5.js ml5.js is a friendly, beginner-oriented JavaScript machine learning library for the browser, built on top of TensorFlow.js. It provides acc… | 23 | 6583 | active |
| axa-group/nlp.js NLP.js is a natural language processing library for Node.js that provides intent classification, named entity recognition, sentiment analys… | 23 | 6580 | active |
| AILab-CVC/YOLO-World YOLO-World is a real-time open-vocabulary object detection model and Python toolkit from Tencent AI Lab and HUST, published at CVPR 2024. I… | 26 | 6579 | active |
| iperov/DeepFaceLive DeepFaceLive is a real-time face-swap application for PC streaming and video calls, using trained face models (DFM) applied to webcam or vi… | 10 | 31022 | maintenance |
| open-mmlab/mmdetection3d MMDetection3D is OpenMMLab's next-generation platform for general 3D object detection, built on PyTorch. It provides a modular toolbox with… | 23 | 6535 | active |
| Argos Translate Argos Translate is an open-source offline neural machine translation library written in Python, built on OpenNMT. It can be used as a Pytho… | 62 | 6532 | active |
| shibing624/pycorrector pycorrector is a Python toolkit for Chinese text error correction, supporting phonetic, visual, and grammar error types with models includi… | 78 | 6528 | active |
| rtqichen/torchdiffeq torchdiffeq is a PyTorch library of differentiable ordinary differential equation (ODE) solvers, best known as the canonical implementation… | 35 | 6482 | stable |
| PufferLib PufferLib is a fast, open-source reinforcement learning library that trains tiny, super-human models in seconds, achieving 1M+ environment … | 87 | 6476 | active |
| open-mmlab/mmcv MMCV is the foundational computer vision library for the OpenMMLab ecosystem, providing image/video I/O, data transformations, and CUDA ope… | 48 | 6475 | stable |
| HVision-NKU/StoryDiffusion StoryDiffusion is the official implementation of a NeurIPS 2024 Spotlight paper introducing Consistent Self-Attention for character-consist… | 25 | 6475 | active |
| DEAP DEAP is a Python framework for evolutionary computation that supports genetic algorithms, genetic programming, evolution strategies, and mu… | 63 | 6450 | stable |
| ByteDance-Seed/Depth-Anything-3 Depth Anything 3 (DA3) is a transformer-based model that predicts spatially consistent depth and geometry from any number of visual inputs,… | 56 | 6445 | active |
| haifengl/smile SMILE is a comprehensive, high-performance machine learning framework for the JVM with idiomatic APIs for Java, Scala, and Kotlin. It cover… | 100 | 6423 | active |
| WhisperKit A Swift SDK providing turn-key on-device speech AI frameworks for Apple Silicon, including WhisperKit (speech-to-text with Whisper), Speake… | 89 | 6397 | active |
| mayocream/koharu Koharu is a local-first desktop application that automates manga translation using machine learning, combining text/bubble detection, OCR, … | 82 | 6394 | active |
| tensorflow/serving TensorFlow Serving is a flexible, high-performance serving system for machine learning models designed for production environments. It mana… | 89 | 6364 | stable |
| StyleTTS StyleTTS 2 is a PyTorch text-to-speech model that uses style diffusion and adversarial training with large speech language models (e.g., Wa… | 30 | 6362 | active |
| KevinMusgrave/pytorch-metric-learning A PyTorch library providing modular losses, miners, samplers, and testers for deep metric learning. It supports building complete train/tes… | 43 | 6342 | active |