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/lucid Lucid is a collection of infrastructure and tools for research in neural network interpretability, built on TensorFlow 1.x. It provides fea… | 10 | 4704 | maintenance |
| sjtuytc/UnboundedNeRFPytorch A PyTorch implementation benchmarking state-of-the-art unbounded (large-scale) neural radiance field methods like NeRF++, DVGO, and Block-N… | 23 | 1323 | active |
| yeyupiaoling/VoiceprintRecognition-Pytorch A PyTorch-based voiceprint recognition (speaker recognition) framework implementing models such as ECAPA-TDNN, ResNetSE, ERes2Net, and CAM+… | 75 | 1322 | active |
| ImprintLab/Medical-SAM-Adapter Medical SAM Adapter (MSA) is a Python framework that fine-tunes Meta's Segment Anything Model for medical image segmentation using lightwei… | 35 | 1322 | active |
| meta-pytorch/segment-anything-fast A fast, batched offline inference-oriented fork of Meta's Segment Anything (SAM) image segmentation model. It applies optimizations like bf… | 42 | 1321 | active |
| stared/livelossplot A Python library that draws live training loss and metric plots inside Jupyter Notebooks for Keras, PyTorch, and other deep learning framew… | 85 | 1320 | stable |
| sanchit-gandhi/whisper-jax An optimized JAX implementation of OpenAI's Whisper speech recognition model, built on Hugging Face Transformers, offering up to 70x faster… | 31 | 4680 | maintenance |
| ChenmienTan/RL2 RL2 (Ray Less Reinforcement Learning) is a concise Python library for post-training large language models with reinforcement learning, SFT,… | 54 | 1316 | active |
| sicara/easy-few-shot-learning A Python library (easyfsl) with ready-to-use code and tutorial notebooks for few-shot image classification and meta-learning, built on PyTo… | 23 | 1316 | stable |
| albermax/innvestigate iNNvestigate is a Python toolbox providing a common interface and out-of-the-box implementations of many neural network explanation methods… | 26 | 1307 | active |
| Vincentqyw/image-matching-webui A Gradio-based web UI that matches keypoints between two images using many state-of-the-art image matching algorithms (LoFTR, SuperGlue, Li… | 85 | 1306 | active |
| huawei-noah/Efficient-Computing A collection of efficient deep learning methods from Huawei Noah's Ark Lab, covering model compression, knowledge distillation, pruning, qu… | 32 | 1306 | active |
| fundamentalvision/BEVFormer BEVFormer is the official PyTorch implementation of an ECCV 2022 paper that learns bird's-eye-view (BEV) representations from multi-camera … | 23 | 4624 | maintenance |
| luosiallen/latent-consistency-model Official implementation of Latent Consistency Models (LCM), a diffusion-based approach for synthesizing high-resolution images with few-ste… | 28 | 4607 | maintenance |
| buoyancy99/diffusion-forcing Official research code for the NeurIPS paper 'Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion', implementing a metho… | 61 | 1299 | active |
| STVIR/pysot PySOT is a Python research platform by SenseTime for single object visual tracking, implementing algorithms such as SiamRPN, SiamRPN++, DaS… | 41 | 4602 | maintenance |
| W2GenAI-Lab/LucidFlux LucidFlux is a caption-free photo-realistic image restoration model built on a large-scale diffusion transformer, released with inference a… | 52 | 1298 | active |
| Xiangyue-Zhang/auto-deep-researcher-24x7 An open-source Python framework where LLM agents autonomously run deep learning experiments 24/7, covering hypothesis formation, code imple… | 50 | 1293 | active |
| plemeri/transparent-background A Python tool and CLI that removes backgrounds from images and videos using the InSPyReNet deep learning model (ACCV 2022). It supports ima… | 60 | 1292 | active |
| locuslab/TCN PyTorch implementation of Temporal Convolutional Networks (TCN) with benchmarks from the paper 'An Empirical Evaluation of Generic Convolut… | 32 | 4569 | maintenance |
| plaidml/plaidml PlaidML is a portable tensor compiler that enables deep learning on hardware (especially GPUs and embedded devices) not well supported by m… | 10 | 4566 | maintenance |
| RoyalVane/CLAN Official PyTorch implementation of CLAN, a CVPR 2019 (oral) / TPAMI 2022 method for unsupervised domain adaptation in semantic segmentation… | 32 | 1289 | stable |
| PrunaAI/pruna Pruna is an open-source Python model optimization framework that makes AI models faster, smaller, cheaper, and greener via caching, quantiz… | 80 | 1288 | active |
| e3nn/e3nn e3nn is a Python/PyTorch library for building E(3)-equivariant neural networks that respect 3D rotation, translation, and mirror symmetries… | 64 | 1288 | active |
| vipshop/cache-dit Cache-DiT is a PyTorch-native inference engine that accelerates Diffusion Transformer (DiT) models with hybrid caching, parallelism, quanti… | 79 | 1287 | active |
| Phantom-video/HuMo HuMo is a research model and Python codebase from Tsinghua University and ByteDance for human-centric video generation using collaborative … | 43 | 1286 | active |
| dcharatan/pixelsplat pixelSplat is a PyTorch implementation of a feed-forward model that reconstructs 3D radiance fields parameterized by 3D Gaussian primitives… | 27 | 1282 | stable |
| DreamLM/Dream Dream 7B is an open diffusion large language model (dLLM) with base and instruct checkpoints, plus inference and training code built on Hug… | 40 | 1279 | active |
| Tianxiaomo/pytorch-YOLOv4 A minimal PyTorch implementation of YOLOv4 (and YOLOv4-tiny) supporting inference and training, with tools to convert Darknet weights to Py… | 32 | 4520 | maintenance |
| AaronJackson/vrn Research code for the ICCV 2017 paper 'Large Pose 3D Face Reconstruction from a Single Image via Direct Volumetric CNN Regression'. It uses… | 32 | 4518 | maintenance |
| Stable-X/Stable3DGen Stable3DGen is a modular Python framework for generating 3D assets from images, adapted from Microsoft's TRELLIS with NVIDIA library depend… | 30 | 1278 | active |
| Tencent-Hunyuan/SRPO SRPO is Tencent Hunyuan's research code for fine-tuning diffusion image generation models (e.g., FLUX.1.dev) by aligning the full diffusion… | 51 | 1276 | active |
| nianticlabs/monodepth2 Monodepth2 is the reference PyTorch implementation of the ICCV 2019 paper 'Digging into Self-Supervised Monocular Depth Prediction'. It tra… | 32 | 4504 | maintenance |
| google-research/simclr Google Research's official implementation of SimCLR and SimCLRv2, a framework for contrastive learning of visual representations, with 65 p… | 10 | 4503 | maintenance |
| ModelCloud/GPTQModel GPTQModel is a production-ready Python toolkit for quantizing (compressing) large language models using GPTQ, AWQ, and related methods, wit… | 92 | 1274 | active |
| Ma-Lab-Berkeley/CRATE CRATE is the official PyTorch implementation of the Coding RAte reduction TransformEr, a family of 'white-box' transformer architectures de… | 30 | 1274 | active |
| luo3300612/Visualizer A lightweight Python library that extracts attention maps and other local variables from deep inside PyTorch models for visualization. It w… | 32 | 1272 | stable |
| leoxiaobin/deep-high-resolution-net.pytorch Official PyTorch implementation of HRNet (Deep High-Resolution Representation Learning for Human Pose Estimation, CVPR 2019), which maintai… | 32 | 4478 | maintenance |
| amaiya/ktrain ktrain is a lightweight Python wrapper around TensorFlow Keras that provides pre-canned, low-code models for text, vision, graph, and tabul… | 23 | 1268 | active |
| meta-pytorch/data TorchData is a PyTorch library providing scalable, performant data loading utilities, including StatefulDataLoader, a drop-in replacement f… | 73 | 1267 | active |
| Bjarten/early-stopping-pytorch A small PyTorch utility package providing an EarlyStopping class that monitors validation loss during training and stops when it stops impr… | 34 | 1267 | stable |
| lucidrains/flamingo-pytorch A PyTorch implementation of DeepMind's Flamingo visual language model architecture, providing the Perceiver Resampler and Gated Cross-Atten… | 23 | 1267 | active |
| ToniRV/NeRF-SLAM NeRF-SLAM is a real-time dense monocular SLAM system that combines neural radiance fields (Instant-NGP) with probabilistic volumetric fusio… | 32 | 1266 | active |
| Deformable 3D Gaussians Official PyTorch implementation of the CVPR 2024 paper 'Deformable 3D Gaussians for High-Fidelity Monocular Dynamic Scene Reconstruction'. … | 19 | 1264 | stable |
| Francis-Rings/StableAvatar StableAvatar is an end-to-end video diffusion transformer that generates infinite-length, high-quality talking avatar videos from a referen… | 62 | 1263 | active |
| meta-pytorch/attention-gym Attention Gym is a collection of tools, examples, and reference implementations for working with PyTorch's FlexAttention API. It provides a… | 91 | 1260 | active |
| Fictionarry/ER-NeRF ER-NeRF is the official PyTorch implementation of an ICCV 2023 paper on region-aware Neural Radiance Fields for high-fidelity talking portr… | 21 | 1259 | stable |
| nv-tlabs/GET3D GET3D is NVIDIA's PyTorch implementation of a generative model that synthesizes high-quality 3D textured meshes (cars, chairs, animals, bui… | 32 | 4433 | maintenance |
| thu-ml/SpargeAttn SpargeAttn is the official implementation of a training-free sparse and quantized attention method that accelerates inference for language,… | 48 | 1257 | active |
| martinpacesa/BindCraft BindCraft is a Python-based computational pipeline for de novo protein binder design that combines AlphaFold2 backpropagation, ProteinMPNN,… | 70 | 1256 | active |
| MoonshotAI/FlashKDA FlashKDA is a set of high-performance CUDA kernels (built on CUTLASS) implementing Kimi Delta Attention, a linear attention mechanism, for … | 56 | 1256 | active |
| dmlc/dlpack DLPack is a minimal, stable open in-memory tensor structure defined as a single C ABI header, enabling zero-copy tensor exchange between de… | 73 | 1253 | stable |
| facebookresearch/llm-transparency-tool An interactive toolkit from Meta Research for analyzing the internal workings of Transformer-based language models. It visualizes contribut… | 10 | 1253 | active |
| LTH14/fractalgen A PyTorch implementation of Fractal Generative Models (FractalGen), enabling pixel-by-pixel high-resolution image generation. It includes p… | 21 | 1247 | active |
| alibaba/Tora Tora is Alibaba's official implementation of a trajectory-oriented Diffusion Transformer (DiT) for controllable video generation, integrati… | 61 | 1243 | active |
| showlab/Tune-A-Video Tune-A-Video is the official PyTorch implementation of an ICCV 2023 paper that fine-tunes pre-trained text-to-image diffusion models (like … | 32 | 4363 | maintenance |
| declare-lab/tango Tango is a family of latent diffusion models for text-to-audio generation, with Tango 2 improving prompt alignment via DPO-based fine-tunin… | 42 | 1241 | active |
| jcjohnson/fast-neural-style A Torch (Lua) implementation of feedforward neural style transfer from the ECCV 2016 paper 'Perceptual Losses for Real-Time Style Transfer … | 32 | 4362 | maintenance |
| chengzeyi/Comfy-WaveSpeed A ComfyUI custom node plugin that acts as an all-in-one inference optimization solution for diffusion models, built around First Block Cach… | 63 | 1238 | stable |
| zai-org/SCAIL-2 Official implementation of SCAIL-2, an open-source model for end-to-end controlled character animation that drives character videos from re… | 55 | 1238 | active |
| facebookresearch/deit Official PyTorch repository for DeiT and related vision transformer architectures (CaiT, ResMLP, PatchConvnet, DeiT III), providing trainin… | 10 | 4348 | maintenance |
| pytorch/serve TorchServe is a flexible, production-ready model server for serving, optimizing, and scaling PyTorch models over HTTP with support for CPU,… | 10 | 4344 | maintenance |
| ElectricAlexis/NotaGen NotaGen is a symbolic music generation model that produces high-quality classical sheet music using LLM-style training paradigms: pre-train… | 29 | 1236 | active |
| open-mmlab/playground OpenMMLab Playground is a central hub collecting and showcasing community projects that extend OpenMMLab libraries with Segment Anything Mo… | 31 | 1235 | active |
| qualcomm/ai-hub-models Qualcomm AI Hub Models is a curated collection of 300+ state-of-the-art machine learning models (vision, audio, speech, generative AI) pre-… | 90 | 1230 | active |
| lucidrains/deep-daze Deep Daze is a simple command line tool for text-to-image generation that combines OpenAI's CLIP with a Siren implicit neural representatio… | 23 | 4316 | maintenance |
| microsoft/MInference MInference is a Microsoft library that accelerates long-context LLM inference using dynamic sparse attention, reducing pre-fill latency by … | 60 | 1229 | active |
| Tencent-Hunyuan/HunyuanCustom HunyuanCustom is a multimodal-driven customized video generation framework built on HunyuanVideo, supporting image, text, audio, and video … | 37 | 1229 | active |
| sgl-project/SpecForge SpecForge is a Python framework from the SGLang team for training speculative decoding models such as EAGLE/EAGLE3 draft heads. Trained mod… | 64 | 1228 | active |
| NVIDIA/BigVGAN BigVGAN is NVIDIA's official PyTorch implementation of a universal neural vocoder (ICLR 2023) that generates high-fidelity raw audio wavefo… | 23 | 1228 | stable |
| datawhalechina/torch-rechub Torch-RecHub is a lightweight PyTorch framework for building recommendation system models with 30+ out-of-the-box algorithms covering ranki… | 96 | 1227 | active |
| zkonduit/ezkl EZKL is a Rust-based library and command-line tool that converts deep learning models and arbitrary computational graphs (exported as ONNX)… | 69 | 1227 | active |
| alibaba/x-deeplearning X-DeepLearning (XDL) is an industrial deep learning framework from Alibaba optimized for high-dimension sparse data scenarios such as adver… | 23 | 4301 | maintenance |
| MotrixLab/SMPLer-X Official code for SMPLer-X, a family of foundation models for expressive human pose and shape estimation (EHPS) that unifies body, hand, an… | 56 | 1226 | stable |
| run-house/kubetorch Kubetorch is a Python library that lets you distribute and run ML workloads (training, inference, data processing) on Kubernetes directly f… | 76 | 1225 | active |
| Project-MONAI/research-contributions A collection of peer-reviewed research prototype implementations built on the MONAI framework for medical imaging AI. It serves as a fast-t… | 39 | 1225 | active |
| lucidrains/perceiver-pytorch A PyTorch implementation of the Perceiver architecture (General Perception with Iterative Attention) and its follow-up Perceiver IO. It pro… | 66 | 1221 | active |
| eduardoleao052/js-pytorch JS-PyTorch is a deep learning library for JavaScript that closely mirrors PyTorch's syntax, providing tensor operations, automatic differen… | 17 | 1221 | active |
| fudan-generative-vision/champ Champ is a research framework for controllable and consistent human image animation using 3D parametric guidance (SMPL-based depth, normal,… | 26 | 4265 | maintenance |
| willisma/SiT Official PyTorch implementation of Scalable Interpolant Transformers (SiT), a family of generative models built on Diffusion Transformers t… | 50 | 1217 | active |
| hasktorch/hasktorch Hasktorch is a Haskell library for tensor math and neural networks, built on bindings to the C++ libtorch libraries that power PyTorch. It … | 73 | 1216 | active |
| frotms/PaddleOCR2Pytorch A PyTorch port of PaddleOCR that lets you run PaddleOCR-trained models (detection, recognition, and document structure parsing) without the… | 70 | 1213 | active |
| Picsart-AI-Research/Text2Video-Zero Official implementation of Text2Video-Zero, a zero-shot text-to-video generation method that adapts text-to-image diffusion models like Sta… | 31 | 4244 | maintenance |
| ifzhang/FairMOT FairMOT is a research implementation of a one-shot multi-object tracking model that jointly performs object detection and re-identification… | 32 | 4241 | maintenance |
| NVIDIA/kvpress kvpress is a Python library from NVIDIA that implements multiple KV cache compression methods and benchmarks for long-context LLM inference… | 86 | 1211 | active |
| haidog-yaqub/MeanFlow An unofficial PyTorch implementation of MeanFlow and iMF, one-step generative modeling methods based on flow matching. It provides config-d… | 57 | 1208 | active |
| DachunKai/EvTexture Official PyTorch implementation of EvTexture and EvTexture++, event-driven video super-resolution models that use event-camera signals to e… | 50 | 1208 | active |
| toshas/torch-fidelity A PyTorch library providing accurate and efficient implementations of generative model evaluation metrics such as FID, Inception Score, KID… | 64 | 1206 | active |
| NVIDIA/earth2studio Earth2Studio is a Python deep-learning framework from NVIDIA for building, exploring, and deploying AI-driven weather and climate workflows… | 88 | 1205 | active |
| metavoiceio/metavoice-src MetaVoice-1B is a 1.2B parameter foundational text-to-speech model trained on 100K hours of speech, focused on emotional rhythm and tone in… | 26 | 4203 | maintenance |
| Artelnics/opennn OpenNN is an open-source C++ library for building, training, and deploying neural networks for advanced analytics. It is dependency-free, o… | 94 | 1202 | active |
| openvinotoolkit/nncf NNCF is Intel's Neural Network Compression Framework, a Python library providing post-training and training-time compression algorithms (qu… | 92 | 1202 | active |
| Cerebras/modelzoo Cerebras Model Zoo is a collection of reference deep learning model implementations (Llama, Mixtral, DINOv2, Llava, etc.) with configs and … | 74 | 1202 | active |
| AlgRUC/JittorGeometric JittorGeometric is a graph machine learning library built on the Jittor deep learning framework, providing implementations of 40+ Graph Neu… | 59 | 1200 | active |
| mlmed/torchxrayvision TorchXRayVision is an open-source PyTorch library providing pre-trained deep learning models and a unified interface for publicly available… | 89 | 1197 | active |
| a-r-j/graphein Graphein is a Python library for constructing graph and mesh representations of proteins, RNA, molecules, and biological interaction networ… | 74 | 1197 | active |
| autonomousvision/stylegan-t Official training code for StyleGAN-T, an ICML 2023 paper on fast large-scale text-to-image synthesis using GANs. It provides dataset prepa… | 32 | 1197 | active |
| ddlBoJack/emotion2vec Official PyTorch implementation of emotion2vec, a self-supervised pre-trained model for speech emotion representation. It provides code for… | 27 | 1197 | active |
| DeepRec-AI/DeepRec DeepRec is a high-performance deep learning framework for recommendation models, built on TensorFlow 1.15 with Intel and NVIDIA TensorFlow … | 23 | 1197 | active |
| facebookresearch/esm Meta FAIR's Evolutionary Scale Modeling (ESM) library providing Transformer protein language models with pretrained weights, including ESM-… | 10 | 4170 | maintenance |