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 |
|---|---|---|---|
| lucidrains/perceiver-pytorch A PyTorch implementation of the Perceiver architecture (General Perception with Iterative Attention) and its follow-up Perceiver IO. It pro… | 66 | 1221 | active |
| eduardoleao052/js-pytorch JS-PyTorch is a deep learning library for JavaScript that closely mirrors PyTorch's syntax, providing tensor operations, automatic differen… | 17 | 1221 | active |
| gabrielchua/RAGxplorer RAGxplorer is a Python library and hosted Streamlit app for visualizing Retrieval Augmented Generation pipelines. It chunks PDFs, embeds th… | 17 | 1221 | active |
| aTrainTranscription/aTrain aTrain is a desktop GUI application for offline transcription of speech recordings using Whisper-based machine learning models, with speake… | 77 | 1220 | active |
| compdemocracy/polis Polis is an open-source, AI-powered sentiment gathering platform for large-scale open-ended feedback, mapping high-dimensional opinion spac… | 77 | 1220 | active |
| EvolvingLMMs-Lab/LLaVA-OneVision-2 A fully open framework for training multimodal large language models, releasing models, datasets, and training recipes for the LLaVA-OneVis… | 70 | 1220 | active |
| simondlevy/TinyEKF TinyEKF is a lightweight, header-only C/C++ implementation of the Extended Kalman Filter designed for microcontrollers like Arduino and STM… | 63 | 1220 | stable |
| ardha27/AI-Song-Cover-RVC A collection of Google Colab and Kaggle notebooks that form an all-in-one toolkit for creating AI song covers with RVC (Retrieval-based Voi… | 64 | 1219 | active |
| yeyupiaoling/Whisper-Finetune A toolkit for fine-tuning OpenAI's Whisper speech recognition models using LoRA, supporting training with or without timestamps and even wi… | 63 | 1219 | active |
| aydinnyunus/ai-captcha-bypass A Python command-line tool that uses multimodal LLMs (GPT-4o, Gemini) to automatically solve various CAPTCHA types, including text, reCAPTC… | 58 | 1219 | active |
| microsoft/malmo Project Malmo is a platform for artificial intelligence experimentation and research built on top of Minecraft, providing a gym-like enviro… | 10 | 4270 | maintenance |
| fudan-generative-vision/champ Champ is a research framework for controllable and consistent human image animation using 3D parametric guidance (SMPL-based depth, normal,… | 26 | 4265 | maintenance |
| willisma/SiT Official PyTorch implementation of Scalable Interpolant Transformers (SiT), a family of generative models built on Diffusion Transformers t… | 50 | 1217 | active |
| hasktorch/hasktorch Hasktorch is a Haskell library for tensor math and neural networks, built on bindings to the C++ libtorch libraries that power PyTorch. It … | 73 | 1216 | active |
| mlmed/torchxrayvision TorchXRayVision is an open-source PyTorch library providing pre-trained deep learning models and a unified interface for publicly available… | 95 | 1215 | active |
| dendenxu/fast-gaussian-rasterization A drop-in replacement for diff-gaussian-rasterization that renders 3D Gaussian Splatting scenes using a geometry-shader-based GPU pipeline … | 16 | 1215 | active |
| Zefan-Cai/R-KV R-KV is a training-free, redundancy-aware KV cache compression method for reasoning LLMs, discarding repetitive tokens on-the-fly during de… | 64 | 1214 | active |
| ubicomplab/rPPG-Toolbox rPPG-Toolbox is an open-source Python toolbox for camera-based physiological sensing (remote photoplethysmography), enabling heart rate and… | 47 | 1214 | active |
| frotms/PaddleOCR2Pytorch A PyTorch port of PaddleOCR that lets you run PaddleOCR-trained models (detection, recognition, and document structure parsing) without the… | 70 | 1213 | active |
| Picsart-AI-Research/Text2Video-Zero Official implementation of Text2Video-Zero, a zero-shot text-to-video generation method that adapts text-to-image diffusion models like Sta… | 31 | 4244 | maintenance |
| ifzhang/FairMOT FairMOT is a research implementation of a one-shot multi-object tracking model that jointly performs object detection and re-identification… | 32 | 4241 | maintenance |
| NVIDIA/kvpress kvpress is a Python library from NVIDIA that implements multiple KV cache compression methods and benchmarks for long-context LLM inference… | 86 | 1211 | active |
| graphframes/graphframes GraphFrames is a package for Apache Spark that provides DataFrame-based graph processing with distributed graph algorithms like PageRank, c… | 94 | 1210 | stable |
| Tencent-Hunyuan/HunyuanWorld-Mirror HunyuanWorld-Mirror is a feed-forward 3D reconstruction model from Tencent that predicts camera poses, intrinsics, depth maps, point clouds… | 52 | 1210 | active |
| omicverse/omicverse OmicVerse is a Python library for multi-omics analysis covering bulk, single-cell, and spatial RNA-seq workflows. It is part of the scverse… | 93 | 1209 | active |
| haidog-yaqub/MeanFlow An unofficial PyTorch implementation of MeanFlow and iMF, one-step generative modeling methods based on flow matching. It provides config-d… | 56 | 1208 | active |
| DachunKai/EvTexture Official PyTorch implementation of EvTexture and EvTexture++, event-driven video super-resolution models that use event-camera signals to e… | 50 | 1208 | active |
| pymc-labs/CausalPy CausalPy is a Python package for causal inference in quasi-experimental settings, offering methods like difference-in-differences, syntheti… | 91 | 1207 | active |
| cvs-health/uqlm UQLM is a Python library for detecting LLM hallucinations using uncertainty quantification techniques. It provides five categories of score… | 86 | 1207 | active |
| toshas/torch-fidelity A PyTorch library providing accurate and efficient implementations of generative model evaluation metrics such as FID, Inception Score, KID… | 64 | 1206 | active |
| JuliaStats/Distributions.jl A Julia package providing a comprehensive collection of probability distributions and associated functions. It implements moments, entropy,… | 97 | 1205 | stable |
| NVIDIA/earth2studio Earth2Studio is a Python deep-learning framework from NVIDIA for building, exploring, and deploying AI-driven weather and climate workflows… | 88 | 1205 | active |
| lquesada/ComfyUI-Inpaint-CropAndStitch A set of ComfyUI custom nodes that crop an image around a masked area before sampling and stitch the inpainted result back afterward. This … | 70 | 1205 | active |
| googleapis/go-genai The official Google Gen AI SDK for Go, providing a client interface to Google's generative models such as Gemini via the Gemini Developer A… | 85 | 1204 | active |
| metavoiceio/metavoice-src MetaVoice-1B is a 1.2B parameter foundational text-to-speech model trained on 100K hours of speech, focused on emotional rhythm and tone in… | 26 | 4203 | maintenance |
| Artelnics/opennn OpenNN is an open-source C++ library for building, training, and deploying neural networks for advanced analytics. It is dependency-free, o… | 94 | 1202 | active |
| openvinotoolkit/nncf NNCF is Intel's Neural Network Compression Framework, a Python library providing post-training and training-time compression algorithms (qu… | 92 | 1202 | active |
| Cerebras/modelzoo Cerebras Model Zoo is a collection of reference deep learning model implementations (Llama, Mixtral, DINOv2, Llava, etc.) with configs and … | 74 | 1202 | active |
| SimAI SimAI is a large-scale network simulation toolkit from Alibaba Cloud for modeling AI training and inference workloads on GPU clusters, publ… | 65 | 1201 | active |
| xorbitsai/xorbits Xorbits is an open-source distributed computing framework that scales Python data science and machine learning workloads from a laptop to l… | 61 | 1200 | active |
| AlgRUC/JittorGeometric JittorGeometric is a graph machine learning library built on the Jittor deep learning framework, providing implementations of 40+ Graph Neu… | 59 | 1200 | active |
| AirSLAM AirSLAM is an efficient, illumination-robust point-line visual SLAM system supporting stereo visual odometry/VIO, offline map optimization,… | 43 | 1200 | active |
| JackHopkins/factorio-learning-environment An open-source framework for developing and evaluating LLM agents in the game of Factorio, providing an open-ended, non-saturating benchmar… | 83 | 1199 | active |
| Anserini Pyserini is a Python toolkit for reproducible information retrieval research supporting both sparse (via Anserini/Lucene) and dense (via Fa… | 77 | 1198 | active |
| Calamari Calamari is a Python-based OCR engine for line-based automatic text recognition, built on OCRopy and Kraken with a TensorFlow deep-learning… | 59 | 1198 | active |
| deepseek-ai/DeepSeek-VL DeepSeek-VL is an open-source vision-language foundation model for real-world multimodal understanding, released with model weights and inf… | 26 | 4183 | maintenance |
| a-r-j/graphein Graphein is a Python library for constructing graph and mesh representations of proteins, RNA, molecules, and biological interaction networ… | 74 | 1197 | active |
| trianglesplatting/triangle-splatting Official implementation of 'Triangle Splatting for Real-Time Radiance Field Rendering' (3DV 2026), which uses 3D triangles as rendering pri… | 40 | 1197 | active |
| autonomousvision/stylegan-t Official training code for StyleGAN-T, an ICML 2023 paper on fast large-scale text-to-image synthesis using GANs. It provides dataset prepa… | 32 | 1197 | active |
| zai-org/CogAgent CogAgent is an open-source vision-language model (VLM) based GUI agent that understands screen captures and natural language to automate in… | 30 | 1197 | active |
| ddlBoJack/emotion2vec Official PyTorch implementation of emotion2vec, a self-supervised pre-trained model for speech emotion representation. It provides code for… | 27 | 1197 | active |
| DeepRec-AI/DeepRec DeepRec is a high-performance deep learning framework for recommendation models, built on TensorFlow 1.15 with Intel and NVIDIA TensorFlow … | 23 | 1197 | active |
| shallowdream204/DreamClear DreamClear is a diffusion-transformer based real-world image restoration model for high-fidelity super-resolution, published at NeurIPS 202… | 24 | 1196 | active |
| deepfence/FlowMeter FlowMeter is a Go utility that analyzes network packet headers, groups packets into flows, and uses machine learning to classify flows as b… | 10 | 1196 | active |
| NVIDIA/audio-flamingo NVIDIA's PyTorch implementation of the Audio Flamingo series of large audio-language models (AF1, AF2, AF3, and Music Flamingo) for audio u… | 47 | 1195 | active |
| facebookresearch/esm Meta FAIR's Evolutionary Scale Modeling (ESM) library providing Transformer protein language models with pretrained weights, including ESM-… | 10 | 4170 | maintenance |
| Soul-AILab/SoulX-LiveAct SoulX-LiveAct is the official inference code for a real-time human animation framework that generates lifelike, audio/multimodal-controlled… | 51 | 1193 | active |
| lmstudio-ai/mlx-engine mlx-engine is the Apple MLX-based LLM inference engine that powers LM Studio on Mac, built on mlx-lm with support for vision models via mlx… | 68 | 1192 | active |
| getkeops/keops KeOps (pykeops) is a Python library for computing kernel reductions over large arrays on CPUs and GPUs using efficient C++/CUDA routines wi… | 61 | 1192 | active |
| GAIR-NLP/ASI-Arch A multi-agent framework that lets an LLM autonomously conduct end-to-end research on neural network architecture discovery, specifically li… | 40 | 1192 | active |
| chensjtu/GaussianObject GaussianObject is a research framework for high-quality 3D object reconstruction from as few as four input images using Gaussian splatting,… | 26 | 1192 | active |
| warmshao/FasterLivePortrait A real-time portrait animation application based on LivePortrait that animates still photos or videos using a driving video, image, audio, … | 35 | 1191 | active |
| zai-org/VisualGLM-6B VisualGLM-6B is an open-source multimodal conversational language model supporting images, Chinese, and English, built on ChatGLM-6B with a… | 30 | 4154 | maintenance |
| MoonshotAI/MoonEP MoonEP is an Expert Parallelism communication library for Mixture-of-Experts training that keeps token loads perfectly balanced across rank… | 58 | 1189 | active |
| functime-org/functime functime is a Python library for production-ready global forecasting and time-series feature extraction on large panel datasets, built on l… | 65 | 1188 | active |
| chakki-works/seqeval seqeval is a Python library for evaluating sequence labeling tasks such as named-entity recognition, part-of-speech tagging, and semantic r… | 23 | 1188 | stable |
| YOPO YOPO is a learning-based one-stage planner for quadrotor autonomous navigation in obstacle-dense environments, integrating perception, mapp… | 76 | 1187 | active |
| pqpo/SmartCropper An Android library for smart image cropping that automatically detects document borders using OpenCV (with an optional TensorFlow Lite HED … | 62 | 4136 | maintenance |
| CarlGao4/Demucs-Gui A graphical desktop application wrapping the Demucs AI music source-separation model, letting users split songs into stems (vocals, drums, … | 43 | 1186 | active |
| DAMO-NLP-SG/VideoLLaMA3 VideoLLaMA 3 is a frontier multimodal foundation model for image and video understanding, released with checkpoints, inference code, and de… | 34 | 1186 | active |
| SHI-Labs/Neighborhood-Attention-Transformer Official PyTorch implementation of the Neighborhood Attention Transformer (NAT/DiNAT), a family of efficient vision transformers with local… | 32 | 1186 | stable |
| bodaay/HuggingFaceModelDownloader A Go-based CLI utility for downloading models and datasets from the HuggingFace Hub with parallel, resumable downloads. It includes an inte… | 95 | 1185 | active |
| auto-novel/auto-novel AutoNovel is a website application that automatically machine-translates light novels (web novels, bunko novels, and local files) using LLM… | 76 | 1185 | active |
| MIT-SPARK/Hydra Hydra is a C++ system that incrementally builds hierarchical 3D Scene Graphs from sensor data in real time. It is developed by MIT SPARK as… | 68 | 1184 | active |
| chengxuxin/extreme-parkour Official code for 'Extreme Parkour with Legged Robots' (ICRA 2024), a reinforcement learning framework for training quadruped robots to per… | 28 | 1184 | active |
| XPixelGroup/DiffBIR DiffBIR is a blind image restoration framework that uses generative diffusion priors to restore degraded real-world images. It provides pre… | 30 | 4125 | maintenance |
| csguoh/MambaIR MambaIR and MambaIRv2 are PyTorch-based image restoration models built on Mamba state-space models, published at ECCV 2024 and CVPR 2025. T… | 51 | 1182 | active |
| NVlabs/Deep_Object_Pose NVIDIA's Deep Object Pose Estimation (DOPE), a deep learning system for detecting known objects and estimating their 6-DoF pose from RGB ca… | 44 | 1182 | active |
| mlfoundations/open_flamingo OpenFlamingo is an open-source PyTorch implementation of DeepMind's Flamingo, a large multimodal vision-language model that interleaves ima… | 23 | 4118 | maintenance |
| orpatashnik/StyleCLIP Official implementation of StyleCLIP, a method for text-driven manipulation of StyleGAN-generated imagery using CLIP. It provides three app… | 32 | 4117 | maintenance |
| Tencent-Hunyuan/MixGRPO MixGRPO is a research framework from Tencent Hunyuan implementing a mixed ODE-SDE GRPO algorithm for efficient reinforcement learning fine-… | 56 | 1181 | active |
| chongzhou96/EdgeSAM EdgeSAM is the official PyTorch implementation of a distilled, accelerated variant of the Segment Anything Model (SAM) designed for on-devi… | 34 | 1180 | active |
| hijohnnylin/neuronpedia Neuronpedia is an open source AI interpretability platform for exploring, visualizing, and steering the internals of language models. It ho… | 93 | 1178 | active |
| balancap/SSD-Tensorflow A TensorFlow re-implementation of the Single Shot MultiBox Detector (SSD) for object detection, including VGG-based SSD-300 and SSD-512 net… | 32 | 4101 | maintenance |
| tjiiv-cprg/EPro-PnP EPro-PnP is a probabilistic Perspective-n-Points (PnP) layer for end-to-end 6DoF monocular object pose estimation networks, built on PyTorc… | 37 | 1177 | stable |
| magicleap/SuperGluePretrainedNetwork SuperGlue is a PyTorch implementation of a graph neural network with an optimal matching layer that matches sparse image features between t… | 32 | 4099 | maintenance |
| jenly1314/MLKit MLKit is an easy-to-use Kotlin wrapper library around Google ML Kit for Android, exposing text recognition, barcode scanning, image labelin… | 72 | 1176 | active |
| bytedance/1d-tokenizer A research repository from ByteDance containing code and pretrained model weights for 1D visual tokenizers (TiTok, TA-TiTok, FlowTok) and i… | 26 | 1176 | active |
| MIC-DKFZ/batchgenerators A Python framework for data augmentation of 2D and 3D images, developed by the German Cancer Research Center for medical image classificati… | 77 | 1175 | stable |
| iver56/torch-audiomentations A PyTorch library for fast audio data augmentation, inspired by audiomentations. It provides GPU-accelerated, differentiable audio transfor… | 54 | 1174 | active |
| princeton-nlp/MeZO MeZO is a memory-efficient zeroth-order optimizer that fine-tunes language models using only forward passes, with the same memory footprint… | 30 | 1174 | stable |
| baichuan-inc/Baichuan2 Baichuan 2 is a family of open large language models (7B and 13B, Base and Chat variants with 4-bit quantized versions) trained by Baichuan… | 29 | 4082 | maintenance |
| NVlabs/imaginaire NVIDIA's PyTorch library containing optimized implementations of image and video synthesis methods, including GAN-based image-to-image tran… | 32 | 4079 | maintenance |
| onnx/onnxmltools ONNXMLTools is a Python library that converts machine learning models from various toolkits (Keras/TensorFlow, scikit-learn, Core ML, XGBoo… | 77 | 1172 | active |
| nv-tlabs/LLaMA-Mesh LLaMA-Mesh is a fine-tuned large language model from NVIDIA Research that generates and understands 3D meshes by representing vertex coordi… | 24 | 1172 | active |
| LibCity LibCity is an open-source PyTorch library for urban spatial-temporal data mining, providing a unified pipeline for traffic prediction resea… | 23 | 1172 | active |
| Mini Pupper Mini Pupper is an open-source ROS-based quadruped robot dog kit built around Raspberry Pi, with software for SLAM, navigation, and OpenCV-b… | 77 | 1171 | active |
| geopavlakos/hamer HaMeR (Hand Mesh Recovery) is a transformer-based model that reconstructs 3D hand meshes from single monocular images using the MANO parame… | 53 | 1171 | active |
| 8080labs/ppscore ppscore is a Python library implementing the Predictive Power Score (PPS), an asymmetric, data-type-agnostic metric that detects linear and… | 51 | 1171 | stable |
| GaParmar/clean-fid Clean-FID is a PyTorch library for computing the Frechet Inception Distance (FID) with correct image resizing and quantization steps, fixin… | 44 | 1171 | stable |