Open-source deep-learning software

2663 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.

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

Product	Health v2	Stars	Maturity
tensorflow/tensorflow TensorFlow is an open‑source end‑to‑end platform for building and deploying machine‑learning models. It provides stable Python and C++ APIs…	83	200740	active
Transformers 🤗 Transformers is a Python library that provides a unified API for thousands of pretrained models across text, vision, audio, and multimoda…	95	167798	active
Transformers Hugging Face Transformers is a Python library that serves as the model-definition framework for state-of-the-art machine learning models ac…	95	167798	stable
openai/whisper OpenAI's Whisper is a general-purpose speech recognition model and Python library built on a Transformer sequence-to-sequence architecture.…	64	109525	stable
pytorch/pytorch PyTorch is an open-source Python library for tensor computation and deep neural networks with strong GPU acceleration. It provides a dynami…	95	104835	active
deepseek-ai/DeepSeek-R1 DeepSeek-R1 is a family of open-weight large language models trained with large-scale reinforcement learning for reasoning, including DeepS…	21	91900	active
tensorflow/models The TensorFlow Model Garden is a repository of official and community implementations of state-of-the-art machine learning models built wit…	82	77653	active
Keras Keras is a high‑level, multi‑backend deep learning library for Python. It provides a user‑friendly API that works with TensorFlow, JAX, PyT…	96	64373	active
Ultralytics YOLO Ultralytics YOLO is a Python package and CLI providing a family of real-time computer vision models (YOLO26, YOLO11, YOLOv8) for object det…	96	62597	active
Faceswap FaceSwap is an open‑source Python application that uses deep learning to detect and swap faces in images and videos. It provides both a gra…	78	57578	active
Faceswap Faceswap is a free, open-source, multi-platform deepfakes tool that uses deep learning to recognize and swap faces in pictures and videos. …	78	57578	active
face_recognition face_recognition is a Python library that provides simple APIs for detecting, locating, and recognizing human faces in images. It also ship…	59	56796	active
Qlib Qlib is an AI-oriented quantitative investment platform from Microsoft that supports the full quant research workflow, from data processing…	66	48694	active
google-research/bert Google's BERT repository provides TensorFlow implementations and pre‑trained checkpoints of the Bidirectional Encoder Representations from …	10	40038	active
jax-ml/jax JAX is a Python library that provides composable transformations such as automatic differentiation, JIT compilation, and vectorization for …	98	36460	active
XingangPan/DragGAN Official PyTorch implementation of DragGAN (SIGGRAPH 2023), an interactive point-based image manipulation method built on StyleGAN3. Users …	30	35748	stable
huggingface/diffusers Hugging Face Diffusers is a Python library providing state-of-the-art pretrained diffusion models for generating images, videos, and audio …	95	34736	stable
OpenPose OpenPose is an open‑source C++ library that provides real‑time multi‑person keypoint detection for body, face, hands, and feet. It leverage…	23	34493	active
openai/CLIP OpenAI's CLIP is a PyTorch library providing pretrained contrastive language-image models that encode images and text into a shared embeddi…	62	34362	stable
tinygrad/tinygrad tinygrad is a small, hackable deep learning framework providing a PyTorch-like tensor library with autograd, plus a visible IR and compiler…	88	33740	active
Lightning-AI/pytorch-lightning PyTorch Lightning is a deep learning framework that organizes PyTorch code to eliminate boilerplate for training, finetuning, and scaling A…	93	31417	stable
karpathy/llm.c llm.c is a pure C/CUDA implementation for training large language models like GPT-2 and GPT-3, with no dependency on PyTorch or Python. It …	34	31176	active
JaidedAI/EasyOCR EasyOCR is a ready-to-use Python OCR library built on PyTorch that extracts text from images, supporting 80+ languages and popular writing …	44	30080	stable
hpcaitech/Open-Sora Open-Sora is an open-source initiative for efficiently producing high-quality videos from text prompts using deep learning models. It provi…	47	29872	active
MLX MLX is an array computation framework for machine learning on Apple silicon, developed by Apple ML research. It offers NumPy-like Python AP…	95	28839	active
LeRobot LeRobot is Hugging Face's PyTorch library for real-world robotics, providing a hardware-agnostic Python interface for controlling robots, a…	90	28232	active
fastai fastai is an open‑source deep learning library built on PyTorch that provides high‑level components for rapid model development. It offers …	92	28224	active
fastai fastai is a deep learning library built on PyTorch that offers high-level components for quickly achieving state-of-the-art results in visi…	92	28224	stable
Stability-AI/generative-models Stability AI's official repository of generative diffusion models, including Stable Diffusion, Stable Video, and SV4D 2.0 for image, video,…	42	27300	active
Apollo Apollo is an open-source autonomous driving platform written in C++. It provides a modular architecture for perception, planning, control, …	54	26853	active
Apollo Apollo is an open-source autonomous driving platform providing a high-performance, modular software stack for developing, testing, and depl…	54	26853	active
lucidrains/vit-pytorch A PyTorch library implementing the Vision Transformer (ViT) and dozens of ViT variants (NaViT, MaxViT, MobileViT, Dino, masked autoencoders…	87	25551	active
junyanz/pytorch-CycleGAN-and-pix2pix Official PyTorch implementations of CycleGAN and pix2pix for paired and unpaired image-to-image translation. It includes training and testi…	44	25261	stable
Dao-AILab/flash-attention Official implementation of FlashAttention, FlashAttention-2, -3, and -4: fast and memory-efficient exact attention kernels for GPUs. It acc…	92	25116	active
PaddlePaddle PaddlePaddle is an open‑source deep learning platform that provides a unified framework for model development, training, and inference. It …	92	24105	active
PaddlePaddle PaddlePaddle is an industrial-grade deep learning framework written in C++ with Python APIs, supporting high-performance single-machine and…	92	24105	active
PyTorch Geometric PyTorch Geometric (PyG) is a Python library built on top of PyTorch for building and training Graph Neural Networks. It provides a unified …	78	24098	active
PyTorch Geometric PyG (PyTorch Geometric) is a Python library built on PyTorch for writing and training Graph Neural Networks on structured and irregular dat…	78	24098	stable
pytorch/examples A curated collection of short, high‑quality PyTorch example projects covering vision, text, reinforcement learning, and more. The repositor…	46	24052	active
facebookresearch/audiocraft A PyTorch library from Meta for audio processing and generation with deep learning, featuring the EnCodec neural audio codec and generative…	58	23699	active
DeepFace DeepFace is a lightweight Python library for face recognition and facial attribute analysis, wrapping state-of-the-art models like VGG-Face…	93	23482	stable
Oxc Oxc is a Rust‑based collection of high‑performance JavaScript and TypeScript tools, including a parser, linter, formatter, transformer, min…	95	23109	active
microsoft/unilm Microsoft's collection of large-scale self-supervised pre-trained models spanning tasks, 100+ languages, and modalities (text, image, layou…	66	22219	active
ONNX Runtime ONNX Runtime is a cross-platform, high-performance machine-learning accelerator for running inference and training on ONNX models. It suppo…	100	22115	stable
onnx/onnx ONNX (Open Neural Network Exchange) is an open standard and Python library for representing machine learning and deep learning models as an…	95	21602	stable
huggingface/candle Candle is a minimalist machine learning framework for Rust focused on performance and ease of use, with CPU and CUDA GPU support. It ships …	74	21182	active
triton-lang/triton Triton is an open‑source programming language and compiler for writing high‑performance custom deep‑learning primitives. It targets GPUs an…	93	20386	active
karpathy/llama2.c A minimal, dependency-free implementation of the Llama 2 LLM architecture that trains in PyTorch and runs inference in a single ~700-line p…	29	20192	active
QwenLM/Qwen3-VL Qwen3-VL is a series of open-weight multimodal vision-language models from Alibaba's Qwen team, available in Dense and MoE architectures wi…	49	19998	active
ML-Agents The Unity Machine Learning Agents Toolkit (ML-Agents) lets Unity games and simulations serve as environments for training intelligent agent…	66	19717	active
WebLLM WebLLM is a high‑performance in‑browser LLM inference engine written in TypeScript. It runs large language models directly in the browser u…	90	19160	active
TensorFlow.js TensorFlow.js is a hardware‑accelerated JavaScript library for building, training, and running machine‑learning models in the browser and N…	59	19147	active
TensorFlow.js TensorFlow.js is a hardware-accelerated JavaScript library for training and deploying machine learning models in the browser and Node.js. I…	59	19147	stable
state-spaces/mamba Mamba is a PyTorch library implementing the selective state space model (SSM) architecture for linear-time sequence modeling, with hardware…	80	18874	active
NVIDIA-NeMo/Speech NVIDIA NeMo Speech is an open‑source Python toolkit for building, training, and deploying speech AI models such as automatic speech recogni…	95	18521	active
NVIDIA/Megatron-LM NVIDIA's GPU-optimized library for training large transformer models at scale, comprising Megatron-LM (reference training scripts) and Mega…	100	18178	active
sczhou/CodeFormer CodeFormer is a PyTorch-based blind face restoration model using a codebook lookup transformer, published at NeurIPS 2022. It restores and …	43	18167	stable
karpathy/micrograd A tiny scalar-valued autograd engine implementing reverse-mode automatic differentiation over a dynamically built DAG, with a small PyTorch…	71	17953	stable
deepseek-ai/Janus Janus-Series is DeepSeek's family of unified multimodal models (Janus, Janus-Pro, JanusFlow) that combine multimodal understanding and imag…	23	17755	active
Wan Wan is an open-source family of large-scale video generation models (Wan2.1/Wan2.2) with Python inference code and pretrained weights for t…	62	17695	active
lllyasviel/FramePack FramePack is a next-frame-prediction video diffusion framework and desktop application that generates long videos progressively while compr…	28	17304	active
Robbyant/lingbot-map LingBot-Map is a feed-forward 3D foundation model that reconstructs scenes from streaming image data using a Geometric Context Transformer.…	57	17135	active
KindXiaoming/pykan Pykan is a Python library implementing Kolmogorov-Arnold Networks (KANs), an alternative to multi-layer perceptrons with learnable activati…	16	16331	active
MNN MNN is a lightweight, high-performance deep learning inference engine developed by Alibaba, supporting LLMs, vision, audio, and multimodal …	90	16188	active
microsoft/Swin-Transformer Official PyTorch implementation of the Swin Transformer, a hierarchical vision transformer using shifted windows that serves as a general-p…	32	16081	stable
tracel-ai/burn Burn is a Rust-based tensor library and deep learning framework supporting training and inference through a unified API. It JIT-compiles te…	86	16008	active
GeeeekExplorer/nano-vllm A lightweight vLLM-style LLM inference engine implemented from scratch in about 1,200 lines of Python. It offers fast offline inference wit…	51	15834	active
SWivid/F5-TTS F5-TTS is the official implementation of a fully non-autoregressive text-to-speech system based on flow matching with a Diffusion Transform…	84	15313	active
Tencent-Hunyuan/Hunyuan3D-2 Tencent's open-source Hunyuan3D-2 system for generating high-resolution 3D assets using large-scale diffusion models, supporting image-to-3…	40	14994	active
BrainJS/brain.js brain.js is a GPU-accelerated neural network library written in JavaScript/TypeScript that runs in both browsers and Node.js. It supports f…	23	14860	active
HumanAIGC/AnimateAnyone Animate Anyone is the official research implementation of a diffusion-based image-to-video synthesis method that animates a static characte…	43	14777	active
RWKV RWKV is a novel language model architecture that combines RNN efficiency (linear time, constant memory, no KV-cache) with transformer-level…	66	14744	active
facebookresearch/vggt VGGT (Visual Geometry Grounded Transformer) is a feed-forward transformer model from Meta AI and Oxford VGG that infers 3D geometry—camera …	55	14499	active
PaddlePaddle/PaddleDetection PaddleDetection is an object detection toolkit built on the PaddlePaddle deep learning framework. It provides implementations of detection,…	67	14431	active
Flair Flair is a Python NLP framework built on PyTorch for state-of-the-art natural language processing tasks such as named entity recognition, p…	48	14392	active
dmlc/dgl DGL (Deep Graph Library) is a Python package for deep learning on graph-structured data, built on top of existing frameworks like PyTorch, …	34	14281	stable
Eclipse Deeplearning4j Eclipse Deeplearning4J is an open-source deep learning framework and ecosystem for the JVM, including the ND4J linear algebra library, the …	76	14269	active
mlfoundations/open_clip OpenCLIP is an open-source PyTorch implementation of CLIP and related multimodal contrastive models, with many pretrained image/text checkp…	80	14157	active
Physical-Intelligence/openpi Open-source repository from Physical Intelligence containing vision-language-action (VLA) models for robotics, including π₀, π₀-FAST, and π…	64	13991	active
CompVis/stable-diffusion The original reference implementation of Stable Diffusion, a latent text-to-image diffusion model trained on LAION-5B data with a CLIP text…	32	73487	maintenance
microsoft/TRELLIS TRELLIS is Microsoft's large-scale 3D asset generation model that creates high-quality 3D assets from text or image prompts. It uses a unif…	58	13679	active
modelscope/DiffSynth-Studio DiffSynth-Studio is an open-source diffusion model engine from the ModelScope community that integrates mainstream image, video, and audio …	77	13229	active
ACE-Step ACE-Step 1.5 is an open-source music generation foundation model combining a language model planner with a Diffusion Transformer to create …	77	13070	active
zai-org/CogVideo CogVideo/CogVideoX is an open-source family of text-to-video and image-to-video generation models from Zhipu AI (THUDM), with inference and…	42	13055	active
jacobgil/pytorch-grad-cam A PyTorch library providing state-of-the-art pixel attribution (saliency) methods like GradCAM, ScoreCAM, and AblationCAM for explainable A…	72	13002	active
PaddlePaddle/PaddleFormers PaddleFormers is a Transformers-style library built on PaddlePaddle providing a model zoo of 100+ large language models and vision-language…	88	12983	active
Vision Transformer Google Research's official JAX/Flax implementation of Vision Transformer (ViT) and MLP-Mixer architectures, with released pretrained checkp…	76	12744	stable
sapientinc/HRM Official PyTorch implementation of the Hierarchical Reasoning Model (HRM), a 27M-parameter recurrent architecture with high-level and low-l…	49	12642	active
axolotl-ai-cloud/axolotl Axolotl is a free, open-source, config-driven framework for fine-tuning large language models, supporting SFT, preference learning (DPO/KTO…	98	12591	active
Tencent-Hunyuan/HunyuanVideo HunyuanVideo is Tencent's open-source framework for large-scale video generation, providing PyTorch model definitions, pre-trained weights,…	59	12577	active
guoyww/AnimateDiff Official implementation of AnimateDiff, a plug-and-play motion modeling module that turns personalized text-to-image diffusion models (e.g.…	30	12274	active
PKU-YuanGroup/Open-Sora-Plan Open-Sora Plan is an open-source effort to reproduce OpenAI's Sora text-to-video model, providing training and inference code for video gen…	45	12209	active
xmu-xiaoma666/External-Attention-pytorch A PyTorch library (fightingcv-attention) providing clean, minimal implementations of numerous attention mechanisms, MLP variants, re-parame…	61	12186	active
Tongyi-MAI/Z-Image Z-Image is a 6B-parameter text-to-image generation foundation model family built on a single-stream diffusion transformer, with a distilled…	43	12057	active
nerfstudio Nerfstudio is a Python library and CLI toolkit providing a simple, modular API for creating, training, and testing Neural Radiance Fields (…	34	12019	active
InstantID InstantID is a tuning-free, zero-shot identity-preserving image generation method built on diffusion models, generating customized images i…	27	11994	active
NVIDIA/cosmos NVIDIA Cosmos is an open platform of omnimodal world foundation models, datasets, and tools for building Physical AI systems such as robots…	69	11892	active
ludwig-ai/ludwig Ludwig is a declarative, low-code deep learning framework for training, fine-tuning, and deploying AI models — from LLMs to tabular, image,…	96	11775	active
qubvel-org/segmentation_models.pytorch A PyTorch library providing neural networks for image semantic segmentation with a simple high-level API. It includes 12 encoder-decoder ar…	68	11745	stable
milesial/Pytorch-UNet A PyTorch implementation of the U-Net architecture for semantic segmentation of high-resolution images, originally built for Kaggle's Carva…	23	11668	active