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 |
|---|---|---|---|
| TensorFlow Extended (TFX) TensorFlow Extended (TFX) is an end-to-end, Google-production-scale platform for building and deploying production machine learning pipelin… | 87 | 2191 | stable |
| MoonshotAI/MoBA MoBA (Mixture of Block Attention) is a PyTorch implementation of a trainable block-sparse attention mechanism for long-context large langua… | 24 | 2191 | active |
| Vaex Vaex is a high-performance Python DataFrame library for lazy, out-of-core processing of large tabular datasets, using memory mapping and ze… | 52 | 8512 | maintenance |
| Vespa314/chan.py An open-source Python framework implementing Chanlun (缠论) technical analysis for stock trading, computing fractals, strokes, segments, pivo… | 75 | 2188 | active |
| yyfz/Pi3 Pi3 (π³) is a feed-forward neural network for visual geometry reconstruction that eliminates the need for a fixed reference view, using a p… | 56 | 2188 | active |
| withcatai/node-llama-cpp node-llama-cpp is a Node.js library providing bindings to llama.cpp for running LLMs locally, with pre-built binaries and automatic GPU sup… | 95 | 2187 | active |
| yatengLG/ISAT_with_segment_anything ISAT_with_segment_anything is an interactive semi-automatic image annotation tool built on the Segment Anything Model family (SAM, SAM2, SA… | 81 | 2186 | active |
| tensorflow/quantum TensorFlow Quantum is a Python framework for hybrid quantum-classical machine learning that integrates Cirq quantum circuits with TensorFlo… | 80 | 2186 | active |
| hls4ml hls4ml is a Python package that converts machine learning models from Keras, PyTorch, and ONNX into high-level synthesis (C++) code for FPG… | 80 | 2185 | active |
| Harry24k/adversarial-attacks-pytorch Torchattacks is a PyTorch library providing implementations of adversarial attacks to generate adversarial examples against deep learning m… | 23 | 2185 | active |
| tandpfun/wardrobe A self-hosted web application that detects garments in photos, extracts clean product cutouts, and generates modeled editorial previews usi… | 51 | 2175 | active |
| learnsyslab/gym-pybullet-drones A Python library providing PyBullet-based Gymnasium environments for simulating quadcopter control in single- and multi-agent reinforcement… | 68 | 2173 | active |
| ViTAE-Transformer/ViTPose Official PyTorch implementation of ViTPose and ViTPose++, Vision Transformer models for human and generic body pose estimation from NeurIPS… | 55 | 2173 | stable |
| Tencent-Hunyuan/HunyuanVideo-Avatar HunyuanVideo-Avatar is Tencent's open-source model and inference code for high-fidelity audio-driven human animation, generating talking av… | 42 | 2173 | active |
| TsingZ0/PFLlib PFLlib is a beginner-friendly Python library and benchmark for traditional and personalized federated learning, offering 39 algorithms, 24 … | 55 | 2166 | active |
| PrimeIntellect-ai/prime-rl prime-rl is a Python framework for large-scale, fully asynchronous reinforcement learning training of language models, built on FSDP2 for t… | 88 | 2164 | active |
| 1038lab/ComfyUI-RMBG A ComfyUI custom node package for advanced image background removal and segmentation of objects, faces, clothing, and fashion elements. It … | 66 | 2162 | active |
| hexo-ai/sia SIA is a Python framework implementing a self-improving AI loop in which a Meta-Agent generates a task-specific Target Agent and a Feedback… | 56 | 2161 | active |
| JasonMa0012/JTRP JTRP is a Unity HDRP-based toon shading (cel shading / NPR) render pipeline toolkit supporting DX12 ray tracing, built as a second developm… | 23 | 2161 | active |
| SpikingJelly SpikingJelly is an open-source deep learning framework for Spiking Neural Networks (SNNs) built on PyTorch. It provides a beginner-friendly… | 77 | 2159 | active |
| rupeshs/fastsdcpu FastSD CPU is a Python application that runs Stable Diffusion image generation quickly on CPUs and Intel AI PCs using Latent Consistency Mo… | 75 | 2158 | active |
| PKU-YuanGroup/Helios Helios is a 14B autoregressive diffusion model for real-time, minute-scale video generation supporting text-to-video, image-to-video, and v… | 56 | 2158 | active |
| facebookresearch/mae A PyTorch/GPU re-implementation of the Masked Autoencoders (MAE) paper for self-supervised vision learning. It includes pre-training code, … | 10 | 8362 | maintenance |
| Open Motion Planning Library (OMPL) OMPL is an open-source C++ library of over 40 sampling-based motion planning algorithms (RRT-Connect, PRM, RRT*, KPIECE, etc.) across more … | 90 | 2156 | stable |
| official-pikafish/Pikafish Pikafish is a free, open-source UCI xiangqi (Chinese chess) engine derived from Stockfish, using NNUE neural network evaluation to analyze … | 86 | 2156 | active |
| espressif/esp-who ESP-WHO is an image processing development platform from Espressif providing face detection, face recognition, pedestrian detection, and QR… | 63 | 2156 | active |
| jd-opensource/JoyAI-Image JoyAI-Image is a unified multimodal foundation model for image understanding, text-to-image generation, and instruction-guided image editin… | 54 | 2156 | active |
| NannyML/nannyml NannyML is an open-source Python library for post-deployment machine learning monitoring. It estimates model performance without access to … | 36 | 2155 | active |
| google-deepmind/pysc2 PySC2 is DeepMind's Python component of the StarCraft II Learning Environment, exposing Blizzard's StarCraft II Machine Learning API as a P… | 23 | 8329 | maintenance |
| ZiYang-xie/WorldGen WorldGen is a Python library that generates full 3D scenes in seconds from text prompts or images, supporting 360-degree consistent explora… | 51 | 2150 | active |
| baowenbo/DAIN DAIN is a PyTorch-based deep learning model for depth-aware video frame interpolation, synthesizing intermediate frames between existing on… | 32 | 8316 | maintenance |
| google/trax Trax is an end-to-end deep learning library built on JAX and TensorFlow that focuses on clear code and speed, developed and maintained by t… | 10 | 8306 | maintenance |
| tumuyan/RealSR-NCNN-Android An Android application for image super-resolution and upscaling built on NCNN and MNN inference engines, bundling models like RealSR, Real-… | 77 | 2144 | active |
| serge-sans-paille/pythran Pythran is an ahead-of-time compiler that translates a subset of Python, focused on scientific computing with NumPy, into fast native C++ e… | 75 | 2144 | active |
| xinsir6/ControlNetPlus ControlNet++ is an all-in-one ControlNet model and architecture supporting 10+ control types for text-to-image generation and image editing… | 24 | 2143 | active |
| Lumiwealth/lumibot Lumibot is a Python framework for building, backtesting, and deploying algorithmic trading strategies and AI-agent trading teams across sto… | 96 | 2142 | active |
| intel/openvino-plugins-ai-audacity A set of AI-enabled effects, generators, and analyzers for Audacity, powered by Intel OpenVINO running fully locally on CPU, GPU, or NPU. I… | 75 | 2142 | active |
| cvxpy/cvxpylayers CVXPYlayers is a Python library for constructing differentiable convex optimization layers in PyTorch, JAX, and MLX using CVXPY. It solves … | 90 | 2141 | active |
| PKU-YuanGroup/LLaVA-CoT LLaVA-CoT is an 11B vision language model with training, inference, and dataset-generation code for spontaneous step-by-step multimodal rea… | 44 | 2131 | active |
| LiheYoung/Depth-Anything Depth Anything is a monocular depth estimation foundation model trained on 1.5M labeled and 62M+ unlabeled images, released as a Python lib… | 26 | 8240 | maintenance |
| utkuozbulak/pytorch-cnn-visualizations A PyTorch library implementing a wide range of convolutional neural network visualization and interpretability techniques, including Grad-C… | 32 | 8236 | maintenance |
| tairov/llama2.mojo A single-file implementation of Llama 2 transformer inference written in pure Mojo, leveraging SIMD and vectorization for fast CPU inferenc… | 71 | 2129 | active |
| theamusing/perfectPixel A Python library that automatically detects the optimal grid size in AI-generated pixel art images and refines them into clean, perfectly a… | 42 | 2129 | active |
| lukemelas/EfficientNet-PyTorch A PyTorch implementation of the EfficientNet convolutional neural network family with pretrained ImageNet weights. It provides a simple pip… | 23 | 8219 | maintenance |
| autonomousvision/sdfstudio SDFStudio is a unified and modular framework for neural implicit surface reconstruction built on top of nerfstudio. It provides unified imp… | 32 | 2124 | active |
| marlbenchmark/on-policy The official PyTorch implementation of Multi-Agent PPO (MAPPO), an on-policy reinforcement learning algorithm for cooperative multi-agent s… | 32 | 2123 | active |
| ozgrozer/ai-renamer A Node.js CLI tool that uses local AI models (via Ollama or LM Studio) or OpenAI to intelligently rename files based on their contents, inc… | 24 | 2123 | active |
| nv-tlabs/vipe ViPE is an open-source video processing engine from NVIDIA that estimates camera intrinsics, camera motion, and dense near-metric depth map… | 77 | 2122 | active |
| geatpy-dev/geatpy Geatpy is a high-performance genetic and evolutionary algorithm toolbox for Python supporting single-objective, multi-objective, and many-o… | 48 | 2122 | active |
| sunnypilot/sunnypilot sunnypilot is an open-source driver assistance system forked from comma.ai's openpilot, providing adaptive cruise control, automated lane c… | 95 | 2121 | active |
| martin226/vibe-draw Vibe Draw is a self-hosted web application that turns rough 2D sketches into 3D models and interactive 3D worlds using AI. It combines a Ne… | 30 | 2121 | active |
| 3DTopia/LGM LGM is the official PyTorch implementation of an ECCV 2024 Oral paper that generates high-resolution 3D models from text prompts or single-… | 26 | 2118 | active |
| eloialonso/diamond DIAMOND is a Python implementation of a reinforcement learning agent trained entirely inside a diffusion-based world model, presented as a … | 25 | 2118 | active |
| deepspeedai/DeepSpeed-MII DeepSpeed-MII is a Python library for high-throughput, low-latency large language model inference, built on DeepSpeed. It provides blocked … | 39 | 2114 | active |
| facebookresearch/ai4animationpy AI4AnimationPy is a Python framework for AI-driven character animation using neural networks, providing motion capture processing, training… | 56 | 2111 | active |
| automl/auto-sklearn auto-sklearn is an automated machine learning (AutoML) toolkit that acts as a drop-in replacement for scikit-learn estimators, automaticall… | 65 | 8135 | maintenance |
| patrick-kidger/diffrax Diffrax is a JAX-based library providing numerical differential equation solvers for ODEs, SDEs, and CDEs. It is fully autodifferentiable a… | 80 | 2106 | active |
| Avalanche Avalanche is an end-to-end continual learning library built on PyTorch, developed by ContinualAI. It provides modules for benchmarks, train… | 24 | 2104 | active |
| kohya-ss/musubi-tuner Musubi Tuner is a set of Python scripts for training LoRA (Low-Rank Adaptation) adapters for video and image generation model architectures… | 88 | 2103 | active |
| River-Zhang/ICEdit ICEdit (In-Context Edit) is a research framework for instruction-based image editing built on large-scale Diffusion Transformers, using a L… | 42 | 2102 | active |
| SUDO-AI-3D/zero123plus Zero123++ is a diffusion base model that generates consistent multi-view images from a single input image, intended as a stepping stone for… | 28 | 2099 | active |
| Open-Reasoner-Zero/Open-Reasoner-Zero Open-Reasoner-Zero is an open-source implementation of large-scale reinforcement learning training for reasoning-oriented language models, … | 28 | 2098 | active |
| RenderKit/oidn Intel Open Image Denoise is an open-source library of high-performance, deep-learning-based denoising filters for images rendered with ray … | 88 | 2097 | stable |
| deepcharles/ruptures ruptures is a Python library for offline change point detection in non-stationary signals. It provides exact and approximate detection algo… | 67 | 2096 | active |
| bigcode-project/starcoder2 StarCoder2 is a family of open code generation language models (3B, 7B, 15B) trained on 600+ programming languages from The Stack v2, with … | 27 | 2095 | active |
| HRM-Text HRM-Text is a 1B-parameter text generation model based on the hierarchical recurrent HRM architecture, released with a complete pretraining… | 57 | 2093 | active |
| vitoplantamura/OnnxStream A lightweight C++ inference library for ONNX models that streams weights to run large models in very little memory, accelerated by XNNPACK.… | 56 | 2089 | active |
| PaddlePaddle/PaddleGAN PaddleGAN is a Python library providing high-performance implementations of classic and state-of-the-art Generative Adversarial Networks bu… | 23 | 8048 | maintenance |
| NanoNets/docext docext is an on-premises document intelligence toolkit powered by vision-language models, offering OCR-free structured data extraction, PDF… | 47 | 2086 | active |
| Rand Rand is a set of Rust crates providing random number generation, including a standard RNG trait, fast pseudo-random and cryptographically-s… | 96 | 2085 | stable |
| OpenSeeFace OpenSeeFace is a robust realtime face and facial landmark tracking library that runs on CPU at 30-60 fps using ONNX-converted MobileNetV3 m… | 85 | 2084 | active |
| alex-damian/pulse PULSE is a Python research implementation of a CVPR 2020 paper that upscales low-resolution face photos by searching the latent space of a … | 32 | 8026 | maintenance |
| alganzory/HaramBlur HaramBlur is a browser extension that automatically detects and blurs inappropriate images and videos on web pages using on-device machine … | 19 | 2083 | active |
| ali-vilab/In-Context-LoRA Official repository for In-Context LoRA (IC-LoRA), a framework for adapting Diffusion Transformers to diverse visual generation tasks via L… | 22 | 2081 | active |
| stepfun-ai/Step-3.5-Flash Step 3.5 Flash is an open-source 196B-parameter sparse Mixture-of-Experts foundation model (11B active per token) from StepFun, focused on … | 46 | 2080 | active |
| uber/orbit Orbit is a Python package for Bayesian time series forecasting and inference with an initialize-fit-predict interface. It uses probabilisti… | 76 | 2079 | stable |
| NVIDIA-NeMo/RL NeMo RL is NVIDIA's open-source post-training library for scaling reinforcement learning methods (GRPO, PPO, DPO, SFT, distillation) on LLM… | 79 | 2077 | active |
| jeshraghian/snntorch snnTorch is a Python library for gradient-based deep learning with spiking neural networks, built as an extension of PyTorch. It provides s… | 68 | 2075 | active |
| marcoslucianops/DeepStream-Yolo A collection of configuration files, parsers, and conversion utilities for running YOLO-family object detection models on NVIDIA DeepStream… | 57 | 2074 | active |
| MatrAIx-ai/MatrAIx-Persona-8B MatrAIx is a population-scale, persona-driven evaluation framework that instantiates sampled persona records as LLM agents to simulate hete… | 56 | 2074 | active |
| WenjieDu/PyPOTS PyPOTS is a Python toolbox for machine learning and data mining on partially-observed time series with missing values. It integrates 50+ st… | 89 | 2072 | active |
| WhatDreamsCost/WhatDreamsCost-ComfyUI A collection of free custom ComfyUI nodes and workflows, centered on LTX Director, a timeline-based tool for directing LTX video generation… | 54 | 2072 | active |
| bytedance/Protenix Protenix is an open-source PyTorch reproduction of AlphaFold 3 for high-accuracy biomolecular structure prediction, covering proteins, nucl… | 78 | 2070 | active |
| facebookresearch/ConvNeXt-V2 Official PyTorch implementation of ConvNeXt V2, a family of pure convolutional neural network models co-designed with a fully convolutional… | 10 | 2065 | stable |
| brightmart/text_classification A collection of deep learning baseline models for text classification in NLP, implemented in TensorFlow. It covers classic architectures li… | 32 | 7938 | maintenance |
| OpenMOSS/MOSS-Transcribe-Diarize MOSS-Transcribe-Diarize 0.9B is an open-source end-to-end audio understanding model that jointly performs multi-speaker speech transcriptio… | 57 | 2064 | active |
| lupantech/AgentFlow AgentFlow is a trainable, tool-integrated agentic framework that coordinates planner, executor, verifier, and generator modules through an … | 44 | 2064 | active |
| facebookresearch/theseus Theseus is a PyTorch-based library for building custom differentiable nonlinear optimization layers, supporting problems in robotics and vi… | 23 | 2060 | active |
| ricky0123/vad A JavaScript/TypeScript library that runs Silero VAD via ONNX Runtime Web to detect voice activity directly in the browser. It provides a s… | 96 | 2059 | active |
| zju3dv/GVHMR GVHMR is a research codebase implementing the SIGGRAPH Asia 2024 paper 'World-Grounded Human Motion Recovery via Gravity-View Coordinates'.… | 66 | 2059 | active |
| jd-opensource/JoyAI-Echo JoyAI-Echo is a Python framework for long-horizon audio-visual generation, producing coherent multi-shot videos up to ~5 minutes with paire… | 57 | 2057 | active |
| shanglianlm0525/PyTorch-Networks A collection of PyTorch implementations of classic and modern CNN architectures, covering classification, detection, segmentation, face, an… | 49 | 2056 | active |
| DeePMD-kit DeePMD-kit is a deep learning package for building many-body potential energy representations and running molecular dynamics simulations. I… | 92 | 2055 | stable |
| jaywalnut310/vits VITS is the official PyTorch implementation of an end-to-end text-to-speech model based on a conditional variational autoencoder with adver… | 32 | 7894 | maintenance |
| n00mkrad/flowframes Flowframes is a Windows GUI application for AI-based video frame interpolation, supporting RIFE (Pytorch & NCNN), DAIN (NCNN), and FLAVR (P… | 66 | 2048 | active |
| Sygil-Dev/sygil-webui A browser-based web UI for generating images with Stable Diffusion, built in Python with Gradio and Streamlit frontends. It supports text-t… | 70 | 7863 | maintenance |
| EvalAI EvalAI is an open-source platform for evaluating and comparing machine learning and AI algorithms at scale. It provides a central leaderboa… | 67 | 2045 | active |
| serengil/retinaface RetinaFace is a Python library for deep learning based face detection, built on TensorFlow and derived from the insightface project's Retin… | 66 | 2042 | active |
| juanmc2005/diart Diart is a Python framework for building AI-powered real-time audio applications, best known for state-of-the-art streaming speaker diariza… | 59 | 2041 | active |
| Yi Yi is a family of open-source large language models trained from scratch by 01.AI, including base and chat models in multiple sizes, with b… | 28 | 7830 | maintenance |