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 |
|---|---|---|---|
| tianweiy/DMD2 DMD2 is the official PyTorch implementation of Improved Distribution Matching Distillation, a NeurIPS 2024 method that distills diffusion m… | 25 | 1464 | active |
| NVIDIA/tacotron2 NVIDIA's PyTorch implementation of the Tacotron 2 text-to-speech model, which synthesizes mel spectrograms from text for vocoder-based audi… | 32 | 5295 | maintenance |
| yunjey/stargan Official PyTorch implementation of StarGAN, a unified generative adversarial network for multi-domain image-to-image translation (CVPR 2018… | 32 | 5295 | maintenance |
| Turing-Project/WriteGPT WriteGPT is a generative text-creation AI framework built on GPT-2 and other models (EAST, CRNN, BERT), fine-tuned to generate Chinese exam… | 23 | 5282 | maintenance |
| alibaba/Logics-Parsing Logics-Parsing is an end-to-end document parsing model from Alibaba that converts document images into structured output using a single mul… | 61 | 1460 | active |
| pyradiomics PyRadiomics is an open-source Python package for extracting radiomics features from 2D and 3D medical images and binary masks. It provides … | 41 | 1460 | active |
| tensorflow/tpu A collection of reference models and tools for training machine learning models on Google Cloud TPUs, maintained as a public mirror by the … | 68 | 5278 | maintenance |
| zsyOAOA/InvSR InvSR is a Python research library implementing arbitrary-steps image super-resolution via diffusion inversion, leveraging pre-trained diff… | 48 | 1459 | active |
| ByteDance-Seed/m3-agent M3-Agent is a multimodal agent framework from ByteDance Seed that processes real-time visual and auditory inputs to build entity-centric lo… | 45 | 1459 | active |
| supermaven-inc/supermaven-nvim The official Neovim plugin for Supermaven, an AI-powered code completion service. It provides inline suggestions in Neovim with configurabl… | 25 | 1458 | active |
| aeon-toolkit/aeon aeon is a scikit-learn compatible Python toolkit for machine learning on time series, covering classification, regression, clustering, fore… | 93 | 1457 | active |
| Walter0807/MotionBERT Official PyTorch implementation of MotionBERT (ICCV 2023), a unified pretrained model for learning human motion representations from 2D ske… | 61 | 1454 | active |
| InternAgent InternAgent-1.5 is a Python-based unified agentic framework for long-horizon autonomous scientific discovery, orchestrating multi-agent sys… | 58 | 1454 | active |
| zama-ai/concrete-ml Concrete ML is a privacy-preserving machine learning library built on top of Zama's Concrete FHE compiler. It lets data scientists convert … | 65 | 1453 | active |
| YanjieZe/3D-Diffusion-Policy 3D Diffusion Policy (DP3) is a visual imitation learning algorithm that combines compact 3D point cloud representations with diffusion poli… | 43 | 1453 | active |
| dbolya/yolact YOLACT is a PyTorch implementation of a fully convolutional model for real-time instance segmentation, accompanying the YOLACT and YOLACT++… | 47 | 5242 | maintenance |
| qupath/qupath QuPath is an open-source desktop application for bioimage analysis, aimed especially at digital pathology and whole-slide imaging. It provi… | 75 | 1450 | active |
| dleemiller/WordLlama WordLlama is a fast, lightweight Python NLP toolkit built on LLM token embeddings for tasks like similarity computation, ranking, fuzzy ded… | 42 | 1450 | active |
| google-deepmind/rlax RLax is a JAX-based library of building blocks for implementing reinforcement learning agents, providing mathematical operations like value… | 79 | 1449 | active |
| microsoft/KBLaM Official implementation of KBLaM, a method for augmenting pre-trained LLMs with external knowledge by encoding a knowledge base into contin… | 68 | 1449 | active |
| zylo117/Yet-Another-EfficientDet-Pytorch A PyTorch re-implementation of Google's EfficientDet object detection model with pretrained weights matching near-SOTA accuracy at real-tim… | 23 | 5233 | maintenance |
| ThoughtfulDev/EagleEye EagleEye is a Python-based OSINT tool that identifies social media profiles (Instagram, Facebook, Twitter, YouTube) of a person using face … | 32 | 5231 | maintenance |
| amdegroot/ssd.pytorch A PyTorch implementation of the Single Shot MultiBox Detector (SSD) object detection model from the 2016 paper by Wei Liu et al. It include… | 32 | 5222 | maintenance |
| SociallyIneptWeeb/AICoverGen AICoverGen is a Python WebUI (with CLI support) that generates AI song covers by converting vocals in a song to any RVC v2 trained voice, s… | 29 | 1445 | active |
| deepseek-ai/EPLB EPLB is DeepSeek's open-source Expert Parallelism Load Balancer for Mixture-of-Experts models. It computes balanced expert replication and … | 23 | 1443 | active |
| SakanaAI/evolutionary-model-merge Official repository for SakanaAI's Evolutionary Model Merge research, providing code and resources to reproduce paper evaluations of models… | 16 | 1442 | active |
| tianweiy/CausVid CausVid is a research codebase implementing a fast autoregressive video diffusion model distilled from a bidirectional diffusion transforme… | 33 | 1441 | active |
| tensorspace-team/tensorspace TensorSpace is a neural network 3D visualization framework built on TensorFlow.js, Three.js, and Tween.js. It provides Keras-like APIs to b… | 23 | 5198 | maintenance |
| nilearn/nilearn Nilearn is a Python library providing statistical and machine-learning tools for analyzing brain imaging data such as fMRI and MRI volumes … | 91 | 1439 | active |
| neuralchen/SimSwap SimSwap is a PyTorch-based face-swapping framework that performs arbitrary face swaps on images and videos using a single trained model. It… | 23 | 5192 | maintenance |
| sql-machine-learning/sqlflow SQLFlow is a compiler that extends SQL with AI-oriented syntax (training, prediction, evaluation, explanation, and mathematical programming… | 23 | 5190 | maintenance |
| voice-cloning-app/Voice-Cloning-App A Python/PyTorch desktop application for cloning and synthesizing human voices from audio datasets. It handles the full pipeline from autom… | 23 | 1438 | active |
| zuruoke/watermark-removal A machine learning tool that removes watermarks from images using deep learning image inpainting, based on Contextual Attention and Gated C… | 77 | 5181 | maintenance |
| salesforce/CodeGen CodeGen is a family of open-source large language models (350M to 16B parameters) from Salesforce AI Research for program synthesis, genera… | 67 | 5181 | maintenance |
| ymcui/Chinese-ELECTRA Chinese-ELECTRA is a collection of pre-trained Chinese ELECTRA language models released by the HIT-iFLYTEK joint lab, built on the official… | 63 | 1433 | stable |
| FireRedTeam/FireRedTTS2 FireRedTTS-2 is a long-form streaming text-to-speech system for multi-speaker dialogue generation, built in PyTorch with a dual-transformer… | 36 | 1433 | active |
| microsoft/rStar Microsoft's research repository for rStar2-Agent, a 14B math reasoning model trained with agentic reinforcement learning that autonomously … | 38 | 1432 | active |
| apache/cloudberry Apache Cloudberry is an advanced open-source Massively Parallel Processing (MPP) database derived from Greenplum and built on a modern Post… | 76 | 1431 | active |
| paul-buerkner/brms brms is an R package providing a formula-based interface (similar to lme4) for fitting Bayesian generalized (non-)linear multivariate multi… | 66 | 1430 | stable |
| zsyOAOA/ResShift ResShift is an efficient diffusion model for image super-resolution that transfers between low- and high-resolution images by shifting resi… | 59 | 1430 | active |
| facebookresearch/vggsfm VGGSfM is a deep learning-based Structure from Motion pipeline from Meta AI and Oxford VGG that recovers camera poses and 3D point clouds f… | 26 | 1430 | active |
| jasonmayes/Real-Time-Person-Removal A browser-based demo that removes people from complex video backgrounds in real time using TensorFlow.js. It learns the static background o… | 23 | 5151 | maintenance |
| winedarksea/AutoTS AutoTS is a Python library for automated time series forecasting, offering dozens of sklearn-style models (statistical, ML, deep learning) … | 88 | 1429 | active |
| HongyuanLuke/frequencylaw Official code repository for the paper 'Textual Frequency Law on Large Language Models', implementing TFL, TFD, and CTFT methods for studyi… | 46 | 1428 | active |
| kootenpv/whereami A Python CLI tool that uses WiFi access point signal strengths and a scikit-learn RandomForest model to predict your indoor location. It le… | 32 | 5142 | maintenance |
| wenet-e2e/wespeaker WeSpeaker is a research and production-oriented toolkit for speaker embedding learning, supporting speaker verification, recognition, and d… | 66 | 1427 | active |
| XPandora/PhysGaussian PhysGaussian is a research library that integrates Material Point Method (MPM) physics simulation with 3D Gaussian Splatting representation… | 52 | 1427 | active |
| argilla-io/argilla Argilla is an open-source collaboration tool for AI engineers and domain experts to build, annotate, and curate high-quality datasets for N… | 78 | 5140 | maintenance |
| roshan-research/hazm Hazm is a Python library for natural language processing on Persian (Farsi) text, offering normalization, tokenization, stemming, lemmatiza… | 70 | 1426 | active |
| apple/ml-aim Apple's official repository for AIM (Autoregressive Image Models), providing code and pretrained checkpoints for AIMv1 and AIMv2 large visi… | 69 | 1426 | active |
| nv-tlabs/GEN3C GEN3C is NVIDIA's research codebase for a generative video model that achieves precise camera control and temporal 3D consistency using a 3… | 56 | 1425 | active |
| mdbloice/Augmentor Augmentor is a standalone Python library for image augmentation in machine learning, providing a pipeline of stochastic operations like rot… | 32 | 5131 | maintenance |
| SagiPolaczek/NeuralSVG Official PyTorch implementation of NeuralSVG, an ICCV 2025 paper that generates layered, editable SVG vector graphics from text prompts. It… | 44 | 1421 | active |
| lyst/lightfm LightFM is a Python library implementing hybrid recommendation algorithms that combine collaborative filtering with user and item metadata … | 23 | 5116 | maintenance |
| AgentCPM AgentCPM-GUI is an open-source 8B-parameter on-device GUI agent built on MiniCPM-V that takes Android screenshots as input and autonomously… | 44 | 1420 | active |
| MLOps Python Package A Python package template that provides a production-grade code base with MLOps best practices for building and deploying machine learning … | 92 | 1419 | active |
| HugoTini/DeepBump DeepBump is a machine-learning tool that generates normal, height, and curvature maps from single pictures, using a U-Net (MobileNetV2) mod… | 24 | 1419 | active |
| oracle/tribuo Tribuo is a Java machine learning library from Oracle Labs providing classification, regression, clustering, anomaly detection, and multi-l… | 55 | 1418 | active |
| NVlabs/BundleSDF BundleSDF is a CVPR 2023 research implementation from NVIDIA for near real-time 6-DoF pose tracking of unknown rigid objects from monocular… | 62 | 1417 | stable |
| ARahim3/mlx-tune A Python library for fine-tuning LLMs, vision-language, audio (TTS/STT), embedding, OCR, and JEPA models natively on Apple Silicon Macs usi… | 69 | 1416 | active |
| jrzaurin/pytorch-widedeep A PyTorch library for multimodal deep learning that combines tabular data with text and images using Wide and Deep model architectures. It … | 55 | 1416 | active |
| mratsim/Arraymancer Arraymancer is a fast, ergonomic N-dimensional tensor (ndarray) library written in Nim, inspired by NumPy and PyTorch. It provides CPU, CUD… | 67 | 1415 | active |
| SakanaAI/ShinkaEvolve ShinkaEvolve is a Python framework that combines LLMs with evolutionary algorithms to automatically evolve and improve scientific code. It … | 76 | 1414 | active |
| Terrain Diffusion Terrain Diffusion is a Python framework that uses diffusion models as a learned, deterministic replacement for Perlin noise, generating inf… | 58 | 1414 | active |
| Lyken17/pytorch-OpCounter THOP (PyTorch-OpCounter) is a Python library that counts the MACs/FLOPs and parameters of PyTorch models. It profiles arbitrary nn.Modules … | 32 | 5083 | maintenance |
| explosion/spacy-transformers A spaCy v3 extension package that provides pipeline components for using pretrained transformer models like BERT, RoBERTa, XLNet, and GPT-2… | 65 | 1413 | stable |
| CSAILVision/semantic-segmentation-pytorch A PyTorch implementation of semantic segmentation (scene parsing) models for the MIT ADE20K dataset, including pretrained model zoo and tra… | 32 | 5081 | maintenance |
| affinelayer/pix2pix-tensorflow A TensorFlow implementation of pix2pix, a conditional GAN that learns a mapping from input images to output images. It is a faithful port o… | 32 | 5078 | maintenance |
| koaning/scikit-lego scikit-lego is a Python library of extra building blocks—custom transformers, models, metrics, and datasets—that are fully compatible with … | 88 | 1412 | active |
| AgibotTech/genie_sim Genie Sim is AgiBot's open-source simulation platform for embodied intelligence, providing environment reconstruction, LLM-driven scene gen… | 70 | 1412 | active |
| lucidrains/self-rewarding-lm-pytorch A PyTorch library implementing the Self-Rewarding Language Model training framework from MetaAI, along with the SPIN training method. It pr… | 17 | 1412 | active |
| nyrahealth/CrisperWhisper CrisperWhisper 2.0 is a controllable speech recognition model and Python library that transcribes audio either verbatim (including fillers,… | 88 | 1411 | active |
| dailenson/SDT Official PyTorch implementation of the CVPR 2023 paper 'Disentangling Writer and Character Styles for Handwriting Generation' (SDT). It gen… | 40 | 1411 | active |
| scraed/LanPaint LanPaint is a training-free inpainting sampler for stable diffusion models, implemented as a ComfyUI custom node. It uses iterative 'think … | 83 | 1409 | active |
| PrimiHub PrimiHub is an open-source privacy-preserving computing platform built by a team of cryptography experts, supporting secure multi-party com… | 40 | 1409 | active |
| lucidrains/transfusion-pytorch A PyTorch implementation of Transfusion, MetaAI's approach to predicting the next token and diffusing images with a single multi-modal mode… | 69 | 1408 | active |
| SpikingBrain SpikingBrain-7B is a brain-inspired large language model that combines hybrid efficient attention, MoE modules, and spike encoding, with a … | 51 | 1408 | active |
| om-ai-lab/OmDet OmDet-Turbo is a PyTorch implementation of a transformer-based open-vocabulary object detection model that detects arbitrary user-defined o… | 54 | 1406 | active |
| open-gigaai/giga-world-policy GigaWorld-Policy is a World Action Model (WAM) for robot policy learning that jointly models actions and future visual observations during … | 53 | 1406 | active |
| XiaoMi/mace MACE (Mobile AI Compute Engine) is a deep learning inference framework optimized for mobile heterogeneous computing on Android, iOS, Linux … | 23 | 5052 | maintenance |
| zju3dv/street_gaussians Street Gaussians is a research implementation of the ECCV 2024 paper 'Modeling Dynamic Urban Scenes with Gaussian Splatting', which reconst… | 69 | 1405 | active |
| gnobitab/InstaFlow InstaFlow is a one-step text-to-image generation model based on Rectified Flow, enabling ultra-fast Stable Diffusion inference without iter… | 29 | 1403 | active |
| Irodori-TTS Irodori-TTS is a Flow Matching-based text-to-speech model with training and inference code, built on a Rectified Flow Diffusion Transformer… | 58 | 1401 | active |
| McGill-NLP/webllama WebLlama is a framework for building Llama-3-powered agents that browse the web by following natural language instructions and dialogue. It… | 56 | 1399 | active |
| yfeng95/PRNet PRNet is a Python/TensorFlow implementation of the ECCV 2018 Position Map Regression Network for joint 3D face reconstruction and dense ali… | 32 | 5018 | maintenance |
| Denys88/rl_games RL Games is a high-performance reinforcement learning library built on PyTorch, focused on training agents in massively parallel GPU-based … | 74 | 1397 | active |
| Zejun-Yang/AniPortrait AniPortrait is a Python framework from Tencent that generates photorealistic portrait animations from an audio clip and a reference image, … | 26 | 5014 | maintenance |
| SebastienZh/StockTradebyZ A semi-automated stock screening application for China's A-share market that fetches daily K-line data via Tushare, applies quantitative pr… | 48 | 1395 | active |
| aklofas/kicad-happy A collection of AI coding agent skills that turn Claude Code, OpenAI Codex, and similar agents into KiCad electronics design assistants. It… | 82 | 1394 | active |
| galilai-group/lejepa LeJEPA is a Python framework for scalable, theoretically grounded self-supervised representation learning based on Joint-Embedding Predicti… | 42 | 1393 | active |
| ByteDance-Seed/SeedVR SeedVR/SeedVR2 are diffusion-transformer based models for generic real-world and AIGC video and image restoration, with SeedVR2 using adver… | 44 | 1390 | active |
| cszn/BSRGAN BSRGAN is a PyTorch implementation of a practical degradation model for deep blind image super-resolution, presented at ICCV 2021. It provi… | 32 | 1390 | stable |
| logtd/ComfyUI-Fluxtapoz A set of ComfyUI custom nodes for editing and stylizing images with Flux models, implementing techniques like RF-Inversion, RF-Edit, Firefl… | 24 | 1390 | active |
| Junyi42/monst3r MonST3R is the official PyTorch implementation of an ICLR 2025 paper that estimates per-timestep geometry (pointmaps) from dynamic videos i… | 22 | 1390 | active |
| PyO3/rust-numpy Rust bindings for the NumPy C-API built on PyO3, enabling zero-copy conversion between Rust ndarray types and NumPy arrays. It supports wri… | 85 | 1388 | active |
| QwenLM/Qwen3-VL-Embedding Qwen3-VL-Embedding and Qwen3-VL-Reranker are state-of-the-art multimodal embedding and reranking models built on the Qwen3-VL foundation mo… | 52 | 1387 | active |
| haoheliu/voicefixer VoiceFixer is a Python library and CLI tool for general speech restoration, using a pretrained neural vocoder to restore degraded human spe… | 23 | 1387 | stable |
| tiny-tpu-v2/tiny-tpu A minimal tensor processing unit (TPU) written in Verilog, reinvented from Google's TPU V1/V2 architectures. It includes a systolic array o… | 50 | 1386 | active |
| zjunlp/KnowLM KnowLM is an open-source framework for building knowledgeable large language models, covering data processing, pre-training, fine-tuning, k… | 31 | 1386 | active |
| Kratos Multiphysics Kratos Multiphysics is a C++ framework with an extensive Python interface for building parallel, multi-disciplinary simulation software suc… | 91 | 1385 | active |