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 |
|---|---|---|---|
| High-Logic/Genie-TTS GENIE is a lightweight Python inference engine for the open-source GPT-SoVITS text-to-speech project, optimized for fast CPU-based speech s… | 58 | 1806 | active |
| thu-ml/RoboticsDiffusionTransformer RDT-1B is a 1B-parameter diffusion foundation model for robot bimanual manipulation, pre-trained on 1M+ multi-robot episodes to predict rob… | 48 | 1806 | active |
| Vitis AI AMD Vitis AI is an integrated development environment and stack for accelerating AI inference on AMD/Xilinx adaptable platforms, including … | 64 | 1805 | active |
| HuangJunJie2017/BEVDet BEVDet is a Python research codebase implementing the BEVDet series of bird's-eye-view (BEV) 3D object detection models for autonomous driv… | 23 | 1805 | active |
| Vchitect/VBench VBench is a comprehensive benchmark suite for evaluating video generative models across hierarchical evaluation dimensions, with prompt sui… | 58 | 1802 | active |
| zai-org/CogVLM CogVLM is an open-source visual language model (17B) combining a vision encoder with a pretrained language model for image understanding an… | 28 | 6745 | maintenance |
| chungmin99/pyroki PyRoki is a modular Python toolkit for robot kinematic optimization built on JAX, providing differentiable forward kinematics from URDFs, c… | 50 | 1797 | active |
| chynl/snake A Python application that plays the game of Snake using two AI approaches: a graph-search algorithm based on Hamiltonian paths and shortest… | 64 | 1796 | active |
| MTG/sms-tools sms-tools is a Python library of sound analysis and synthesis models for music applications, including DFT, STFT, sinusoidal, harmonic, and… | 61 | 1795 | active |
| FL33TW00D/whisper-turbo Whisper Turbo is a fast, cross-platform, GPU-accelerated implementation of OpenAI's Whisper speech recognition model, built on the Ratchet … | 19 | 1794 | active |
| QwenLM/Qwen-VL Official repository for Qwen-VL, Alibaba Cloud's large vision-language model family, including the pretrained Qwen-VL and instruction-tuned… | 29 | 6725 | maintenance |
| Webreaper/Damselfly Damselfly is a server-based photograph management application designed to index and search very large image collections using metadata such… | 84 | 1790 | active |
| lancopku/pkuseg-python pkuseg is a Python toolkit for multi-domain Chinese word segmentation based on the paper by Luo et al. (2019). It provides pre-trained mode… | 23 | 6710 | maintenance |
| lehaifeng/T-GCN A collection of research source code implementing Temporal Graph Convolutional Networks (T-GCN) and related variants for urban traffic flow… | 46 | 1787 | active |
| octo-models/octo Octo is an open-source generalist robot policy: a transformer-based diffusion policy pretrained on 800k robot trajectories from the Open X-… | 18 | 1787 | active |
| nsfw-filter/nsfw-filter A free, open-source, privacy-focused browser extension that blocks NSFW images using on-device AI classification with TensorFlow.js. It hid… | 91 | 1786 | active |
| DataArcTech/DataArc-SynData-Toolkit DataArc SynData Toolkit is a Python-based synthetic data generation platform for creating customized LLM training data from local corpora, … | 59 | 1783 | active |
| VAST-AI-Research/UniRig UniRig is the official implementation of a SIGGRAPH 2025 framework that automatically rigs 3D models by predicting skeletons and skinning w… | 55 | 1783 | active |
| Simple-Efficient/RL-Factory RLFactory is a reinforcement learning post-training framework for training agentic LLM models, decoupling the environment from RL training … | 31 | 1783 | active |
| CalculatedContent/WeightWatcher WeightWatcher is an open-source Python diagnostic tool for analyzing pre-trained deep neural networks without access to training or test da… | 56 | 1777 | active |
| Totoro97/NeuS Official PyTorch implementation of NeuS, a neural implicit surface reconstruction method that learns SDF-based surfaces via volume renderin… | 32 | 1775 | stable |
| isnowfy/snownlp SnowNLP is a Python library for processing Simplified Chinese text, offering word segmentation, part-of-speech tagging, sentiment analysis,… | 32 | 6635 | maintenance |
| Farama-Foundation/Minari Minari is a Python library providing a standard format for offline reinforcement learning datasets, with popular reference datasets and uti… | 86 | 1772 | active |
| huchenlei/ComfyUI-layerdiffuse A ComfyUI custom node plugin implementing LayerDiffuse, enabling generation of transparent images with RGBA output and foreground/backgroun… | 26 | 1771 | active |
| brainflow-dev/brainflow BrainFlow is a C/C++ core library with bindings for nine languages that provides a uniform SDK for acquiring, parsing, and analyzing data f… | 94 | 1769 | active |
| pondorasti/emojis A web application that generates custom Slack emojis from text prompts using AI image generation (SDXL emoji model) via Replicate. It is a … | 71 | 1769 | active |
| levihsu/OOTDiffusion Official implementation of OOTDiffusion, a latent diffusion model for controllable virtual try-on that generates images of a person wearing… | 26 | 6607 | maintenance |
| GAP-LAB-CUHK-SZ/gaustudio GauStudio is a modular PyTorch framework for 3D Gaussian Splatting (3DGS) research and development, supporting novel view synthesis, 3D rec… | 46 | 1763 | active |
| character-ai/Ovi Ovi is a video-plus-audio generation model from Character AI that simultaneously generates synchronized video and audio from text or text+i… | 38 | 1763 | active |
| ILGPU ILGPU is a just-in-time compiler for high-performance GPU programs (kernels) written in .NET languages like C# and F#, entirely in C# with … | 71 | 1762 | active |
| tsurumeso/vocal-remover A Python command-line tool that uses deep neural networks to separate vocals from instrumental tracks in songs. It includes pretrained mode… | 65 | 1761 | stable |
| parthsarthi03/raptor The official Python implementation of RAPTOR (Recursive Abstractive Processing for Tree-Organized Retrieval), a retrieval-augmented generat… | 26 | 1761 | active |
| robocasa/robocasa RoboCasa is a large-scale simulation framework for training and benchmarking generally capable robots on everyday tasks, focused on kitchen… | 78 | 1759 | active |
| AnswerDotAI/ModernBERT ModernBERT is the research repository for a modernized BERT-family bidirectional encoder trained on 2 trillion tokens with an 8192-token co… | 52 | 1758 | active |
| xinntao/ESRGAN ESRGAN (Enhanced SRGAN) is a PyTorch-based image super-resolution model that won the PIRM 2018 Challenge on Perceptual Super-Resolution. Th… | 32 | 6571 | maintenance |
| kijai/ComfyUI-Florence2 A ComfyUI custom node plugin that runs Microsoft's Florence-2 vision-language model for image captioning, object detection, segmentation, a… | 57 | 1756 | active |
| Maximal Update Parametrization (μP) The `mup` Python package implements Maximal Update Parametrization (μP) for PyTorch models, enabling optimal hyperparameters to remain stab… | 10 | 1756 | stable |
| T5 The official T5 library from Google Research, implementing the text-to-text transfer transformer for NLP tasks like summarization, question… | 67 | 6562 | maintenance |
| SamsungSAILMontreal/TinyRecursiveModels Official codebase for the Tiny Recursive Model (TRM) paper, a recursive reasoning approach where a tiny 7M-parameter neural network iterati… | 10 | 6561 | maintenance |
| mahmoodlab/CLAM CLAM is an open-source Python toolkit for data-efficient, weakly supervised classification of whole-slide images (WSIs) in computational pa… | 36 | 1755 | active |
| MaliosDark/wifi-3d-fusion WiFi-3D-Fusion is an open-source research project that estimates 3D human pose from WiFi CSI (Channel State Information) signals using deep… | 22 | 1755 | active |
| datastax/jvector JVector is an embedded approximate nearest neighbor (ANN) search engine for the JVM, implementing a graph index that merges HNSW's hierarch… | 90 | 1754 | active |
| webonnx/wonnx Wonnx is a GPU-accelerated ONNX inference runtime written entirely in Rust, built on wgpu and usable natively or in the browser via WebGPU … | 10 | 1754 | active |
| decisionintelligence/TFB TFB is a comprehensive and fair benchmarking framework for time series forecasting methods, covering deep learning, machine learning, and s… | 70 | 1753 | active |
| NimaNzrii/comfyui-photoshop A Photoshop plugin that integrates ComfyUI's AI image generation directly into the Photoshop workspace. It lets users run Stable Diffusion … | 68 | 1753 | active |
| maxbbraun/trump2cash A Python stock trading bot that watches Donald Trump's tweets, detects mentions of publicly traded companies, and uses sentiment analysis t… | 10 | 6545 | maintenance |
| software-mansion/react-native-executorch React Native ExecuTorch is a declarative React Native library for running AI models on-device, powered by Meta's ExecuTorch runtime. It shi… | 88 | 1750 | active |
| Zaneham/Booth Booth is an open-source compiler that takes CUDA C, HIP, or Triton kernel source and emits binaries for AMD RDNA 2/3/4 GPUs, NVIDIA PTX, Te… | 79 | 1750 | active |
| CompVis/taming-transformers The official implementation of 'Taming Transformers for High-Resolution Image Synthesis' (CVPR 2021), combining a convolutional VQGAN codeb… | 32 | 6530 | maintenance |
| codertimo/BERT-pytorch A PyTorch implementation of Google AI's 2018 BERT model with simple, readable code. It provides CLI tools for building vocabulary and pre-t… | 23 | 6529 | maintenance |
| Informer The official PyTorch implementation of Informer, an efficient Transformer architecture for long sequence time-series forecasting that won t… | 41 | 6528 | maintenance |
| Stability-AI/StableCascade Official codebase for Stable Cascade, a text-to-image generation model built on the Würstchen architecture with a highly compressed latent … | 26 | 6528 | maintenance |
| kyleskom/NBA-Machine-Learning-Sports-Betting A Python application that predicts NBA game winners and totals (over/under) using XGBoost and neural network models trained on historical t… | 78 | 1747 | active |
| rusty1s/pytorch_scatter A PyTorch extension library providing highly optimized scatter and segment (sparse update) operations with sum, mean, min, and max reductio… | 57 | 1747 | stable |
| TencentARC/BrushNet BrushNet is the official PyTorch implementation of an ECCV 2024 plug-and-play image inpainting model that embeds pixel-level masked image f… | 26 | 1744 | active |
| clmtrackr clmtrackr is a JavaScript library for fitting facial models to faces in videos or images using Constrained Local Models with regularized la… | 23 | 6500 | maintenance |
| facebookresearch/Kats Kats is a Python toolkit from Facebook's Infrastructure Data Science team for time series analysis, covering statistics, anomaly and change… | 66 | 6497 | maintenance |
| facebookresearch/multimodal TorchMultimodal is a PyTorch library from Meta for training state-of-the-art multimodal multi-task models at scale, covering both content u… | 77 | 1740 | active |
| SHI-Labs/OneFormer OneFormer is a universal image segmentation framework (CVPR 2023) that unifies semantic, instance, and panoptic segmentation in a single tr… | 32 | 1740 | stable |
| verlab/accelerated_features XFeat is a lightweight, fast learned keypoint detector and descriptor for local feature extraction and image matching, supporting both spar… | 16 | 1740 | active |
| s60sc/ESP32-CAM_MJPEG2SD An ESP32/ESP32-S3 camera application that detects motion and records JPEG frames to SD card as AVI files, optionally with WAV audio from a … | 93 | 1739 | active |
| alkaline-ml/pmdarima pmdarima is a Python statistical library that brings R's auto.arima functionality to Python, wrapping statsmodels ARIMA/SARIMAX models behi… | 81 | 1738 | active |
| MultimediaTechLab/YOLO Official MIT-licensed implementation of the YOLOv9, YOLOv7, and YOLO-RD real-time object detection models, including pre-trained weights, t… | 51 | 1738 | active |
| Xiaojiu-z/EasyControl EasyControl is the official implementation of an ICCV 2025 paper adding efficient and flexible conditional control to Diffusion Transformer… | 32 | 1738 | active |
| openai/consistency_models Official PyTorch implementation of Consistency Models, a generative image model family from OpenAI supporting consistency distillation, con… | 10 | 6483 | maintenance |
| google/automl Google Brain's AutoML repository containing implementations of AutoML models and libraries such as EfficientNet, EfficientNetV2, and Effici… | 10 | 6474 | maintenance |
| Pupil Pupil is an open source eye tracking platform consisting of applications (Pupil Capture, Player, Service) that work with Pupil Labs wearabl… | 64 | 1735 | active |
| robin-shaun/XTDrone XTDrone is a customizable UAV simulation platform built on PX4, ROS, and Gazebo, supporting multi-rotors, fixed-wing, VTOL vehicles, and ot… | 77 | 1733 | active |
| muscriptor/muscriptor MuScriptor is a multi-instrument music transcription model by Kyutai and Mirelo that converts audio recordings into MIDI and sheet music. I… | 78 | 1732 | active |
| cleverhans-lab/cleverhans CleverHans is a Python library for benchmarking machine learning systems' vulnerability to adversarial examples, providing reference implem… | 23 | 6455 | maintenance |
| jakobrunge/tigramite Tigramite is a Python package for causal inference on time series data, providing causal discovery methods such as PCMCI, PCMCIplus, LPCMCI… | 47 | 1731 | active |
| NVIDIA/FasterTransformer NVIDIA's highly optimized C++/CUDA library for fast inference of Transformer-based models such as BERT, GPT, and encoder-decoder models, wi… | 23 | 6450 | maintenance |
| google-research/circuit_training AlphaChip (circuit_training) is Google's open-source framework for generating chip floorplans using distributed deep reinforcement learning… | 58 | 1730 | active |
| McGill-NLP/llm2vec LLM2Vec is a Python library that converts decoder-only large language models into powerful text encoders via bidirectional attention, maske… | 46 | 1729 | active |
| birdnet-team/BirdNET-Analyzer BirdNET-Analyzer is an open-source Python tool from the Cornell Lab of Ornithology that uses deep learning to identify bird species from au… | 77 | 1728 | active |
| AvaLovelace1/BrickGPT BrickGPT is the official implementation of an ICCV 2025 Best Paper approach that generates physically stable, buildable toy brick (LEGO) mo… | 54 | 1726 | active |
| meta-pytorch/tnt TNT (torchtnt) is a lightweight library from Meta providing tools and utilities for PyTorch training workflows. It offers building blocks f… | 77 | 1723 | active |
| facebookresearch/coconut Official PyTorch implementation of Coconut, a method for training large language models to reason in a continuous latent space instead of e… | 58 | 1723 | active |
| facebookresearch/ConvNeXt Official PyTorch implementation of ConvNeXt, a pure convolutional neural network architecture from the CVPR 2022 paper 'A ConvNet for the 2… | 10 | 6408 | maintenance |
| lotus-data/lotus LOTUS is a Python library providing a Pandas-like API of LLM-powered semantic operators (map, filter, extract, aggregate, top-k) for bulk p… | 85 | 1719 | active |
| tenstorrent/tt-metal TT-Metal is Tenstorrent's open-source software stack containing TT-NN, a Python and C++ neural network operator library, and TT-Metalium, a… | 99 | 1718 | active |
| Enzyme Enzyme is a high-performance automatic differentiation plugin for LLVM and MLIR that computes derivatives and gradients of arbitrary existi… | 95 | 1718 | active |
| One-2-3-45/One-2-3-45 One-2-3-45 is the official PyTorch implementation of a NeurIPS 2023 paper that converts any single image into a full 360-degree 3D textured… | 29 | 1717 | stable |
| lightly-ai/lightly-train LightlyTrain is a Python framework for training computer vision models, covering pretraining of vision foundation models (DINOv2/v3) on unl… | 83 | 1716 | active |
| sihyun-yu/REPA REPA is the official PyTorch implementation of the ICLR 2025 paper 'Representation Alignment for Generation', a regularization technique th… | 24 | 1716 | active |
| cocodataset/cocoapi Official API for the COCO (Common Objects in Context) dataset, providing Matlab, Python, and Lua interfaces to load, parse, and visualize C… | 32 | 6387 | maintenance |
| openappsec/openappsec open-appsec is an open-source machine learning security engine that provides preemptive web application and API threat protection against O… | 90 | 1715 | active |
| 4paradigm/OpenMLDB OpenMLDB is an open-source machine learning database that acts as a feature platform, computing consistent features for both offline traini… | 65 | 1714 | active |
| trailofbits/buttercup Buttercup is a Cyber Reasoning System (CRS) developed by Trail of Bits for the DARPA AI Cyber Challenge that automatically finds and patche… | 56 | 1714 | active |
| lsdefine/simple_GRPO A minimal (~200 lines) Python implementation of GRPO reinforcement learning for training LLMs to develop r1-like reasoning, built on PyTorc… | 41 | 1714 | active |
| feizc/FluxMusic FluxMusic is the official PyTorch implementation of a rectified flow Transformer model for text-to-music generation, from the paper 'Flux t… | 24 | 1709 | active |
| iMoonLab/yolov13 Official PyTorch implementation of YOLOv13, a real-time object detection model family (Nano to X-Large) featuring Hypergraph-based Adaptive… | 29 | 1708 | active |
| jiaweizzhao/GaLore GaLore is a PyTorch library implementing Gradient Low-Rank Projection for memory-efficient full-parameter training of large language models… | 26 | 1708 | active |
| NuPIC NuPIC is Numenta's open-source machine intelligence platform implementing Hierarchical Temporal Memory (HTM), a computational theory of the… | 23 | 6352 | maintenance |
| micknoise/Maximilian Maximilian is a cross-platform C++ audio synthesis and signal processing (DSP) library with JavaScript bindings for Web Audio. It provides … | 23 | 1707 | active |
| kubeflow/katib Kubeflow Katib is a Kubernetes-native AutoML system for hyperparameter tuning, early stopping, and neural architecture search. It is ML-fra… | 70 | 1706 | stable |
| omerbt/TokenFlow TokenFlow is the official PyTorch implementation of an ICLR 2024 paper for text-driven, temporally consistent video editing using a pre-tra… | 29 | 1706 | stable |
| zai-org/ImageReward ImageReward is the first general-purpose human preference reward model for text-to-image generation, trained on 137k expert comparison pair… | 49 | 1705 | stable |
| airockchip/rknn-llm RKLLM is Rockchip's software stack for converting, quantizing, and running large language models on Rockchip NPU chips such as RK3588 and R… | 83 | 1704 | active |
| BindsNET/bindsnet BindsNET is a Python package for simulating spiking neural networks (SNNs) built on PyTorch tensor functionality, running on CPUs or GPUs. … | 81 | 1704 | active |