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 |
|---|---|---|---|
| osmr/imgclsmob A research sandbox providing (re)implementations of numerous deep learning computer vision models for classification, segmentation, detecti… | 23 | 3012 | active |
| MeiGen-AI/MultiTalk MultiTalk is an audio-driven framework for generating multi-person conversational videos from multi-stream audio, a reference image, and a … | 53 | 3000 | active |
| benedekrozemberczki/pytorch_geometric_temporal PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric providing spatiotemporal signal processing with … | 64 | 2992 | active |
| openai/shap-e Shap-E is OpenAI's official release of a generative model that produces 3D implicit functions conditioned on text prompts or images. It inc… | 30 | 12263 | maintenance |
| Biohub/esm A Python library and model release providing ESMC protein language models, ESMFold2 structure prediction, and the ESM Atlas of 6.8 billion … | 88 | 2987 | active |
| sovrasov/flops-counter.pytorch A Python library (ptflops) that computes the theoretical number of multiply-add operations (FLOPs) and parameter counts for neural network … | 36 | 2983 | active |
| patrick-kidger/equinox Equinox is a Python library providing neural networks and scientific computing utilities for JAX, using PyTorch-like class-based syntax whe… | 88 | 2976 | stable |
| zju3dv/LoFTR LoFTR is a detector-free local image feature matching method using Transformers, released with PyTorch inference and training code plus pre… | 32 | 2963 | stable |
| sunsmarterjie/yolov12 YOLOv12 is a PyTorch implementation of attention-centric real-time object detectors, published at NeurIPS 2025. It provides detection model… | 66 | 2961 | active |
| karpathy/char-rnn char-rnn is a Torch/Lua implementation of multi-layer recurrent neural networks (RNN, LSTM, GRU) for character-level language modeling. It … | 32 | 12120 | maintenance |
| dreamquark-ai/tabnet A PyTorch implementation of the TabNet deep learning architecture for tabular data, providing attentive, interpretable models for classific… | 23 | 2956 | stable |
| iscyy/ultralyticsPro A PyTorch-based collection of improved YOLO-family object detection models (YOLOv5 through YOLOv13, RT-DETR) with pluggable modules for bac… | 45 | 2954 | active |
| turboderp/exllama ExLlama is a standalone Python/C++/CUDA implementation of the Llama architecture optimized for running 4-bit GPTQ quantized weights on mode… | 30 | 2947 | active |
| bghira/SimpleTuner SimpleTuner is a Python fine-tuning toolkit for image, video, and audio diffusion models built on Hugging Face Diffusers. It provides a web… | 93 | 2940 | active |
| sokrypton/ColabFold ColabFold is a set of tools and Google Colab notebooks that make protein structure prediction with AlphaFold2, AlphaFold3, ESMFold, and rel… | 92 | 2940 | active |
| Tencent-Hunyuan/HunyuanWorld-1.0 Tencent HunyuanWorld-1.0 is an open-source 3D world generation model that creates immersive, explorable, and interactive 3D worlds from tex… | 50 | 2936 | active |
| InternLM/InternLM-XComposer InternLM-XComposer is a family of large vision-language models from the InternLM team, including XComposer2.5 for long-context multimodal u… | 35 | 2927 | active |
| state-spaces/s4 Official implementations of Structured State Space (S4) sequence models and related variants like S4D, HiPPO, and SaShiMi, built in PyTorch… | 32 | 2927 | active |
| keras-team/keras-tuner KerasTuner is a hyperparameter optimization framework for Keras models with a define-by-run syntax for configuring search spaces. It ships … | 47 | 2923 | active |
| decoderesearch/circuit-tracer A Python library for circuit tracing and attribution graphs in language model interpretability, based on cross-layer MLP transcoders. It co… | 78 | 2918 | active |
| eric-mitchell/direct-preference-optimization A reference implementation of Direct Preference Optimization (DPO) for training language models from human preference data, built on Huggin… | 29 | 2916 | stable |
| AllenNLP AllenNLP is an open-source NLP research library built on PyTorch for developing state-of-the-art deep learning models on linguistic tasks. … | 10 | 11886 | maintenance |
| Saiyan-World/goku Goku is a family of flow-based (rectified flow Transformer) foundation models for joint image and video generation, released by HKU and Byt… | 21 | 2906 | active |
| learnables/learn2learn learn2learn is a PyTorch library for meta-learning research, providing utilities for few-shot task creation, high-level wrappers for algori… | 45 | 2894 | active |
| explosion/thinc Thinc is a lightweight deep learning library from the makers of spaCy offering a type-checked, functional-programming API for composing neu… | 74 | 2891 | active |
| KellerJordan/Muon Muon is a PyTorch optimizer for the hidden layers of neural networks, based on orthogonalized momentum updates via Newton-Schulz iteration.… | 56 | 2887 | active |
| werner-duvaud/muzero-general A commented and documented PyTorch implementation of DeepMind's MuZero reinforcement learning algorithm, designed to be easily adapted to a… | 32 | 2870 | active |
| physical-superintelligence-lab/Psi0 Psi-Zero (Ψ₀) is an open vision-language-action (VLA) foundation model for dexterous humanoid loco-manipulation, combining a Qwen3-VL backb… | 59 | 2859 | active |
| linto-ai/whisper-timestamped A Python library extending OpenAI's Whisper models to produce accurate word-level timestamps and confidence scores during multilingual spee… | 77 | 2855 | active |
| OpenGVLab/InternImage InternImage is a large-scale CNN-based vision foundation model that uses deformable convolutions as its core operator, released with pretra… | 25 | 2854 | stable |
| huggingface/nanotron Nanotron is a minimalistic Python library from Hugging Face for pretraining large language models with 3D parallelism (data, tensor, and pi… | 63 | 2843 | active |
| pytorch/xla PyTorch/XLA is a Python package that connects the PyTorch deep learning framework to XLA devices such as Google Cloud TPUs via the XLA deep… | 63 | 2801 | active |
| lucidrains/DALLE2-pytorch A PyTorch implementation of OpenAI's DALL-E 2 text-to-image synthesis model, focusing on the diffusion prior network that predicts image em… | 23 | 11302 | maintenance |
| NVlabs/stylegan2 The official TensorFlow implementation of StyleGAN2, NVIDIA's improved style-based generative adversarial network for high-quality uncondit… | 32 | 11180 | maintenance |
| NVIDIA/FastPhotoStyle FastPhotoStyle is NVIDIA's official PyTorch implementation of the ECCV 2018 paper 'A Closed-form Solution to Photorealistic Image Stylizati… | 23 | 11171 | maintenance |
| apple/turicreate Turi Create is a Python library from Apple that simplifies building custom machine learning models for tasks like image classification, obj… | 10 | 11155 | maintenance |
| voxelmorph/voxelmorph VoxelMorph is a Python library for learning-based image registration and alignment, using unsupervised deep learning to model deformations … | 75 | 2754 | active |
| artidoro/qlora QLoRA is the official implementation of the QLoRA paper, an efficient finetuning approach that backpropagates through a frozen 4-bit quanti… | 30 | 11035 | maintenance |
| qualcomm/aimet AIMET (AI Model Efficiency Toolkit) is a Python library from Qualcomm providing advanced quantization and compression techniques for traine… | 99 | 2723 | active |
| intel/neural-compressor Intel Neural Compressor is an open-source Python library providing state-of-the-art low-bit quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4),… | 95 | 2722 | active |
| prophesier/diff-svc Diff-SVC is a deep learning project that performs singing voice conversion using diffusion models, transforming input singing audio into a … | 58 | 2719 | active |
| yuqinie98/PatchTST Official PyTorch implementation of PatchTST, an ICLR 2023 Transformer model for long-term time series forecasting based on patching and cha… | 32 | 2719 | stable |
| lengstrom/fast-style-transfer A TensorFlow implementation of fast neural style transfer that applies the style of famous paintings to photos and videos in real time. It … | 32 | 10957 | maintenance |
| bmild/nerf The official TensorFlow implementation of NeRF (Neural Radiance Fields), the ECCV 2020 paper representing scenes as neural radiance fields … | 36 | 10944 | maintenance |
| yuweihao/MambaOut MambaOut is a PyTorch implementation of Gated CNN models from the CVPR 2025 paper 'MambaOut: Do We Really Need Mamba for Vision?', which qu… | 16 | 2710 | stable |
| KimMeen/Time-LLM Time-LLM is the official PyTorch implementation of an ICLR 2024 paper that reprograms frozen large language models (Llama, GPT-2, BERT) for… | 44 | 2707 | active |
| TMElyralab/MusePose MusePose is a diffusion-based, pose-guided image-to-video generation framework for creating virtual human videos, where a character in a re… | 25 | 2706 | active |
| dreamzero0/dreamzero DreamZero is NVIDIA's World Action Model (WAM) that jointly predicts future video and actions from a pretrained video diffusion backbone, e… | 48 | 2700 | active |
| HiLab-git/SSL4MIS A benchmark and code collection of semi-supervised learning methods for medical image segmentation, re-implementing approaches like Mean Te… | 40 | 2699 | active |
| princeton-vl/DROID-SLAM DROID-SLAM is a deep learning-based visual SLAM system for monocular, stereo, and RGB-D cameras that estimates camera trajectory and dense … | 38 | 2699 | active |
| roboflow/maestro maestro is a Python library from Roboflow that streamlines fine-tuning of multimodal vision-language models such as Florence-2, PaliGemma 2… | 61 | 2696 | active |
| naiveHobo/InvoiceNet InvoiceNet is a deep neural network application with a GUI for extracting structured information from invoice documents in PDF, JPG, and PN… | 32 | 2696 | active |
| baaivision/EVA EVA is a family of large-scale vision foundation models from BAAI, including masked image models (EVA-01/02) and scaled CLIP models (EVA-CL… | 23 | 2694 | active |
| SkyReels SkyReels V1 is an open-source human-centric video foundation model with Text-to-Video and Image-to-Video variants, fine-tuned from HunyuanV… | 22 | 2694 | active |
| JIA-Lab-research/LISA LISA (Large Language Instructed Segmentation Assistant) is a multimodal large language model that performs reasoning-based image segmentati… | 29 | 2687 | active |
| openai/DALL-E The official PyTorch package for the discrete VAE (dVAE) component of OpenAI's DALL·E model. It does not include the transformer that gener… | 10 | 10820 | maintenance |
| MrGiovanni/UNetPlusPlus Official implementation of UNet++, a nested U-Net architecture for medical image segmentation, in both Keras and PyTorch. It redesigns skip… | 73 | 2685 | stable |
| InfiniteYou InfiniteYou (InfU) is a research framework from ByteDance for identity-preserved text-to-image generation built on Diffusion Transformers l… | 34 | 2685 | active |
| Tencent-Hunyuan/HY-World-2.0 HY-World 2.0 is Tencent Hunyuan's open-source multi-modal world model framework that reconstructs, generates, and simulates 3D worlds from … | 55 | 2680 | active |
| LAM LHM is a PyTorch-based large reconstruction model that reconstructs high-fidelity animatable 3D human avatars from a single image in second… | 49 | 2679 | active |
| stochasticai/xTuring xTuring is a Python library for fine-tuning, evaluating, and running open-source large language models such as LLaMA, GPT-J, GPT-2, Qwen, a… | 65 | 2675 | active |
| StableSR StableSR is a Python research library that leverages pre-trained Stable Diffusion priors for real-world blind image super-resolution. It pr… | 21 | 2672 | stable |
| Tencent/MimicMotion MimicMotion is a diffusion-based framework from Tencent for generating high-quality human motion videos guided by pose sequences, featuring… | 44 | 2668 | active |
| open-gigaai/giga-brain-0 GigaBrain-0/0.7 is an open-source vision-language-action (VLA) model family for generalist embodied agents, powered by world models and a t… | 60 | 2659 | active |
| phillipi/pix2pix The original Torch (Lua) implementation of pix2pix, a conditional GAN for image-to-image translation tasks such as synthesizing photos from… | 32 | 10659 | maintenance |
| FeatherTalk An ultralight talking-head (digital human) model that animates a person's face from audio input and runs in real time on mobile devices. It… | 61 | 2648 | active |
| swz30/Restormer Restormer is an efficient Transformer architecture for high-resolution image restoration, published as a CVPR 2022 Oral paper. It provides … | 41 | 2648 | stable |
| ultralytics/yolov3 Ultralytics' PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection. It provides training, validation… | 66 | 10613 | maintenance |
| AudioLDM AudioLDM 2 is a Python library and CLI for generating audio, music, and speech from text prompts using latent diffusion models. It includes… | 29 | 2642 | active |
| OmniSVG/OmniSVG OmniSVG is a family of end-to-end multimodal SVG generation models built on pre-trained Vision-Language Models, released with inference cod… | 48 | 2626 | active |
| lucidrains/audiolm-pytorch A PyTorch implementation of AudioLM, Google Research's language modeling approach to audio generation, including a MIT-licensed SoundStream… | 34 | 2625 | active |
| meta-pytorch/torchrec TorchRec is a PyTorch domain library for building recommendation systems at scale. It provides distributed sharding of large embedding tabl… | 95 | 2615 | active |
| crowsonkb/k-diffusion A PyTorch library implementing Karras et al. (2022) diffusion models with enhancements like improved sampling algorithms and transformer-ba… | 49 | 2599 | active |
| CodeGeeX CodeGeeX is a family of open multilingual code generation large language models (13B and successors CodeGeeX2/CodeGeeX4) pre-trained on 20+… | 24 | 2597 | active |
| atong01/conditional-flow-matching TorchCFM is a PyTorch library implementing Conditional Flow Matching (CFM), a simulation-free training objective for continuous normalizing… | 72 | 2596 | active |
| asteroid-team/asteroid Asteroid is a PyTorch-based audio source separation toolkit for researchers, providing modular building blocks (filterbanks, encoders, mask… | 66 | 2593 | active |
| median-research-group/LibMTL LibMTL is an open-source PyTorch library for Multi-Task Learning (MTL). It provides implementations of many MTL architectures and gradient-… | 38 | 2593 | active |
| facebookresearch/demucs Demucs is a state-of-the-art music source separation model from Meta AI that splits songs into stems like drums, bass, and vocals using a h… | 10 | 10373 | maintenance |
| vita-epfl/Stable-Video-Infinity Stable Video Infinity (SVI) is a research codebase for infinite-length video generation using video diffusion transformers with an error-re… | 60 | 2579 | active |
| Tencent-Hunyuan/HY-Motion-1.0 HY-Motion 1.0 is a series of text-to-3D human motion generation models from Tencent Hunyuan, based on Diffusion Transformer and Flow Matchi… | 55 | 2574 | active |
| advimman/lama LaMa is a PyTorch-based image inpainting model that fills large missing regions in images using fast Fourier convolutions, generalizing wel… | 32 | 10291 | maintenance |
| LTH14/JiT A PyTorch/GPU re-implementation of JiT (Just image Transformer), a minimalist pixel-space diffusion model for high-resolution image generat… | 39 | 2553 | active |
| DeepDetect DeepDetect is an open-source deep learning runtime, CLI, and REST server written in C++ for training and inference across images, text, tab… | 95 | 2551 | active |
| mozilla/TTS A deep learning library for advanced text-to-speech generation, built on PyTorch with models like Tacotron2, Glow-TTS, and various vocoders… | 23 | 10173 | maintenance |
| learning-at-home/hivemind Hivemind is a PyTorch library for decentralized deep learning across the Internet, enabling training of large models on hundreds of volunte… | 53 | 2541 | active |
| X-PLUG/mPLUG-Owl mPLUG-Owl is a family of open-source multimodal large language models that combine visual encoders with LLMs (built on LLaMA) for image and… | 33 | 2539 | active |
| VITA-MLLM/VITA VITA is an open-source interactive omni multimodal large language model (VITA-1.5) that supports real-time vision and speech interaction, s… | 26 | 2536 | active |
| bilibili/Index-anisora Index-AniSora is Bilibili's open-source anime video generation model, capable of creating video shots in diverse anime styles from images, … | 60 | 2523 | active |
| amap-cvlab/ABot-World ABot-World is an open-source interactive world model that generates infinite, explorable video worlds from a single image, running on a sin… | 57 | 2513 | active |
| HiDream-ai/HiDream-I1 HiDream-I1 is an open-source 17B-parameter text-to-image generative foundation model based on a Sparse Diffusion Transformer, with full and… | 31 | 2510 | active |
| luanfujun/deep-photo-styletransfer Reference implementation of the CVPR 2017 paper 'Deep Photo Style Transfer', performing photorealistic image style transfer using Torch wit… | 32 | 9982 | maintenance |
| google-deepmind/sonnet Sonnet is a Python library built on TensorFlow 2 that provides simple, composable abstractions (snt.Module) for constructing neural network… | 66 | 9973 | maintenance |
| sthalles/SimCLR A PyTorch reference implementation of SimCLR, a self-supervised contrastive learning framework for learning visual representations from unl… | 23 | 2494 | stable |
| GaParmar/img2img-turbo A research library implementing one-step image-to-image translation models (CycleGAN-Turbo and pix2pix-turbo) built on SD-Turbo diffusion m… | 38 | 2488 | active |
| ppogg/YOLOv5-Lite YOLOv5-Lite is a lightweight object detection model family evolved from YOLOv5, with models as small as ~900KB (int8) that run 10-15+ FPS o… | 23 | 2488 | active |
| xuebinqin/U-2-Net Official PyTorch implementation of U^2-Net, a nested U-structure deep network for salient object detection, published in Pattern Recognitio… | 32 | 9875 | maintenance |
| wgsxm/PartCrafter PartCrafter is a structured 3D generative model that jointly generates multiple semantically meaningful 3D mesh parts and objects from a si… | 50 | 2485 | active |
| thuml/Autoformer Autoformer is the official PyTorch implementation of the NeurIPS 2021 paper 'Autoformer: Decomposition Transformers with Auto-Correlation f… | 32 | 2485 | stable |
| alexjc/neural-doodle A Python implementation of Semantic Style Transfer (Champandard, 2016) based on the Neural Patches algorithm. It turns rough doodles into r… | 10 | 9851 | maintenance |
| chemprop/chemprop Chemprop is a Python package implementing message passing neural networks (MPNNs) for predicting molecular and reaction properties. It prov… | 88 | 2480 | active |