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 |
|---|---|---|---|
| SuperBruceJia/EEG-DL EEG-DL is a deep learning library built on TensorFlow for classifying EEG signals, supporting many architectures including CNNs, RNNs, GCNs… | 43 | 1171 | active |
| Tencent/hpc-ops HPC-Ops is a production-grade C++/CUDA operator library for high-performance LLM inference, developed by Tencent's Hunyuan AI Infra team. I… | 58 | 1170 | active |
| majianjia/nnom NNoM is a high-level neural network inference library written in C for microcontrollers. It converts Keras models into optimized on-device … | 23 | 1170 | stable |
| SarahWeiii/CoACD CoACD is a C++ library (with Python bindings and a Unity package) that performs approximate convex decomposition of 3D triangle meshes usin… | 95 | 1169 | active |
| simpler-env/SimplerEnv SIMPLER (SimplerEnv) is a collection of simulated environments built on SAPIEN/ManiSkill for evaluating real-world robot manipulation polic… | 48 | 1169 | active |
| DepthAnything/PromptDA Prompt Depth Anything is a Python library implementing a CVPR 2025 method for high-resolution (up to 4K) accurate metric depth estimation. … | 47 | 1169 | active |
| warp-ctc A fast parallel implementation of the Connectionist Temporal Classification (CTC) loss function for CPU and CUDA GPU, with a simple C inter… | 32 | 4066 | maintenance |
| facebookresearch/encodec EnCodec is a deep learning based neural audio codec from Meta AI that compresses mono 24 kHz and stereo 48 kHz audio to bitrates from 1.5 t… | 32 | 4065 | maintenance |
| espressif/esp-dl ESP-DL is Espressif's lightweight neural network inference framework for ESP-series chips, with a custom .espdl model format, quantization … | 69 | 1168 | active |
| Dao-AILab/quack QuACK is a collection of high-performance GPU kernels (RMSNorm, LayerNorm, softmax, cross-entropy, GEMM with epilogues) written in NVIDIA's… | 83 | 1167 | active |
| extropic-ai/thrml THRML is a JAX library for building and sampling probabilistic graphical models, focused on efficient block Gibbs sampling of energy-based … | 70 | 1167 | active |
| Object Detection Metrics A Python toolkit implementing the most popular metrics (AP, mAP, precision-recall curves) used to evaluate object detection algorithms, wit… | 46 | 1167 | stable |
| deepmodeling/Uni-Mol Uni-Mol is a collection of 3D molecular representation learning frameworks and pretrained models for tasks like molecule property predictio… | 30 | 1167 | active |
| ucla-mobility/OpenCDA OpenCDA is an open-source Python framework for prototyping and evaluating full-stack cooperative driving automation (CDA) applications in a… | 63 | 1166 | active |
| OpenKE OpenKE is an open-source PyTorch-based toolkit for knowledge graph embedding (knowledge representation learning), with C++ accelerated data… | 32 | 4053 | maintenance |
| yosinski/deep-visualization-toolbox A GUI toolbox for visualizing and understanding deep neural networks, showing per-unit activations, backprop/deconv, and regularized-optimi… | 32 | 4052 | maintenance |
| sksq96/pytorch-summary A PyTorch library providing a Keras-style model.summary() that prints layer types, output shapes, parameter counts, and memory estimates. I… | 32 | 4050 | maintenance |
| Tavish9/any4lerobot Any4LeRobot is a curated collection of Python utilities for the Hugging Face LeRobot robotics ecosystem, including dataset conversion scrip… | 63 | 1165 | active |
| cure-lab/MagicDrive MagicDrive is the official PyTorch implementation of an ICLR 2024 paper for controllable street view generation using diffusion models with… | 32 | 1165 | active |
| facebookresearch/VideoPose3D A PyTorch implementation of CVPR 2019 research on 3D human pose estimation in video using temporal convolutions over 2D keypoint trajectori… | 10 | 4049 | maintenance |
| gemelo-ai/vocos Vocos is a fast neural vocoder that synthesizes audio waveforms from acoustic features such as mel-spectrograms or EnCodec tokens. It uses … | 61 | 1164 | stable |
| Text-to-Audio/AudioLCM AudioLCM is a PyTorch implementation of an ACM-MM'24 paper for efficient, high-quality text-to-audio generation using latent consistency mo… | 34 | 1164 | active |
| IST-DASLab/marlin Marlin is a highly optimized FP16xINT4 matrix multiplication CUDA kernel for LLM inference that achieves near-ideal 4x speedups at batch si… | 27 | 1164 | active |
| chrishayuk/larql LARQL decompiles transformer models into a queryable 'vindex' (vector index) format and provides LQL, a SQL-like query language, to browse,… | 80 | 1162 | active |
| MIT-SPARK/VGGT-SLAM VGGT-SLAM is a dense RGB SLAM system that performs real-time feed-forward 3D scene reconstruction, optimizing on the SL(4) manifold using t… | 56 | 1162 | active |
| flagos-ai/FlagGems FlagGems is a high-performance operator library for large language models written in the Triton language, providing backend-neutral GPU ker… | 88 | 1161 | active |
| unitreerobotics/unifolm-world-model-action UnifoLM-WMA-0 is Unitree's open-source world-model-action framework for general-purpose robot learning across multiple robotic embodiments.… | 47 | 1161 | active |
| Photo2Cartoon A Python deep-learning project from Minivision that converts real portrait photos into cartoon-style avatars using unpaired image translati… | 32 | 4032 | maintenance |
| lhotse-speech/lhotse Lhotse is a Python library for flexible, scalable preparation of multimodal (speech, audio, video, image, text) data for machine learning, … | 85 | 1160 | active |
| fundamentalvision/Deformable-DETR Official PyTorch implementation of Deformable DETR, an efficient end-to-end object detector that uses deformable attention to fix DETR's sl… | 32 | 4028 | maintenance |
| Libr-AI/OpenFactVerification Loki is an open-source Python tool that automates fact verification by decomposing texts into claims, retrieving evidence via search, and u… | 16 | 1159 | active |
| chengtan9907/OpenSTL OpenSTL is a comprehensive benchmark and modular framework for spatio-temporal predictive learning, covering video prediction methods acros… | 50 | 1158 | active |
| Linaom1214/TensorRT-For-YOLO-Series A Python and C++ toolkit for running YOLO-series object detection models (YOLOv3 through YOLOv12, YOLOX) with NVIDIA TensorRT, including ON… | 40 | 1158 | active |
| kijai/ComfyUI-IC-Light A ComfyUI custom node that provides a native implementation of IC-Light models for image relighting. It lets users run IC-Light workflows i… | 32 | 1158 | active |
| sirius-ai/LPRNet_Pytorch A PyTorch implementation of LPRNet, a lightweight deep neural network for license plate recognition. It ships with pretrained weights focus… | 32 | 1158 | stable |
| EyeTrackVR EyeTrackVR is a free, open-source, DIY software platform that turns affordable cameras and IR LEDs mounted inside a VR headset into an eye … | 82 | 1157 | active |
| okooo5km/HiPixel HiPixel is a native macOS app for AI-powered image super-resolution, built with SwiftUI and using Upscayl's AI models. It offers batch upsc… | 70 | 1155 | active |
| JunMa11/SegLossOdyssey A curated collection of loss functions for medical image segmentation, accompanying the 'Loss Odyssey in Medical Image Segmentation' survey… | 32 | 4005 | maintenance |
| ShiqiYu/OpenGait OpenGait is a flexible and extensible Python framework for gait recognition research, providing implementations of state-of-the-art models … | 65 | 1154 | active |
| cvg/glue-factory Glue Factory is a PyTorch-based library for training and evaluating deep neural networks that detect and match local visual features (point… | 66 | 1153 | active |
| OpenGVLab/VisionLLM VisionLLM is a series of open-source multimodal large language models from OpenGVLab that unify vision-centric tasks under language instruc… | 30 | 1153 | active |
| PKU-Alignment/omnisafe OmniSafe is a PyTorch-based infrastructural framework for safe reinforcement learning research, providing a unified modular toolkit and com… | 24 | 1153 | active |
| White-box Cartoonization Official TensorFlow implementation of the CVPR 2020 paper 'Learning to Cartoonize Using White-box Cartoon Representations', which converts … | 57 | 4000 | maintenance |
| quark0/darts DARTS is the official PyTorch implementation of the ICLR 2019 paper 'DARTS: Differentiable Architecture Search', which performs neural arch… | 32 | 3997 | maintenance |
| mgonzs13/yolo_ros A ROS 2 wrapper for Ultralytics YOLO models (YOLOv8 through YOLO26) providing object detection, tracking, instance segmentation, human pose… | 96 | 1152 | active |
| facebookresearch/fairseq2 fairseq2 is a PyTorch-based sequence modeling toolkit from Meta FAIR for training custom models for content generation tasks such as langua… | 86 | 1152 | active |
| open-gigaai/giga-world-1 GigaWorld-1 is an open-source framework providing training, inference, data processing, checkpoint conversion, and LoRA merge workflows for… | 51 | 1152 | active |
| shyamsn97/mario-gpt MarioGPT is a Python library and finetuned GPT2 model that generates playable Super Mario Bros levels from text prompts. It accompanies the… | 22 | 1152 | active |
| JDAI-CV/fast-reid FastReID is a PyTorch-based research platform implementing state-of-the-art re-identification algorithms for persons, vehicles, and faces. … | 23 | 3995 | maintenance |
| apple/ml-clara CLaRa is Apple's open-source end-to-end Retrieval-Augmented Generation model that compresses documents into continuous latent representatio… | 59 | 1151 | active |
| FlagOpen/RoboBrain2.5 RoboBrain 2.5 is an open-source embodied AI foundation model from BAAI that combines multimodal large language model capabilities with 3D s… | 47 | 1150 | active |
| Gen-Verse/LatentMAS LatentMAS is a training-free multi-agent reasoning framework that lets LLM agents collaborate in continuous latent space by passing hidden-… | 52 | 1149 | active |
| brightmart/albert_zh A repository providing pre-trained ALBERT models for Chinese language, implemented in TensorFlow with PyTorch and Keras conversions. It inc… | 32 | 3984 | maintenance |
| google/lyra Lyra is a very low-bitrate speech codec from Google that combines traditional codec techniques with generative machine learning models to c… | 23 | 3982 | maintenance |
| nnaisense/evotorch EvoTorch is an open-source evolutionary computation library built on top of PyTorch, developed at NNAISENSE. It provides distribution-based… | 75 | 1148 | active |
| Being-H Being-H is a family of human-centric embodied foundation models, including VLA models (Being-H0.5, Being-H0) and latent world-action models… | 62 | 1148 | active |
| midas-research/audino Audino is an open-source, self-hosted web application for annotating audio, supporting transcription, labeling, and speaker-related tasks. … | 48 | 1148 | active |
| IDEA-Research/Grounding-DINO-1.5-API Python examples and API client for Grounding DINO 1.5/1.6, IDEA Research's open-world (open-set) object detection model series hosted on De… | 25 | 1148 | active |
| amazon-science/mm-cot Official PyTorch implementation of the paper 'Multimodal Chain-of-Thought Reasoning in Language Models', which adds vision features to a tw… | 32 | 3978 | maintenance |
| huawei-noah/trustworthyAI A collection of trustworthy AI projects from Huawei Noah's Ark Lab, centered on gCastle, a causal structure learning toolchain with many gr… | 66 | 1147 | active |
| Snips NLU Snips NLU is a Python library (with a Rust core) that extracts structured meaning from natural language text by detecting user intents and … | 23 | 3974 | maintenance |
| hanruihua/ir-sim IR-SIM is an open-source, Python-based lightweight robot simulator for navigation, control, and learning. It provides a simple YAML-driven … | 99 | 1146 | active |
| xzf-thu/Mega-ASR Mega-ASR is a foundation automatic speech recognition model trained on 2.6M samples spanning 7 atomic acoustic conditions and 54 compound r… | 56 | 1146 | active |
| yohanshin/WHAM WHAM is the official PyTorch implementation of the CVPR 2024 paper 'Reconstructing World-grounded Humans with Accurate 3D Motion'. It estim… | 27 | 1146 | active |
| HengyiWang/spann3r Spann3R is a transformer-based model for dense 3D reconstruction from ordered or unordered image collections, built on the DUSt3R paradigm.… | 23 | 1145 | active |
| Codium-ai/AlphaCodium Official implementation of the AlphaCodium paper, a test-based, multi-stage, iterative flow for LLM code generation on competitive programm… | 27 | 3966 | maintenance |
| atiilla/GeoIntel GeoIntel is a Python tool that uses Google's Gemini API to estimate where a photo was taken through AI-powered geolocation analysis. It off… | 55 | 1144 | active |
| ScorpioLea/AiCE AiCE is a Python tool that predicts high-fitness protein mutations by sampling sequences from protein inverse folding models such as Protei… | 37 | 1144 | active |
| callous-youth/BOAT BOAT is a PyTorch-based library providing a compositional, operation-level toolbox for gradient-based bi-level optimization (BLO). It decom… | 81 | 1143 | active |
| noahcao/OC_SORT OC-SORT is a pure motion-model-based multi-object tracker for video, improving on SORT by fixing Kalman filter limitations to handle occlus… | 64 | 1143 | stable |
| rohitgandikota/sliders Official implementation of Concept Sliders, LoRA adaptors that enable precise, plug-and-play control of attributes in diffusion models like… | 49 | 1143 | active |
| microsoft/onnxruntime-genai ONNX Runtime GenAI is a C++ library with Python, C#, C/C++, and Java APIs for running generative AI models (LLMs, Whisper, vision-language … | 94 | 1142 | active |
| THUMNLab/AutoGL AutoGL is an autoML framework and toolkit for machine learning on graphs, built on PyTorch with PyTorch Geometric and DGL backends. It prov… | 43 | 1142 | active |
| LLM-Pruner LLM-Pruner is a PyTorch library implementing structural pruning of large language models based on gradient information, as published at Neu… | 30 | 1142 | active |
| facebookresearch/StarSpace StarSpace is a general-purpose neural model from Facebook Research that learns entity embeddings for classification, retrieval, ranking, an… | 10 | 3955 | maintenance |
| vec2text/vec2text A Python library for text embedding inversion: training and running models that reconstruct text sequences from their sentence embeddings. … | 54 | 1140 | active |
| smthemex/ComfyUI_Sonic A ComfyUI custom node implementing the Sonic method for audio-driven portrait animation, generating talking-head videos from a single portr… | 53 | 1140 | active |
| clovaai/deep-text-recognition-benchmark Official PyTorch implementation of a four-stage scene text recognition (OCR) framework from an ICCV 2019 paper, with training and evaluatio… | 32 | 3941 | maintenance |
| Kiteretsu77/APISR APISR is a deep-learning based super-resolution tool that restores and enhances low-quality, low-resolution anime images and videos using t… | 34 | 1138 | active |
| rhymes-ai/Allegro Allegro is an open-source text-to-video generation model that produces high-quality 720p videos up to 6 seconds at 15 FPS from text prompts… | 23 | 1138 | active |
| CellProfiler CellProfiler is a free, open-source desktop application for quantitative analysis of biological images, letting biologists build modular im… | 67 | 1137 | active |
| OpenGVLab/SAM-Med2D Official implementation of SAM-Med2D, a fine-tuned Segment Anything Model (SAM) for 2D medical image segmentation, trained on the SA-Med2D-… | 29 | 1137 | active |
| rockbenben/subtitle-translator A free, browser-based batch subtitle translation tool for .srt, .ass, .vtt, and .lrc files that strips timing locally and sends only dialog… | 85 | 1136 | active |
| HarryR/z80ai Z80-μLM is a 2-bit quantized micro language model designed to run on 8-bit Z80 processors with only 64KB of RAM, exportable as ~40KB CP/M .… | 46 | 1136 | active |
| alibaba-damo-academy/RynnVLA-002 RynnVLA-002 is a unified autoregressive Vision-Language-Action and world model that generates robot actions from text and image observation… | 40 | 1136 | active |
| diego-vicente/som-tsp A Python implementation that solves the Traveling Salesman Problem using Self-Organizing Maps, a neural network technique adapted to find a… | 32 | 3931 | maintenance |
| OpenDriveLab/UniVLA UniVLA is an open-source framework for training cross-embodiment vision-language-action (VLA) robot policies using task-centric latent acti… | 40 | 1134 | active |
| facebookresearch/watermark-anything Official PyTorch implementation and pretrained models for the paper 'Watermark Anything with Localized Messages', which embeds multiple loc… | 10 | 1134 | active |
| Janspiry/Image-Super-Resolution-via-Iterative-Refinement An unofficial PyTorch implementation of SR3 (Image Super-Resolution via Iterative Refinement), a diffusion-based model for image super-reso… | 32 | 3922 | maintenance |
| sooftware/conformer An unofficial PyTorch implementation of the Conformer architecture (convolution-augmented Transformer) from the INTERSPEECH 2020 paper, tar… | 65 | 1132 | active |
| localai-org/depth-anything.cpp A from-scratch C++17/ggml port of ByteDance's Depth Anything 2 and 3 models for dependency-free monocular metric depth and camera pose infe… | 56 | 1132 | active |
| BlackJAX BlackJAX is a library of fast, modular MCMC samplers (NUTS, HMC, and more) built on JAX, running on CPU, GPU, and TPU. It is not a full pro… | 94 | 1131 | active |
| leggedrobotics/elevation_mapping_cupy A GPU-accelerated elevation mapping library for robotics, built on CuPy and integrated with ROS, that fuses point clouds into multi-modal t… | 85 | 1131 | active |
| NovaSearch-Team/RAG-Retrieval A Python library and toolkit for unified fine-tuning, inference, and distillation of RAG retrieval models, including embedding models, ColB… | 57 | 1131 | active |
| OpenGVLab/VideoMamba VideoMamba is a state space model (Mamba-based) architecture for efficient video understanding, released with code and pretrained models fr… | 26 | 1130 | active |
| ikostrikov/pytorch-a2c-ppo-acktr-gail A PyTorch implementation of several deep reinforcement learning algorithms: A2C, PPO, ACKTR, and GAIL (imitation learning). It works with O… | 32 | 3907 | maintenance |
| PriesiaMioShirakana/DragonianVoice A C++ inference library for running ONNX-based TTS, SVC (singing voice conversion), and SVS (singing voice synthesis) models, supporting ar… | 36 | 1129 | active |
| brownhci/WebGazer WebGazer.js is a JavaScript eye tracking library that uses a standard webcam to predict a user's gaze location on a web page in real time. … | 58 | 3903 | maintenance |
| open-mmlab/mmtracking MMTracking is OpenMMLab's PyTorch-based toolbox for video perception tasks, unifying video object detection, multiple object tracking, sing… | 23 | 3903 | maintenance |
| Dobiasd/frugally-deep frugally-deep is a lightweight header-only C++ library for running inference (forward passes) on Keras/TensorFlow models without linking ag… | 78 | 1128 | active |