Open-source deep-learning software

2785 open-source libraries, tools and applications tagged deep-learning, ranked by how closely deep-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.

2785 products, primary matches first, then adoption-weighted; health v2 shown.

Product	Health v2	Stars	Maturity
tensorflow/rust TensorFlow Rust provides idiomatic Rust language bindings for TensorFlow via its C API. It lets Rust programs build and run TensorFlow comp…	10	5475	active
karpathy/minGPT A minimal, clean PyTorch re-implementation of OpenAI's GPT covering both training and inference in roughly 300 lines of code. It is designe…	32	24956	maintenance
apple/coremltools Apple's official Python package for converting machine learning models from TensorFlow, PyTorch, scikit-learn, XGBoost, and LibSVM into the…	72	5436	active
facebookresearch/sapiens Sapiens is a family of foundation models from Meta Reality Labs for human-centric vision tasks including 2D pose estimation, body-part segm…	58	5432	active
isl-org/MiDaS MiDaS is a Python library with pretrained models for robust monocular depth estimation from a single image, based on the TPAMI 2022 paper a…	10	5419	stable
NVIDIA/Model-Optimizer NVIDIA Model Optimizer (ModelOpt) is a Python library of state-of-the-art model optimization techniques including quantization, pruning, di…	92	5386	active
InternLM/xtuner XTuner is an open-source LLM training engine from InternLM designed for fine-tuning ultra-large-scale Mixture-of-Experts (MoE) models, with…	68	5211	active
lightvector/KataGo KataGo is an open-source Go (baduk) engine trained via AlphaZero-like self-play, one of the strongest Go bots available. It runs as a GTP e…	96	5204	active
Transformer Lab Transformer Lab is an open-source desktop application (built with Electron and Python) that provides a unified GUI for training, fine-tunin…	82	5195	active
h2oai/h2o-llmstudio H2O LLM Studio is a framework and no-code GUI for fine-tuning state-of-the-art large language models, built by H2O.ai. It supports LoRA and…	96	5184	active
NVIDIAGameWorks/kaolin Kaolin is NVIDIA's PyTorch library of GPU-optimized modules for 3D deep learning research, covering meshes, point clouds, and 3D Gaussian s…	66	5175	active
NCCL NVIDIA's Collective Communication Library (NCCL) is a C++ library providing topology-aware, high-bandwidth inter-GPU communication primitiv…	99	5165	stable
open-mmlab/mmaction2 MMAction2 is OpenMMLab's PyTorch-based toolbox and benchmark for video understanding, covering action recognition, temporal action localiza…	52	5163	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
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
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
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
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
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
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
microsoft/muzic Muzic is a Microsoft Research project providing deep learning models for music understanding and generation, including MusicBERT, SongMASS,…	62	4960	active
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
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
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
cvg/LightGlue LightGlue is a deep neural network library that matches sparse local features across image pairs with high accuracy and fast inference. It …	47	4820	stable
facebookresearch/flow_matching A PyTorch library for implementing flow matching algorithms with continuous, discrete, and Riemannian flow matching implementations. It acc…	45	4801	active
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
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
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
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
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
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
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
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
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
sensity-ai/dot dot (Deepfake Offensive Toolkit) is a Python tool that generates real-time, controllable deepfakes from a webcam feed and injects them into…	23	4589	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
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
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
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
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
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
zai-org/GLM-4.5 GLM-4.5 (and successors GLM-4.6/4.7) is a family of open-weight Mixture-of-Experts foundation models from Z.ai focused on agentic tasks, re…	44	4424	active
huawei-noah/Efficient-AI-Backbones A collection of efficient neural network backbone architectures (GhostNet, TNT, ViG, WaveMLP, TinyNet, etc.) from Huawei Noah's Ark Lab, wi…	24	4423	active
onnxsim/onnxsim ONNX Simplifier (onnxsim) is a tool that simplifies ONNX models by inferring the whole computation graph and replacing redundant operators …	100	4415	active
MedSAM MedSAM is a fine-tuned Segment Anything Model (SAM) foundation model for universal medical image segmentation, trained on over 1.5 million …	26	4411	active
HIP HIP is a C++ runtime API and kernel programming language for AMD GPUs that mirrors the NVIDIA CUDA programming interface. It enables develo…	90	4406	active
iperov/DeepFaceLab DeepFaceLab is the leading open-source Windows application for creating deepfakes, allowing users to swap, de-age, or replace faces and hea…	10	19274	maintenance
AI4Finance-Foundation/ElegantRL ElegantRL is a lightweight, modular deep reinforcement learning library built on PyTorch that implements core model-free RL algorithms (PPO…	50	4368	active
ModelTC/LightLLM LightLLM is a Python-based LLM inference and serving framework designed for lightweight deployment, easy scalability, and high throughput. …	82	4317	active
Tencent-Hunyuan/HunyuanDiT Hunyuan-DiT is Tencent's open-source diffusion transformer model for text-to-image generation with fine-grained Chinese language understand…	46	4288	active
richzhang/PerceptualSimilarity A PyTorch library implementing the LPIPS (Learned Perceptual Image Patch Similarity) metric, which measures perceptual distance between ima…	23	4280	stable
Nixtla/neuralforecast NeuralForecast is a Python library offering a large collection of state-of-the-art neural forecasting models (NBEATS, NHITS, TFT, PatchTST,…	97	4277	active
ThilinaRajapakse/simpletransformers Simple Transformers is a Python library built on Hugging Face Transformers that lets users train, fine-tune, and evaluate Transformer model…	57	4256	active
ArcInstitute/evo2 Evo 2 is a DNA foundation language model (1B-40B parameters) that models genomes at single-nucleotide resolution with up to 1 million base …	56	4247	active
SwanHubX/SwanLab SwanLab is an open-source AI training tracking and visualization platform with a Python SDK, CLI, and cloud or self-hosted dashboard. It in…	88	4237	active
thinking-machines-lab/tinker-cookbook Tinker Cookbook is a Python library of realistic examples and abstractions for post-training (fine-tuning) language models via the Tinker A…	84	4199	active
torchgeo/torchgeo TorchGeo is a PyTorch domain library, similar to torchvision, providing datasets, samplers, transforms, and pre-trained models specific to …	93	4192	active
JEPA Official PyTorch implementation of V-JEPA, a self-supervised method for learning visual representations from video using a joint-embedding …	26	4179	active
SamGeo SamGeo (segment-geospatial) is a Python package that applies Meta AI's Segment Anything Model (SAM, SAM2, SAM3, HQ-SAM) to geospatial data …	94	4147	active
IDEA-CCNL/Fengshenbang-LM Fengshenbang-LM is an open-source suite of Chinese large language models and a PyTorch training framework from IDEA Research's CCNL lab, ai…	67	4123	active
openai/transformer-debugger Transformer Debugger (TDB) is an OpenAI tool for investigating specific behaviors of small language models, combining automated interpretab…	57	4123	active
GuyTevet/motion-diffusion-model Official PyTorch implementation of the Human Motion Diffusion Model (MDM) paper, generating 3D human motion sequences from text prompts usi…	48	4119	active
princeton-vl/RAFT Official PyTorch implementation of RAFT (Recurrent All Pairs Field Transforms for Optical Flow), an ECCV 2020 model for estimating dense op…	46	4103	stable
Tencent-Hunyuan/Hunyuan3D-2.1 Tencent's open-source 3D asset generation model that creates high-fidelity 3D meshes with production-ready PBR materials from single images…	37	4096	active
PaddlePaddle/PaddleRec PaddleRec is a large-scale recommendation algorithm library built on PaddlePaddle, containing classic and state-of-the-art recommendation m…	26	4086	active
QwenLM/Qwen2.5-Omni Qwen2.5-Omni is an end-to-end multimodal model from Alibaba's Qwen team that understands text, images, audio, and video, and generates stre…	29	4075	active
DLRM A PyTorch-based implementation of the Deep Learning Recommendation Model (DLRM) from Facebook Research, which processes dense and sparse fe…	57	4074	stable
FedML FedML (TensorOpera) is a unified Python library for large-scale distributed training, model serving, and federated learning across GPU clou…	41	4070	active
uxlfoundation/oneDNN oneDNN is an open-source cross-platform performance library providing optimized building blocks (primitives) for deep learning applications…	99	4067	stable
tensorflow/tensor2tensor Tensor2Tensor (T2T) is a Python library of deep learning models and datasets built on TensorFlow, developed by the Google Brain team to mak…	10	17473	maintenance
lucidrains/vector-quantize-pytorch A PyTorch library implementing vector and scalar quantization, including Residual VQ and techniques like DiVeQ codebook updates. It origina…	81	4019	active
GVCLab/PersonaLive PersonaLive is a diffusion-based framework for real-time, streamable portrait image animation, generating infinite-length expressive talkin…	58	3995	active
Marin Marin is an open-source Python framework and research program for training foundation models, covering the full pipeline from data curation…	79	3988	active
LLaDA Official PyTorch implementation of LLaDA, a family of large language diffusion models (8B base/instruct, MoE, and iLLaDA variants) with pre…	59	3986	active
IREE IREE is an MLIR-based end-to-end machine learning compiler and runtime that lowers models from frameworks like PyTorch, TensorFlow, JAX, an…	95	3980	active
hustvl/4DGaussians An official PyTorch implementation of 4D Gaussian Splatting (4D-GS) for real-time rendering of dynamic scenes, published at CVPR 2024. It c…	28	3978	active
ali-vilab/VACE VACE is the official implementation of an all-in-one video creation and editing model from Tongyi Lab, built on Wan2.1 diffusion models. It…	38	3962	active
Lightning-AI/LitServe LitServe is a Python framework for building custom AI inference servers with full control over batching, routing, streaming, and scaling lo…	92	3956	active
NeuralOperator A PyTorch library for learning neural operators, which map between function spaces rather than finite-dimensional vectors. It provides the …	81	3955	active
Nunchaku Nunchaku is a high-performance inference engine for 4-bit quantized diffusion models (and LLMs) based on the SVDQuant technique from an ICL…	73	3953	active
TransformerLensOrg/TransformerLens TransformerLens is a Python library for mechanistic interpretability of GPT-style transformer language models. It lets researchers load tho…	98	3949	active
stanford-futuredata/ColBERT ColBERT is a fast and accurate neural retrieval model that encodes passages and queries into token-level embedding matrices and scores them…	40	3941	active