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
princeton-vl/DPVO DPVO is a deep learning-based visual odometry and SLAM system that estimates camera trajectories from video or image sequences using patch-…	75	1127	active
nnsight nnsight is a Python library for interpreting and intervening on the internals of deep learning models, built on PyTorch. It lets researcher…	96	1126	active
lasgroup/SDPO SDPO (Self-Distilled Policy Optimization) is a research library implementing a reinforcement learning framework for post-training large lan…	53	1126	active
cggos/imu_x_fusion A C++/ROS library implementing loosely-coupled IMU + X (GNSS, 6DoF odometry) fusion localization using ESKF, IEKF, UKF variants, and MAP es…	34	1126	active
lkwq007/stablediffusion-infinity A web application for outpainting with Stable Diffusion on an effectively infinite canvas, built with PyScript and Gradio. It extends image…	32	3887	maintenance
hanruihua/NeuPAN NeuPAN is a real-time, map-free robot motion planner that maps raw obstacle point data directly to control actions using an end-to-end MPC-…	63	1124	active
PRIME-RL/TTRL TTRL is an open-source implementation of Test-Time Reinforcement Learning, a method for training LLMs with RL on unlabeled data using major…	49	1124	active
datadreamer-dev/DataDreamer DataDreamer is an open-source Python library for prompting LLMs, generating synthetic datasets, and training or aligning models in reproduc…	32	1124	active
THU-MIG/RepViT Official PyTorch implementation of RepViT, a family of lightweight CNNs designed by integrating efficient ViT architectural designs into Mo…	20	1123	stable
THUDM/SwissArmyTransformer SwissArmyTransformer (sat) is a PyTorch library for developing custom Transformer model variants where models like BERT, GPT, T5, GLM, and …	23	1122	active
hkchengrex/Cutie Cutie is a video object segmentation framework with object-level memory reading, a follow-up to XMem offering better consistency, robustnes…	18	1122	active
premieroctet/photoshot Photoshot is an open-source web application that generates custom AI avatars from user-uploaded selfies using fine-tuned text-to-image mode…	23	3873	maintenance
AdaptiveMotorControlLab/CEBRA CEBRA is a Python library for self-supervised learning of consistent latent embeddings from high-dimensional time-series recordings, using …	87	1121	active
princeton-vl/RAFT-Stereo RAFT-Stereo is a PyTorch implementation of a deep learning model for stereo matching that estimates disparity maps from stereo image pairs …	71	1121	stable
MST (Spectral Compressive Imaging Toolbox) A Python toolbox for spectral compressive imaging reconstruction that implements over 15 algorithms including MST, CST, DAUHST, BiSCI, HDNe…	49	1121	active
FlagAI-Open/FlagAI FlagAI is a Python toolkit for training, fine-tuning, and deploying large-scale AI models across NLP, CV, and vision-language tasks. It int…	60	3870	maintenance
luxkun/ReGoap ReGoap is a generic C# GOAP (Goal Oriented Action Planning) library with adapters for Unity and Godot. It provides an engine-agnostic plann…	54	1120	active
PaddlePaddle/PaddleHelix PaddleHelix is a bio-computing platform built on PaddlePaddle featuring large-scale representation learning and multi-task deep learning fo…	52	1120	active
yangxue0827/RotationDetection AlphaRotate is a TensorFlow-based benchmark and toolbox for rotated (oriented) object detection, implementing detectors such as R2CNN, Reti…	23	1120	active
yandex-research/tabm TabM is a PyTorch-based deep learning model for tabular data that efficiently imitates an ensemble of MLPs through parameter-efficient ense…	32	1119	active
HITsz-TMG/Uni-MoE Uni-MoE is a family of open-source Mixture-of-Experts (MoE) based omnimodal large language models that understand and generate across text,…	65	1118	active
SCLBD/DeepfakeBench DeepfakeBench is a comprehensive benchmark framework for deepfake detection, providing a unified platform for data management, implementati…	33	1118	active
luchris429/purejaxrl PureJaxRL provides end-to-end reinforcement learning training pipelines implemented entirely in JAX, including environments, enabling massi…	31	1118	active
Lasagne Lasagne is a lightweight Python library for building and training neural networks on top of Theano. It supports feed-forward, convolutional…	23	3858	maintenance
louiszengCN/CarlaAir CarlaAir is an open-source simulation infrastructure that combines CARLA's high-fidelity urban driving environments with physics-accurate m…	68	1117	active
grf-labs/grf GRF is an R package (with C++ core) for forest-based statistical estimation and inference, providing non-parametric methods for heterogeneo…	64	1117	stable
autonomousvision/navsim NAVSIM is a data-driven pseudo-simulation framework and benchmark for autonomous vehicle planning, evaluating driving agents non-reactively…	46	1117	active
prometheus-eval/prometheus-eval Prometheus-Eval is a Python library for evaluating LLM generation outputs using the Prometheus family of open evaluator models and GPT-4 as…	20	1117	active
MatchZoo MatchZoo is a Python toolkit for designing, comparing, and sharing deep text matching models. It provides a unified data pipeline, pre-buil…	23	3849	maintenance
openai/improved-diffusion The official codebase for OpenAI's Improved Denoising Diffusion Probabilistic Models paper, providing a Python package for training and sam…	32	3848	maintenance
yahoo/TensorFlowOnSpark TensorFlowOnSpark is a Python library that lets existing TensorFlow programs run distributed training and inference on Apache Spark and Had…	23	3847	maintenance
storyicon/comfyui_segment_anything A ComfyUI custom node that combines GroundingDINO and Segment Anything (SAM) to segment any element in an image using semantic text prompts…	28	1114	active
open-mmlab/PowerPaint PowerPaint is a versatile image inpainting model (ECCV 2024) built on diffusion models that handles text-guided object insertion, object re…	67	1113	active
boona13/image-extender An open-source Next.js web app for AI image outpainting that extends images in any direction using Gemini models via OpenRouter, with Poiss…	48	1113	active
zju3dv/PGSR PGSR is a Python implementation of a planar-based Gaussian Splatting method for efficient, high-fidelity 3D surface reconstruction from mul…	24	1113	active
LAMDA-CL/PyCIL PyCIL is a PyTorch-based Python toolbox for class-incremental learning, implementing the largest collection of CIL methods for reproducible…	48	1112	active
szymanowiczs/splatter-image Official PyTorch implementation of 'Splatter Image: Ultra-Fast Single-View 3D Reconstruction' (CVPR 2024), which uses an image-to-image net…	27	1111	active
baofff/U-ViT U-ViT is the official PyTorch implementation of a ViT-based backbone architecture for diffusion models from the CVPR 2023 paper 'All are Wo…	32	1109	stable
WangLibo1995/GeoSeg GeoSeg is an open-source PyTorch-based semantic segmentation toolbox focused on Vision Transformers for remote sensing imagery, featuring t…	32	1109	active
jina-ai/discoart DiscoArt is a Python library that wraps Disco Diffusion (CLIP-guided diffusion) so generative artists and developers can create AI artworks…	23	3818	maintenance
waymo-research/waymax Waymax is a lightweight, multi-agent autonomous driving simulator built entirely in JAX, based on the Waymo Open Motion Dataset. It support…	46	1108	active
gxtrobot/bustag Bustag is a self-hosted web application that periodically scrapes new media title (番号) listings, lets users tag items as liked or disliked,…	23	3817	maintenance
google-ai-edge/litert-torch LiteRT Torch is a Python library that converts PyTorch models into .tflite format for running with LiteRT on Android, iOS, and IoT devices.…	86	1107	active
yzslab/gaussian-splatting-lightning A PyTorch Lightning implementation of 3D Gaussian Splatting with many derived algorithms (Mip-Splatting, LightGaussian, 2DGS, deformable Ga…	55	1107	active
ARM-software/CMSIS-DSP CMSIS-DSP is ARM's optimized embedded compute library providing DSP and math kernels for Cortex-M and Cortex-A processors, including FFT, f…	95	1106	stable
rusty1s/pytorch_sparse A PyTorch extension library providing optimized sparse matrix operations (coalesce, transpose, sparse-dense and sparse-sparse multiplicatio…	57	1106	active
PyABSA PyABSA is a PyTorch-based library providing state-of-the-art models for aspect-based sentiment analysis, including aspect term extraction, …	66	1105	active
fraunhoferportugal/tsfel TSFEL is an open-source Python library for extracting features from time series signals across statistical, temporal, spectral, and fractal…	65	1105	active
google-research/multinerf Google Research's official code release for three NeRF papers: Mip-NeRF 360, Ref-NeRF, and RawNeRF, written in JAX. It trains neural radian…	10	3805	maintenance
aiptimizer/TurboOCR TurboOCR is an extremely fast GPU-accelerated document parser written in C++ that combines OCR, layout analysis, table extraction, and form…	80	1104	active
PharMolix/OpenBioMed OpenBioMed is an open-source toolkit and agent platform for biomedicine and life science, offering multimodal models like BioMedGPT-Mol, 45…	69	1104	active
TencentARC/T2I-Adapter Official implementation of T2I-Adapter, lightweight adapter models that add controllable conditioning (sketch, canny, lineart, depth, pose)…	31	3799	maintenance
gunrock/gunrock Gunrock is a CUDA/C++ library for GPU-accelerated graph analytics built on a high-level, data-centric abstraction over vertex and edge fron…	57	1102	active
nihui/rife-ncnn-vulkan A command-line tool implementing RIFE (Real-Time Intermediate Flow Estimation) for video frame interpolation using the ncnn neural network …	23	1102	active
naturomics/CapsNet-Tensorflow A TensorFlow implementation of CapsNet (Capsule Networks) based on Geoffrey Hinton's paper 'Dynamic Routing Between Capsules'. It supports …	32	3788	maintenance
apache/systemds Apache SystemDS is an open-source machine learning system covering the end-to-end data science lifecycle, from data cleaning and feature en…	83	1100	active
geoopt/geoopt Geoopt is a PyTorch library providing manifold-aware optimizers and manifold tensor containers for Riemannian optimization. It implements R…	67	1100	active
mit-han-lab/streaming-vlm StreamingVLM is a vision-language model framework from MIT Han Lab for real-time understanding of effectively infinite video streams. It ma…	35	1100	active
lucidrains/tab-transformer-pytorch A PyTorch implementation of the TabTransformer architecture, an attention-based neural network for tabular data, also including the FT Tran…	64	1099	stable
NVIDIA/DreamDojo NVIDIA's official PyTorch codebase for DreamDojo, a generalist robot world model pretrained on 44k hours of human egocentric video and post…	45	1099	active
lizhe00/AnimatableGaussians Official PyTorch implementation of the CVPR 2024 paper 'Animatable Gaussians', which learns pose-dependent Gaussian maps for high-fidelity …	27	1099	active
zai-org/CogView4 CogView4, CogView3-Plus and CogView3 are open-source text-to-image generation models from Zhipu AI, with CogView4 being a 6B-parameter DiT-…	25	1099	active
lucidrains/stylegan2-pytorch A simple PyTorch implementation of StyleGAN2, a state-of-the-art generative adversarial network, trainable entirely from the command line w…	23	3779	maintenance
Toni-SM/skrl skrl is an open-source modular Reinforcement Learning library written in Python, implemented in PyTorch, JAX, and NVIDIA Warp. It supports …	85	1098	active
qiskit-community/qiskit-machine-learning Qiskit Machine Learning is an open-source Python library built on the Qiskit SDK that provides building blocks for quantum machine learning…	84	1098	active
welltop-cn/ComfyUI-TeaCache A ComfyUI plugin integrating TeaCache, a training-free caching method that accelerates diffusion model inference by exploiting output diffe…	32	1097	active
microsoft/TimeCraft TimeCraft is a diffusion model-based framework for generating high-quality synthetic time series data across domains. It uses learned seman…	66	1096	active
minghanqin/LangSplat Official implementation of LangSplat, a CVPR 2024 Highlight paper that constructs a 3D language field using 3D Gaussian Splatting with CLIP…	44	1096	active
Eyeline-Labs/Go-with-the-Flow Official implementation of the CVPR 2025 Oral paper 'Go-with-the-Flow', which controls motion in video diffusion models by replacing i.i.d.…	39	1096	active
YOLOs-CPP YOLOs-CPP is a production-ready, cross-platform C++ inference library for the YOLO model family (v5 through YOLO26), built on ONNX Runtime …	82	1095	active
LucasAlegre/sumo-rl SUMO-RL is a Python library that provides Gymnasium and PettingZoo-compatible reinforcement learning environments for traffic signal contro…	51	1095	active
tianxingleo/ACGTI ACGTI is an MBTI-inspired web quiz that maps users' personality dimensions to anime character archetypes. It is a Vue 3 + TypeScript full-s…	69	1093	active
Soul-AILab/SoulX-FlashHead SoulX-FlashHead is a 1.3B-parameter framework for high-fidelity, infinite-length, real-time streaming talking-head portrait video generatio…	50	1093	active
yandex/YaLM-100B YaLM-100B is a GPT-like pretrained language model with 100 billion parameters, trained by Yandex on English and Russian text using DeepSpee…	32	3756	maintenance
RUC-NLPIR/Arbor Arbor is a generalist autonomous research agent that takes a benchmark and a goal, then proposes hypotheses, edits code, and runs real expe…	74	1091	active
NVlabs/Fast-dLLM NVIDIA's official implementation of Fast-dLLM, a family of training-free and fine-tuning-based acceleration techniques for diffusion-based …	54	1090	active
mittagessen/kraken kraken is a turn-key OCR/HTR engine built on neural networks, optimized for historical and non-Latin script material. It provides trainable…	96	1089	active
zju3dv/InfiniDepth InfiniDepth is a CVPR 2026 research library for monocular depth estimation that represents depth as neural implicit fields, allowing depth …	59	1089	active
HswAI2026/JuZhou-V1 JuZhou 1.0 is an ultra-lightweight 0.387B-parameter text-to-image foundation model designed for fully offline, on-device execution on mobil…	51	1089	active
latorc/MahjongCopilot A desktop AI assistant application for the Majsoul online mahjong game, built on the Mortal model via the MJAI protocol. It provides real-t…	18	1089	active
mymusise/ChatGLM-Tuning A Python toolkit for fine-tuning the ChatGLM-6B large language model using LoRA (Low-Rank Adaptation) with the Alpaca dataset. It provides …	10	3739	maintenance
rhymes-ai/Aria Aria is an open multimodal native Mixture-of-Experts (MoE) model with 25.3B total parameters (3.9B activated per token) and a 64K multimoda…	23	1088	active
agi-brain/xuance XuanCe is an open-source Python library of deep reinforcement learning (DRL) and multi-agent reinforcement learning (MARL) algorithm implem…	91	1087	active
alexfrom0815/Online-3D-BPP-PCT A Python research implementation of the ICLR 2022 paper 'Learning Efficient Online 3D Bin Packing on Packing Configuration Trees', using de…	57	1087	active
AlohaMini AlohaMini is an open-source, 3D-printable dual-arm mobile robot with a motorized lift, buildable for under $1,000. It is LeRobot-ready for …	54	1087	active
k2-fsa/ZipVoice ZipVoice is a series of fast, high-quality zero-shot text-to-speech models based on flow matching, with a compact 123M-parameter Zipformer-…	40	1087	active
yerfor/Real3DPortrait Official PyTorch implementation of Real3D-Portrait, an ICLR 2024 Spotlight paper for one-shot realistic 3D talking portrait synthesis. It g…	26	1087	active
mlc-ai/web-stable-diffusion A project that compiles and runs Stable Diffusion text-to-image models entirely inside web browsers using WebGPU and WebAssembly, with no s…	31	3729	maintenance
PointNeXt PointNeXt is the official PyTorch implementation of the NeurIPS'22 paper that improves PointNet++ via better training and model scaling str…	86	1086	stable
microsoft/Biodiversity Microsoft AI for Good Lab's biodiversity research hub providing open-source AI models and tools for wildlife monitoring and conservation, i…	86	1086	active
DSE-MSU/DeepRobust DeepRobust is a PyTorch library for adversarial robustness research, providing implementations of attack and defense methods for both image…	41	1086	active
GraphSAGE Reference implementation of the GraphSAGE algorithm for inductive representation learning on large graphs using stochastic graph convolutio…	32	3728	maintenance
LujiaJin/One-Pot_Multi-Frame_Denoising Official PyTorch implementation of the One-Pot Multi-frame Denoising (OPD) method published at BMVC 2022 and extended in IJCV. It provides …	57	1084	stable
Fannovel16/ComfyUI-Frame-Interpolation A set of custom nodes for ComfyUI that perform video frame interpolation using models like RIFE, FILM, GMFSS, and others. It lets users gen…	49	1084	active
Alpha-VLLM/Lumina-mGPT-2.0 Lumina-mGPT 2.0 is a stand-alone decoder-only autoregressive model trained from scratch that unifies a broad range of image generation task…	39	1084	active
QwenLM/Qwen2.5-Math Qwen2.5-Math is a series of math-specialized large language models (1.5B/7B/72B base and instruct variants plus a 72B reward model) built o…	24	1084	active
kimiyoung/transformer-xl Official implementation of Transformer-XL, an attention-based language model architecture that extends context beyond a fixed length via se…	32	3717	maintenance
bytedance/byteps BytePS is a high-performance parameter server framework for distributed deep neural network training, supporting TensorFlow, Keras, PyTorch…	10	3717	maintenance
XiaomiMiMo/MiMo-Audio Xiaomi's open-source 7B audio language model family (Base and Instruct) plus a 1.2B RVQ audio tokenizer, pretrained on 100M+ hours of audio…	53	1083	active
GAIR-NLP/LIMO LIMO is a research project and training framework demonstrating that large language models can achieve strong mathematical reasoning with o…	33	1083	active