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 |
|---|---|---|---|
| InterpretML InterpretML is a Python library that unifies machine learning interpretability techniques, letting users train inherently interpretable gla… | 89 | 6961 | 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 |
| drumih/turbo-fieldfare TurboFieldfare is a custom Swift + Metal runtime that runs Gemma 4 26B-A4B inference in roughly 2 GB of RAM on Apple Silicon Macs by stream… | 78 | 6880 | 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… | 66 | 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 |
| Andyyyy64/whichllm whichllm is a Python CLI tool that auto-detects your GPU, CPU, and RAM, then ranks local LLMs from HuggingFace that will actually run and p… | 81 | 6724 | active |
| 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 |
| 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 |
| 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 |
| 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 |
| 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 |
| spotify/pedalboard Pedalboard is a Python library by Spotify for reading, writing, and processing audio, with built-in effects like reverb, distortion, and EQ… | 92 | 6329 | active |
| Pachyderm Pachyderm is a data-centric pipeline platform that automates data transformations with built-in data versioning and lineage tracking. It ru… | 30 | 6309 | active |
| Aim Aim is an open-source experiment tracker for AI/ML that logs training runs and metadata, with a web UI to compare runs and a Python API to … | 74 | 6276 | active |
| PaddlePaddle/PaddleX PaddleX is a low-code, all-in-one AI development tool built on the PaddlePaddle framework, bundling 200+ pretrained models into 33 producti… | 86 | 6268 | active |
| tyiannak/pyAudioAnalysis pyAudioAnalysis is a Python library for audio analysis covering feature extraction (MFCCs, spectrograms, chromagrams), supervised and unsup… | 44 | 6268 | active |
| meta-pytorch/gpt-fast A minimal (<1000 lines) PyTorch-native implementation of fast transformer text generation, demonstrating low-latency LLM inference with int… | 41 | 6259 | active |
| Trusted-AI/adversarial-robustness-toolbox Adversarial Robustness Toolbox (ART) is a Python library for machine learning security covering evasion, poisoning, extraction, and inferen… | 56 | 6250 | stable |
| open-edge-platform/anomalib Anomalib is a Python deep learning library for anomaly detection, offering state-of-the-art unsupervised algorithms for detecting and local… | 96 | 6196 | active |
| ladaapp/lada Lada is an open-source tool with both GUI and CLI that restores pixelated/mosaic regions in videos, primarily targeting JAV (Japanese adult… | 59 | 6193 | active |
| skorch-dev/skorch skorch is a Python library that wraps PyTorch neural networks in a scikit-learn compatible API, providing estimators like NeuralNetClassifi… | 81 | 6179 | active |
| Akegarasu/lora-scripts SD-Trainer is a GUI application and set of scripts for training LoRA and Dreambooth fine-tunes of Stable Diffusion diffusion models, wrappi… | 66 | 6166 | active |
| FlashDrive DFlash is a lightweight block diffusion model used as a draft model for speculative decoding of large language models, drafting entire toke… | 65 | 6134 | active |
| KServe KServe is a CNCF incubating, Kubernetes-native platform for serving both generative and predictive AI models at scale. It provides a standa… | 94 | 6133 | stable |
| timeseriesAI/tsai tsai is an open-source deep learning library built on PyTorch and fastai for time series and sequential data tasks such as classification, … | 77 | 6124 | active |
| Cactus Cactus is a hybrid edge-cloud AI inference engine for mobile devices, wearables, smart home devices, and robots, built in C++ with custom q… | 85 | 6111 | active |
| online-ml/river River is a Python library for online machine learning, allowing models to learn incrementally from streaming data one observation at a time… | 90 | 6109 | active |
| FATE FATE (Federated AI Technology Enabler) is an industrial-grade open-source federated learning framework hosted by the Linux Foundation. It e… | 23 | 6107 | active |
| chaiNNer chaiNNer is a free, open-source, node-based desktop application for building image processing pipelines by connecting nodes on a canvas. Or… | 89 | 6094 | active |
| OFA-Sys/Chinese-CLIP Chinese-CLIP is a Chinese version of the CLIP model trained on ~200 million Chinese image-text pairs, built on open_clip. It provides APIs,… | 62 | 6014 | active |
| uber/causalml Causal ML is a Python package providing uplift modeling and causal inference methods built on machine learning algorithms. It estimates the… | 82 | 6013 | stable |
| amazon-science/chronos-forecasting Chronos is a Python library providing pretrained foundation models for time series forecasting, including Chronos-2 which handles univariat… | 91 | 5999 | active |
| volcano-sh/volcano Volcano is a CNCF-hosted, Kubernetes-native batch scheduling system that extends kube-scheduler for high-performance workloads like AI/ML t… | 98 | 5993 | stable |
| fla-org/flash-linear-attention A PyTorch library providing hardware-efficient implementations of emerging sequence model architectures, including linear attention, sparse… | 86 | 5940 | active |
| Giskard-AI/giskard-oss Giskard is an open-source Python library for testing, evaluating, and red-teaming LLM-based agents and RAG systems. Its v3 rewrite provides… | 95 | 5910 | active |
| Livebook Livebook is an open-source web application for writing interactive and collaborative Elixir notebooks, mixing Markdown prose with executabl… | 98 | 5888 | active |
| ChaoningZhang/MobileSAM MobileSAM is the official implementation of a lightweight version of Meta's Segment Anything Model (SAM), replacing the heavyweight image e… | 62 | 5885 | stable |
| xiph/rnnoise RNNoise is a C library that uses a hybrid DSP/recurrent neural network approach for real-time full-band speech noise suppression. It also s… | 23 | 5878 | stable |
| rllm-org/rllm rLLM is an open-source Python framework for training language agents with reinforcement learning, supporting any agent harness (Claude Code… | 75 | 5862 | active |
| PaddlePaddle/PaddleClas PaddleClas is a Python library and toolkit for image classification, recognition, and retrieval built on the PaddlePaddle deep learning fra… | 65 | 5847 | active |
| shibing624/MedicalGPT MedicalGPT is a Python training framework for building medical-domain large language models using the full ChatGPT-style training pipeline:… | 85 | 5843 | active |
| areal-project/AReaL AReaL is a large-scale asynchronous reinforcement learning system that bridges foundation model training with agent-based applications, sup… | 86 | 5842 | active |
| pytorch/torchtitan torchtitan is a PyTorch-native platform for large-scale training of generative AI models, offering a clean-room implementation of PyTorch's… | 81 | 5830 | active |
| Eventual-Inc/Daft Daft is a high-performance distributed data engine with a Python dataframe API, implemented in Rust, designed for AI and multimodal workloa… | 95 | 5804 | active |
| DeepLabCut DeepLabCut is an open-source Python toolbox for markerless 2D and 3D pose estimation of user-defined body parts using deep neural networks … | 94 | 5784 | stable |
| NVIDIA/DALI NVIDIA DALI is a GPU-accelerated data loading and preprocessing library with optimized building blocks and an execution engine for deep lea… | 91 | 5761 | active |
| Nuclio Nuclio is a high-performance open-source serverless (FaaS) platform for real-time event and data processing, deployable standalone via Dock… | 92 | 5760 | stable |
| facebookresearch/fastText fastText is a lightweight open-source C++ library (with Python bindings and a CLI) from Facebook Research for efficiently learning word rep… | 10 | 26525 | maintenance |
| Slang Slang is a shading language and compiler for building large, modular, high-performance GPU shader codebases. It compiles to D3D12, Vulkan, … | 95 | 5752 | active |
| pjreddie/darknet Darknet is an open-source neural network framework written in C and CUDA, best known as the original home of the YOLO real-time object dete… | 32 | 26503 | maintenance |
| apple/ml-depth-pro Depth Pro is Apple's reference implementation of a foundation model for zero-shot metric monocular depth estimation, producing sharp high-r… | 66 | 5737 | active |
| Orange Data Mining Orange is an open-source visual programming toolbox for interactive data mining, machine learning, and data visualization. Users build anal… | 74 | 5722 | stable |
| Gen-Verse/OpenClaw-RL OpenClaw-RL is a framework for training personalized AI agents through reinforcement learning using natural conversation as feedback. It us… | 49 | 5722 | active |
| nerfstudio gsplat is an open-source Python library with CUDA-accelerated, differentiable rasterization of Gaussians, based on 3D Gaussian Splatting fo… | 73 | 5716 | active |
| meta-pytorch/captum Captum is a model interpretability and understanding library for PyTorch, providing implementations of algorithms like Integrated Gradients… | 79 | 5712 | active |
| mlpack mlpack is a fast, header-only C++ machine learning library built on Armadillo, ensmallen, and cereal, offering a wide range of algorithms f… | 82 | 5710 | stable |
| OpenBMB/UltraRAG UltraRAG is a low-code Python framework from OpenBMB for building complex RAG pipelines using MCP servers and YAML-based orchestration with… | 85 | 5706 | active |
| maziyarpanahi/openmed OpenMed is a local-first healthcare AI SDK for clinical named-entity recognition and HIPAA PII/PHI de-identification that runs entirely on-… | 85 | 5699 | active |
| huggingface/alignment-handbook A collection of robust training recipes and scripts from Hugging Face for aligning large language models with human and AI preferences, cov… | 72 | 5696 | active |
| alexlenail/NN-SVG A web-based tool for creating publication-ready neural network architecture diagrams parametrically, supporting FCNN, LeNet-style CNN, and … | 57 | 5691 | stable |
| idealo/imagededup imagededup is a Python library for finding exact and near-duplicate images in a collection using perceptual hashing algorithms (PHash, DHas… | 42 | 5678 | stable |
| Basic Pitch Basic Pitch is a lightweight neural network library from Spotify for automatic music transcription, converting audio recordings of nearly a… | 42 | 5645 | stable |
| lm-sys/RouteLLM RouteLLM is a Python framework for serving and evaluating LLM routers that route queries between strong and weak models to cut costs. It pr… | 25 | 5626 | active |
| lyuwenyu/RT-DETR Official implementation of RT-DETR and RT-DETRv2, real-time object detection transformers that outperform YOLO models, in PyTorch and Paddl… | 74 | 5609 | active |
| cgzirim/seek-tune SeekTune is an open-source implementation of Shazam's audio fingerprinting and song recognition algorithm, written in Go with a web client.… | 46 | 5605 | active |
| ZenML ZenML is an open-source MLOps framework for building, deploying, and managing production ML pipelines, LLM workflows, and AI agents on any … | 97 | 5601 | active |
| matterport/Mask_RCNN A Python implementation of the Mask R-CNN model for object detection and instance segmentation, built on Keras and TensorFlow with a ResNet… | 23 | 25574 | maintenance |
| Netflix/vmaf VMAF is Netflix's Emmy-winning perceptual video quality assessment algorithm, provided as a standalone C library (libvmaf) with a wrapping … | 93 | 5516 | stable |
| obss/sahi SAHI (Slicing Aided Hyper Inference) is a Python vision library for detecting small objects in large images via sliced/tiled inference, wor… | 94 | 5513 | active |
| LaurentMazare/tch-rs tch-rs is a Rust crate providing thin bindings to the C++ API of PyTorch (libtorch), staying close to the original API. It enables tensor o… | 64 | 5503 | active |
| mosaicml/composer Composer is an open-source PyTorch-based deep learning training library by MosaicML (now Databricks) for training neural networks faster an… | 66 | 5499 | active |
| HarisIqbal88/PlotNeuralNet A LaTeX/TikZ-based library for drawing neural network architecture diagrams, with a Python interface for generating the TikZ code. It is co… | 23 | 25035 | maintenance |
| tensorflow/rust TensorFlow Rust provides idiomatic Rust language bindings for TensorFlow via its C API. It lets Rust programs build and run TensorFlow comp… | 10 | 5475 | active |
| RLinf/RLinf RLinf is an open-source, flexible and scalable reinforcement learning training infrastructure for embodied AI (vision-language-action model… | 78 | 5471 | active |
| PeterGriffinJin/Search-R1 Search-R1 is an open-source reinforcement learning framework for training LLMs that interleave reasoning with search engine calls, built on… | 40 | 5446 | active |
| apple/coremltools Apple's official Python package for converting machine learning models from TensorFlow, PyTorch, scikit-learn, XGBoost, and LibSVM into the… | 72 | 5436 | active |
| isl-org/MiDaS MiDaS is a Python library with pretrained models for robust monocular depth estimation from a single image, based on the TPAMI 2022 paper a… | 10 | 5419 | stable |
| roboflow/sports A Python library from Roboflow providing reusable computer vision tools for sports analytics, including ball tracking, player tracking and … | 72 | 5390 | active |
| NVIDIA/Model-Optimizer NVIDIA Model Optimizer (ModelOpt) is a Python library of state-of-the-art model optimization techniques including quantization, pruning, di… | 92 | 5386 | active |
| nmslib/hnswlib A header-only C++ library with Python bindings implementing the HNSW algorithm for fast approximate nearest neighbor search. It supports in… | 75 | 5332 | stable |
| superduper-io/superduper Superduper is an open-source Python framework for building database-integrated AI applications and agents, connecting models, embeddings, a… | 44 | 5329 | active |
| NVIDIA/cuml NVIDIA cuML is a GPU-accelerated machine learning library offering scikit-learn-style estimators that run on NVIDIA GPUs via CUDA. It also … | 91 | 5292 | active |
| HIT-SCIR/ltp LTP (Language Technology Platform) is an open-source neural NLP toolkit for Chinese supporting word segmentation, POS tagging, NER, depende… | 51 | 5265 | active |
| microsoft/SynapseML SynapseML (formerly MMLSpark) is an open-source machine learning library built on Apache Spark that provides simple, composable, distribute… | 86 | 5248 | active |