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 |
|---|---|---|---|
| open-mmlab/mmaction2 MMAction2 is OpenMMLab's PyTorch-based toolbox and benchmark for video understanding, covering action recognition, temporal action localiza… | 52 | 5163 | active |
| facebookresearch/co-tracker CoTracker is a transformer-based model from Meta AI and Oxford VGG that jointly tracks any point (pixel) across a video, handling occlusion… | 57 | 5149 | active |
| ArthurBrussee/brush Brush is a 3D reconstruction engine using Gaussian splatting, built in Rust on the Burn ML framework and WebGPU. It trains and renders spla… | 64 | 5148 | active |
| EasyPhoto EasyPhoto is a Stable Diffusion WebUI plugin for generating AI portraits by training a personal 'digital doppelganger' from 5-20 user photo… | 29 | 5147 | active |
| philz1337x/clarity-upscaler Clarity AI is a free and open-source AI image upscaler and enhancer built on Stable Diffusion, positioned as an alternative to Magnific. It… | 27 | 5138 | active |
| ai-dawang/PlugNPlay-Modules A curated collection of plug-and-play deep learning modules (convolutions, attention mechanisms, downsampling, and feature fusion blocks) i… | 35 | 5120 | active |
| chatopera/Synonyms A Python library for Chinese synonym lookup and semantic similarity computation, built on word vector models for NLP and chatbot use cases.… | 48 | 5107 | active |
| pytorch/executorch ExecuTorch is PyTorch's framework for exporting and running AI models on-device across mobile, embedded, and edge hardware, with a tiny (~5… | 94 | 5101 | active |
| facebookresearch/AugLy AugLy is a Python data augmentation library from Meta AI supporting audio, image, text, and video with over 100 augmentations. It focuses o… | 66 | 5097 | stable |
| manycore-research/SpatialLM SpatialLM is a 3D large language model that processes point cloud data (from monocular video, RGBD images, or LiDAR) and generates structur… | 59 | 5094 | active |
| AILab-CVC/VideoCrafter VideoCrafter is an open-source video generation and editing toolbox built on video diffusion models, offering Text-to-Video and Image-to-Vi… | 45 | 5091 | active |
| quantumlib/Cirq Cirq is a Python framework from Google Quantum AI for writing, manipulating, optimizing, and simulating quantum circuits, with a focus on N… | 86 | 5086 | stable |
| huggingface/text-embeddings-inference Text Embeddings Inference (TEI) is a Rust-based toolkit from Hugging Face for deploying and serving open-source text embedding, re-ranker, … | 93 | 5079 | active |
| Deci-AI/super-gradients SuperGradients is an open-source PyTorch-based training library for building, training, and fine-tuning state-of-the-art computer vision mo… | 50 | 5072 | active |
| fspecii/ace-step-ui A professional React/TypeScript web UI for the open-source ACE-Step 1.5 AI music generation model, providing a Spotify-like interface for c… | 52 | 5045 | active |
| deepseek-ai/DeepSeek-V2 DeepSeek-V2 is a strong, economical Mixture-of-Experts language model released with open weights, inference code, and evaluation tooling. T… | 25 | 5042 | active |
| microsoft/SPTAG SPTAG is a C++ library from Microsoft for large-scale approximate nearest neighbor (ANN) vector search, offering kd-tree/balanced k-means t… | 75 | 5019 | active |
| Plachtaa/VITS-fast-fine-tuning A Python pipeline for fast fine-tuning of VITS text-to-speech models, enabling speaker adaptation in under an hour from short audio, long a… | 10 | 5009 | active |
| sktime/pytorch-forecasting A PyTorch-based Python library for time series forecasting with state-of-the-art deep learning architectures. It provides a high-level API … | 88 | 4999 | active |
| going-doer/Paper2Code PaperCoder is a multi-agent LLM system that automatically generates a code repository from a machine learning research paper via a planning… | 49 | 4982 | active |
| EvoScientist EvoScientist is a Python-based multi-agent framework for building self-evolving AI scientists that plan, research, code, analyze data, and … | 83 | 4980 | active |
| shibing624/text2vec A Python library that converts text (words, sentences, paragraphs) into vector embeddings, implementing Word2Vec, RankBM25, BERT, Sentence-… | 49 | 4978 | active |
| apple/embedding-atlas Embedding Atlas is an open-source tool from Apple for interactive visualization of large embeddings and their metadata, supporting clusteri… | 86 | 4969 | active |
| microsoft/muzic Muzic is a Microsoft Research project providing deep learning models for music understanding and generation, including MusicBERT, SongMASS,… | 62 | 4960 | active |
| Picovoice/porcupine Porcupine is a highly-accurate, lightweight on-device wake word detection engine powered by deep neural networks. It enables always-listeni… | 71 | 4945 | stable |
| Nixtla/statsforecast StatsForecast is a Python library providing fast, scalable implementations of statistical and econometric univariate time series forecastin… | 87 | 4938 | stable |
| KaiyangZhou/deep-person-reid Torchreid is a PyTorch library for deep-learning person re-identification, supporting both image and video reid with end-to-end training an… | 47 | 4926 | stable |
| ArrayFire ArrayFire is a general-purpose tensor/numerical computing library for C, C++, and Python that accelerates array operations on GPUs (CUDA, O… | 65 | 4903 | stable |
| google-deepmind/alphageometry Google DeepMind's implementation of AlphaGeometry and DDAR, AI systems that solve Olympiad-level geometry theorem proving problems. It comb… | 52 | 4899 | active |
| zju3dv/EasyMocap EasyMocap is an open-source Python toolbox for markerless human motion capture and novel view synthesis from RGB videos. It fits parametric… | 66 | 4898 | active |
| libpostal libpostal is a C library for parsing and normalizing street addresses from around the world using statistical NLP and open geo data. It con… | 56 | 4894 | stable |
| tyxsspa/AnyText AnyText is the official implementation of a diffusion-based model for multilingual visual text generation and editing in images, accepted a… | 29 | 4883 | active |
| ACE-Step ACE-Step is an open-source foundation model for music generation that combines diffusion-based generation with a deep compression autoencod… | 46 | 4881 | active |
| PaddlePaddle/VisualDL VisualDL is a deep learning visualization toolkit for PaddlePaddle that provides charts for tracking training metrics, visualizing model st… | 23 | 4881 | stable |
| neilsonnn/image-blaster A Claude skillset that converts a single input image into a full 3D environment, including meshed 3D models (.glb/.obj), Gaussian splats (.… | 48 | 4866 | active |
| Deep Java Library (DJL) Deep Java Library (DJL) is an engine-agnostic, high-level deep learning framework for Java that supports backends like PyTorch, TensorFlow,… | 96 | 4857 | active |
| facebookresearch/flow_matching A PyTorch library for implementing flow matching algorithms with continuous, discrete, and Riemannian flow matching implementations. It acc… | 45 | 4801 | active |
| py-why/EconML EconML is a Python package from Microsoft Research's ALICE project for estimating heterogeneous treatment effects from observational data u… | 83 | 4799 | active |
| Bing-su/adetailer ADetailer is an extension for the Stable Diffusion WebUI (A1111) that automatically detects objects such as faces and hands in generated im… | 74 | 4793 | active |
| UX-Decoder/Segment-Everything-Everywhere-All-At-Once SEEM is the official PyTorch implementation of the NeurIPS 2023 paper 'Segment Everything Everywhere All at Once', a model for universal im… | 21 | 4792 | stable |
| pytorch/ignite PyTorch-Ignite is a high-level library for training and evaluating neural networks in PyTorch flexibly and transparently. It provides an En… | 84 | 4787 | stable |
| SakuraLLM/SakuraLLM SakuraLLM is a series of fine-tuned open large language models for Japanese-to-Chinese translation specialized in light novel and Galgame (… | 58 | 4785 | active |
| SeldonIO/seldon-core Seldon Core 2 is an MLOps and LLMOps framework for deploying, managing, and scaling machine learning models and modular AI applications on … | 66 | 4782 | active |
| facebookresearch/lingua Meta Lingua is a minimal, fast LLM training and inference library built on easy-to-modify PyTorch components for research purposes. It supp… | 33 | 4771 | active |
| OpenDriveLab/UniAD UniAD is a unified end-to-end autonomous driving framework that hierarchically casts perception, prediction, and planning tasks under a pla… | 41 | 4766 | active |
| facebookresearch/vjepa2 Official PyTorch codebase and pretrained models for V-JEPA 2, a self-supervised video encoder trained on internet-scale video, plus V-JEPA … | 49 | 4758 | active |
| Rikorose/DeepFilterNet DeepFilterNet is a low-complexity speech enhancement framework that performs real-time noise suppression on full-band 48kHz audio using dee… | 23 | 4757 | active |
| Flux Flux.jl is a machine learning library written entirely in Julia, providing lightweight abstractions over Julia's native GPU support and aut… | 98 | 4754 | stable |
| hftbacktest A high-frequency trading and market-making backtesting framework written in Rust with Python bindings, which accurately simulates limit ord… | 57 | 4754 | active |
| open-mmlab/mmocr MMOCR is OpenMMLab's PyTorch-based toolbox for text detection, recognition, and key information extraction. It provides a model zoo of OCR … | 23 | 4752 | active |
| rust-ml/linfa Linfa is a Rust machine learning framework offering classical ML algorithms and preprocessing, inspired by Python's scikit-learn. It is org… | 76 | 4751 | active |
| esimov/pigo Pigo is a pure Go library for fast face detection, pupil/eye localization, and facial landmark detection based on the Pixel Intensity Compa… | 28 | 4741 | stable |
| Kimi-Audio Kimi-Audio is an open-source audio foundation model (7B parameters) that unifies audio understanding, generation, and speech conversation i… | 28 | 4736 | active |
| deepseek-ai/Engram Official implementation of Engram, a conditional memory module from DeepSeek that modernizes N-gram embeddings for O(1) lookup as a new spa… | 41 | 4735 | active |
| cjlin1/libsvm LIBSVM is a widely used library and command-line toolset for support vector machine classification, regression, and distribution estimation… | 55 | 4715 | stable |
| PrimeIntellect-ai/verifiers verifiers is a Python library for creating environments to train and evaluate large language models with reinforcement learning. It integra… | 89 | 4712 | active |
| xxnuo/MTranServer MTranServer is an offline machine translation server with very low resource usage and millisecond response times, requiring no GPU. It supp… | 60 | 4711 | active |
| google-deepmind/dm_control Google DeepMind's Python stack for physics-based simulation and reinforcement learning environments built on the MuJoCo physics engine. It … | 99 | 4710 | stable |
| mindspore-ai/mindspore MindSpore is an open-source deep learning framework for training and inference across mobile, edge, and cloud scenarios. It provides automa… | 32 | 4707 | active |
| OpenNMT/CTranslate2 CTranslate2 is a C++ and Python library for fast, memory-efficient inference of Transformer models on CPU and GPU. It uses quantization, la… | 94 | 4684 | stable |
| Kanaries/Rath RATH is an open-source augmented analytics platform that automates exploratory data analysis by discovering patterns, insights, and causal … | 63 | 4683 | active |
| mcmonkeyprojects/SwarmUI SwarmUI (formerly StableSwarmUI) is a modular, self-hosted web user interface for AI image generation, supporting models like Stable Diffus… | 70 | 4675 | active |
| PKU-Alignment/align-anything Align-Anything is a modular Python framework for aligning any-to-any (all-modality) large models with human intentions and values using fee… | 45 | 4671 | active |
| makcedward/nlpaug nlpaug is a Python library for augmenting textual, audio, and spectrogram data to improve machine learning model performance. It provides A… | 100 | 4664 | active |
| Blealtan/efficient-kan An efficient pure-PyTorch implementation of Kolmogorov-Arnold Networks (KAN) that reformulates the computation as matrix multiplications ov… | 25 | 4659 | active |
| Tencent/TNN TNN is a high-performance, lightweight deep learning inference framework developed by Tencent Youtu Lab, supporting mobile, desktop, and se… | 28 | 4656 | active |
| WhisperSpeech/WhisperSpeech WhisperSpeech is an open-source text-to-speech system built by inverting OpenAI's Whisper model, aiming to be 'Stable Diffusion for speech'… | 54 | 4652 | active |
| facebookresearch/vggt-omega VGGT-Omega is a research library from Oxford VGG and Meta AI providing pretrained transformer models for 3D vision tasks such as camera pos… | 58 | 4644 | active |
| joanrod/star-vector StarVector is a foundation model that generates scalable vector graphics (SVG) code from images and text by treating vectorization as a cod… | 46 | 4629 | active |
| hao-ai-lab/FastVideo FastVideo is a unified Python framework for post-training and real-time inference of video diffusion models, covering data preprocessing, f… | 79 | 4616 | active |
| NVlabs/neuralangelo Official PyTorch implementation of Neuralangelo, a CVPR 2023 method for high-fidelity neural surface reconstruction from multi-view images.… | 29 | 4615 | active |
| facebook/prophet Prophet is an open‑source library for automatic forecasting of time‑series data with multiple seasonalities. It provides a simple API in Py… | 99 | 20437 | maintenance |
| royshil/obs-backgroundremoval An OBS Studio plugin that removes and replaces the background in portrait video using ONNX-based machine learning segmentation, acting as a… | 95 | 4607 | active |
| openxla/xla XLA (Accelerated Linear Algebra) is an open-source machine learning compiler that optimizes models from frameworks like PyTorch, TensorFlow… | 77 | 4605 | active |
| Kwai-Kolors/Kolors Kolors is a large-scale latent diffusion model for photorealistic text-to-image synthesis, trained with bilingual (Chinese and English) tex… | 24 | 4598 | active |
| real-stanford/diffusion_policy Official PyTorch implementation of Diffusion Policy, a visuomotor robot policy learning method that represents robot behavior as a conditio… | 31 | 4588 | stable |
| ceres-solver/ceres-solver Ceres Solver is an open-source C++ library for modeling and solving large-scale non-linear optimization problems, including bounded non-lin… | 77 | 4581 | stable |
| fixie-ai/ultravox Ultravox is a fast multimodal LLM that understands human speech directly without a separate ASR stage, by projecting audio into the LLM's e… | 40 | 4577 | active |
| RecBole RecBole is a unified, comprehensive and efficient recommendation library built on Python and PyTorch for reproducing and developing recomme… | 23 | 4569 | stable |
| Tencent-Hunyuan/HunyuanVideo-1.5 HunyuanVideo-1.5 is Tencent's lightweight 8.3B-parameter video generation model supporting text-to-video and image-to-video synthesis. The … | 47 | 4564 | active |
| RuVector RuVector is a Rust-native vector database and memory substrate for AI agents, combining persistent semantic embeddings, graph relationships… | 83 | 4554 | active |
| lucas-maes/le-wm LeWorldModel (LeWM) is the official PyTorch codebase for a JEPA-based world model that trains stably end-to-end from raw pixels using only … | 49 | 4547 | active |
| TencentARC/InstantMesh InstantMesh is a feed-forward framework for generating 3D meshes from a single image using sparse-view large reconstruction models (LRM/Ins… | 25 | 4545 | active |
| NVlabs/tiny-cuda-nn A small, self-contained C++/CUDA framework for training and querying neural networks, featuring a lightning-fast fully fused MLP and a vers… | 66 | 4536 | active |
| OAID/Tengine Tengine is a lightweight, high-performance, modular deep learning inference engine developed by OPEN AI LAB for embedded and edge devices. … | 24 | 4531 | active |
| dedupe Dedupe is a Python library that uses machine learning to perform fuzzy matching, deduplication, and entity resolution on structured data. I… | 44 | 4516 | stable |
| modelscope/ClearerVoice-Studio ClearerVoice-Studio is an open-source, AI-powered speech processing toolkit from ModelScope/Alibaba offering state-of-the-art pretrained mo… | 35 | 4515 | active |
| salesforce/Merlion Merlion is a Python library from Salesforce providing an end-to-end machine learning framework for time series intelligence, covering forec… | 10 | 4486 | active |
| zjunlp/DeepKE DeepKE is an open-source PyTorch-based knowledge extraction toolkit for knowledge graph construction, supporting named entity recognition, … | 61 | 4484 | active |
| DeepXDE DeepXDE is a Python library for scientific machine learning and physics-informed learning, built on top of PyTorch, TensorFlow, JAX, and Pa… | 75 | 4477 | stable |
| Mobile ALOHA Mobile ALOHA is an open-source system for low-cost whole-body teleoperation and data collection with a bimanual mobile manipulation robot, … | 29 | 4476 | active |
| microsoft/LMOps A Microsoft research initiative collecting code and papers on fundamental techniques for building AI products with large language models, c… | 75 | 4475 | active |
| solidSpoon/DashPlayer DashPlayer is an open-source desktop video player built specifically for English learners, using videos as immersive learning material. It … | 96 | 4473 | active |
| astriaai/headshots-starter An open-source Next.js starter kit that generates professional AI headshots from user-uploaded selfies using Astria.ai's fine-tuning and in… | 37 | 4460 | active |
| EvolvingLMMs-Lab/lmms-eval lmms-eval is a unified Python framework for evaluating large multimodal models across text, image, video, and audio tasks, with 100+ benchm… | 87 | 4453 | active |
| layumi/Person_reID_baseline_pytorch A small, friendly PyTorch baseline implementation for person and vehicle re-identification (ReID). It reproduces strong top-conference resu… | 67 | 4450 | stable |
| mosaicml/llm-foundry LLM Foundry is a PyTorch-based codebase for training, finetuning, evaluating, and deploying large language models from 125M to 70B+ paramet… | 62 | 4449 | active |
| xlite-dev/lite.ai.toolkit A lightweight C++ toolkit providing unified APIs for 100+ pre-trained AI models across inference backends like ONNX Runtime, MNN, TensorRT,… | 72 | 4439 | active |
| tensorflow/probability TensorFlow Probability is a Python library for probabilistic reasoning and statistical analysis built on TensorFlow, with a JAX substrate. … | 66 | 4435 | active |
| RestorePhotos A Next.js web application that restores old and blurry face photos using the GFPGAN ML model via the Replicate API. It provides a hosted se… | 32 | 4432 | active |