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 |
|---|---|---|---|
| omoindrot/tensorflow-triplet-loss A TensorFlow implementation of triplet loss with online triplet mining strategies (batch all and batch hard), for training embedding models… | 32 | 1127 | abandoned |
| VikParuchuri/texify Texify is a deep-learning OCR model that converts images or PDFs containing math into markdown and LaTeX renderable by MathJax. It handles … | 10 | 1124 | abandoned |
| da03/Attention-OCR A Python/TensorFlow implementation of a visual attention-based OCR model that combines a sliding CNN, an LSTM, and an attention decoder to … | 32 | 1116 | abandoned |
| szad670401/end-to-end-for-chinese-plate-recognition An end-to-end Chinese license plate recognition model based on MXnet, using multi-label classification. It was trained on ~500k synthetic r… | 32 | 1116 | abandoned |
| pbloem/former A minimal, from-scratch implementation of the transformer architecture in PyTorch, accompanying a tutorial blog post on how modern transfor… | 34 | 1098 | abandoned |
| carpedm20/DiscoGAN-pytorch A PyTorch implementation of DiscoGAN, a generative adversarial network that discovers cross-domain relations in an unsupervised manner. It … | 32 | 1096 | abandoned |
| spragunr/deep_q_rl A Theano/Lasagne-based implementation of the Deep Q-Network (DQN) algorithm from DeepMind's Atari reinforcement learning papers. It trains … | 32 | 1094 | abandoned |
| dmlc/minpy MinPy is a Python library providing a pure NumPy interface backed by MXNet, automatically dispatching operators to GPU when available and f… | 10 | 1094 | abandoned |
| MarvinTeichmann/tensorflow-fcn A single-file TensorFlow implementation of Fully Convolutional Networks (FCN) for semantic segmentation, using VGG16 weights loaded from a … | 32 | 1093 | abandoned |
| SegNet A modified version of the Caffe deep learning framework implementing SegNet, a deep convolutional encoder-decoder architecture for semantic… | 32 | 1084 | abandoned |
| neuronika/neuronika Neuronika is a machine learning framework written in pure Rust providing tensors and dynamic neural networks with reverse-mode automatic di… | 32 | 1083 | abandoned |
| openai/InfoGAN Official research code from OpenAI for reproducing results of the InfoGAN paper, an information-theoretic extension to GANs that learns dis… | 10 | 1068 | abandoned |
| isl-org/MultiObjectiveOptimization Source code for the NeurIPS 2018 paper 'Multi-Task Learning as Multi-Objective Optimization', implementing the MGDA_UB algorithm for multi-… | 10 | 1066 | abandoned |
| DeepLearningKit/DeepLearningKit DeepLearningKit is an open-source deep learning framework for Apple's iOS, OS X and tvOS, written in Swift and using Metal for GPU-accelera… | 32 | 1059 | abandoned |
| facebookarchive/fbcunn Facebook AI Research's CUDA-accelerated deep learning extension modules for the Torch-7 framework, including fast FFT-based convolutions, m… | 10 | 1058 | abandoned |
| TensorFlow.js tfjs-node was the Node.js native binding for TensorFlow.js, providing accelerated training and inference of ML models in JavaScript server … | 10 | 1055 | abandoned |
| shiyybua/NER A Chinese named entity recognition (NER) system built with TensorFlow, using a bidirectional RNN combined with a CRF layer for sequence lab… | 32 | 1054 | abandoned |
| basveeling/wavenet A Keras implementation of DeepMind's WaveNet, a generative neural network model for raw audio synthesis. It supports training on datasets l… | 32 | 1053 | abandoned |
| facebookresearch/VMZ VMZ is a model zoo from Facebook AI's Computer Vision team providing Caffe2 and PyTorch implementations of video classification models such… | 10 | 1053 | abandoned |
| NVIDIA-AI-IOT/redtail NVIDIA Redtail provides deep learning and computer vision components for autonomous visual navigation of drones and ground vehicles, center… | 23 | 1048 | abandoned |
| carpedm20/NTM-tensorflow A TensorFlow implementation of the Neural Turing Machine (NTM) architecture with an LSTM controller and support for multiple read/write hea… | 32 | 1047 | abandoned |
| openai/supervised-reptile Official research code for the Reptile meta-learning algorithm from the paper 'On First-Order Meta-Learning Algorithms'. It trains and eval… | 10 | 1042 | abandoned |
| PRBonn/lidar-bonnetal A deep learning framework for training and deploying semantic segmentation of LiDAR point clouds using range-image representations, develop… | 10 | 1037 | abandoned |
| andrewgordonwilson/bayesgan TensorFlow implementation of the Bayesian GAN from the NIPS 2017 paper by Saatchi and Wilson, which places conditional posteriors over gene… | 32 | 1013 | abandoned |
| coreylynch/async-rl A TensorFlow + Keras implementation of asynchronous 1-step Q-learning from the DeepMind 'Asynchronous Methods for Deep Reinforcement Learni… | 32 | 1003 | abandoned |
| Stable Diffusion web UI Stable Diffusion Web UI is a Python‑based web application that lets you run Stable Diffusion locally through an interactive Gradio interfac… | 51 | 165213 | active |
| deepseek-ai/DeepSeek-V3 Official repository for DeepSeek-V3, a 671B-parameter Mixture-of-Experts language model (37B activated per token) with open weights, infere… | 26 | 104498 | active |
| hacksider/Deep-Live-Cam Deep-Live-Cam is a Python‑based desktop application that performs real‑time face swapping and video deepfakes using a single source image. … | 88 | 97084 | active |
| OpenCV OpenCV is an open‑source computer vision and machine learning library written in C++. It provides over 2,500 optimized algorithms for image… | 85 | 90950 | active |
| karpathy/nanochat nanochat is a minimal, hackable full-stack LLM training harness that covers tokenization, pretraining, finetuning, evaluation, and inferenc… | 59 | 58750 | active |
| facebookresearch/segment-anything Segment Anything Model (SAM) from Meta AI is a promptable image segmentation foundation model that produces high-quality object masks from … | 31 | 54993 | stable |
| ray-project/ray Ray is a unified open-source framework for scaling AI and Python applications, consisting of a core distributed runtime (tasks, actors, obj… | 99 | 44057 | stable |
| OpenBMB/VoxCPM VoxCPM is a tokenizer-free text-to-speech system built on a diffusion autoregressive architecture that generates continuous speech represen… | 74 | 37924 | active |
| myshell-ai/OpenVoice OpenVoice is a Python library and audio foundation model for instant voice cloning, requiring only a short reference audio clip to replicat… | 31 | 37608 | active |
| huggingface/pytorch-image-models timm (PyTorch Image Models) is a Python library that provides a large collection of state‑of‑the‑art vision models, utilities, and training… | 94 | 37222 | active |
| facebookresearch/detectron2 Detectron2 is Facebook AI Research's PyTorch-based library for state-of-the-art object detection, instance/panoptic segmentation, and other… | 66 | 34897 | stable |
| FaceFusion FaceFusion is a Python‑based command‑line application for AI‑powered face manipulation, including swapping faces in images and videos with … | 94 | 30263 | active |
| FaceFusion FaceFusion is an open-source face manipulation platform for face swapping and lip syncing in images and videos, built in Python with GPU ac… | 94 | 30263 | active |
| SYSTRAN/faster-whisper A fast reimplementation of OpenAI's Whisper speech-to-text model built on the CTranslate2 inference engine, offering up to 4x speedup and l… | 51 | 25631 | active |
| haotian-liu/LLaVA LLaVA (Large Language and Vision Assistant) is an open-source multimodal large language model framework implementing visual instruction tun… | 21 | 25067 | active |
| DeepFace DeepFace is a lightweight Python library for face recognition and facial attribute analysis. It provides pre‑trained models to estimate age… | 92 | 23482 | active |
| spmallick/learnopencv LearnOpenCV is a repository of C++ and Python code examples for computer vision, deep learning, and AI. It provides ready‑to‑run notebooks … | 96 | 23170 | active |
| bloc97/Anime4K Anime4K is a set of open-source, high-quality real-time anime upscaling and denoising algorithms implemented as GLSL shaders, primarily for… | 23 | 21485 | stable |
| facebookresearch/sam2 Official code for Meta's Segment Anything Model 2 (SAM 2), a foundation model for promptable visual segmentation in images and videos. It i… | 58 | 20001 | active |
| KlingAIResearch/LivePortrait LivePortrait is a Python-based portrait animation tool from Kuaishou Technology that synthesizes lifelike videos from a single source image… | 59 | 19260 | active |
| lukas-blecher/LaTeX-OCR pix2tex (LaTeX-OCR) is a PyTorch-based vision transformer model that converts images of math formulas into LaTeX code. It ships as a pip-in… | 23 | 16595 | stable |
| NVIDIA/TensorRT-LLM NVIDIA's TensorRT-LLM is a Python library and runtime for defining and serving Large Language Models with state-of-the-art inference optimi… | 83 | 14851 | active |
| k2-fsa/OmniVoice OmniVoice is a massively multilingual zero-shot text-to-speech model supporting 600+ languages, built on a diffusion language model-style a… | 77 | 14762 | active |
| SesameAILabs/csm CSM (Conversational Speech Model) is Sesame's speech generation model that produces conversational audio from text and audio context, using… | 28 | 14728 | active |
| CARLA Simulator CARLA is an open-source simulator for autonomous driving research built on Unreal Engine, providing open digital assets like urban layouts,… | 67 | 14465 | active |
| dlib Dlib is a modern C++ toolkit containing machine learning algorithms, deep learning tools, computer vision, linear algebra, and general-purp… | 84 | 14458 | stable |
| Lightning-AI/litgpt LitGPT is a Python library providing from-scratch, hackable implementations of 20+ open-source large language models with recipes for pretr… | 87 | 13718 | active |
| PaddlePaddle/PaddleNLP PaddleNLP is an easy-to-use NLP and large language model development kit built on the PaddlePaddle deep learning framework, with a large pr… | 59 | 12977 | active |
| libfacedetection An open-source C++ library for CNN-based face detection in images, with the model embedded as static C source so it has no external depende… | 59 | 12807 | stable |
| cupy/cupy CuPy is a NumPy/SciPy-compatible array library for GPU-accelerated computing with Python, running on NVIDIA CUDA or AMD ROCm. It acts as a … | 92 | 12362 | stable |
| SpeechBrain SpeechBrain is an open-source PyTorch-based speech toolkit for building conversational AI systems. It provides training recipes, pretrained… | 85 | 11845 | active |
| Kornia Kornia is a differentiable computer vision library built on PyTorch, offering GPU-accelerated image processing, augmentations, and geometri… | 84 | 11436 | active |
| OpenVINO OpenVINO is an open-source toolkit from Intel for optimizing and deploying deep learning inference across CPU, GPU, and NPU hardware. It su… | 94 | 10996 | stable |
| IDEA-Research/GroundingDINO Official PyTorch implementation of Grounding DINO, an open-set object detector that fuses a Transformer-based DINO detector with grounded v… | 22 | 10682 | stable |
| X-AnyLabeling X-AnyLabeling is a cross-platform desktop application for AI-assisted annotation of text, image, video, and multimodal data. It bundles bui… | 96 | 10642 | active |
| bigscience-workshop/petals Petals is a Python library that lets you run and fine-tune large language models (Llama 3.1, Mixtral, Falcon, BLOOM) on a BitTorrent-style … | 23 | 10614 | active |
| pyannote.audio pyannote.audio is an open-source Python toolkit built on PyTorch for speaker diarization, providing neural building blocks like voice activ… | 92 | 10593 | active |
| zyddnys/manga-image-translator A Python tool that automatically detects, OCRs, translates, inpaints, and re-typesets text in images, primarily for manga and comics. It ru… | 66 | 10477 | active |
| open-mmlab/Amphion Amphion is an open-source Python toolkit for audio, music, and speech generation, supporting tasks like text-to-speech, voice conversion, s… | 47 | 10304 | active |
| ESPnet ESPnet is an end-to-end speech processing toolkit built on PyTorch covering speech recognition, text-to-speech, speech translation, enhance… | 94 | 9989 | active |
| roboflow/rf-detr RF-DETR is a real-time transformer-based model architecture from Roboflow for object detection, instance segmentation, and keypoint detecti… | 86 | 9678 | active |
| PeterL1n/RobustVideoMatting Robust Video Matting (RVM) is a deep learning model and library for real-time human video matting, using a recurrent neural network with te… | 23 | 9533 | stable |
| PaddlePaddle/PaddleSeg PaddleSeg is an end-to-end image segmentation toolkit built on PaddlePaddle, offering a model zoo with dozens of pre-trained models for sem… | 48 | 9395 | active |
| Deep Lake Deep Lake is an open-source database for AI that stores multimodal data (images, video, audio, text, embeddings, annotations) in a format o… | 75 | 9249 | active |
| tile-ai/tilelang TileLang is a Pythonic domain-specific language and compiler (built on TVM) for writing high-performance GPU/CPU/accelerator kernels such a… | 90 | 8724 | active |
| jasonppy/VoiceCraft VoiceCraft is a token infilling neural codec language model for zero-shot speech editing and text-to-speech on in-the-wild data like audiob… | 60 | 8576 | active |
| netease-youdao/EmotiVoice EmotiVoice is an open-source text-to-speech engine supporting English and Chinese with over 2000 voices and prompt-controlled emotional syn… | 60 | 8529 | active |
| OptimalScale/LMFlow LMFlow is an extensible Python toolkit for finetuning and inference of large foundation models such as LLaMA, GPT-2, and Galactica. It prov… | 61 | 8487 | active |
| CASIA-LMC-Lab/FastSAM FastSAM is a CNN-based Segment Anything Model trained on only 2% of the SA-1B dataset, achieving comparable segmentation performance to SAM… | 20 | 8421 | active |
| RapidAI/RapidOCR RapidOCR is an open-source, multi-language OCR toolkit that performs text detection and recognition using models converted to run on ONNX R… | 99 | 8134 | active |
| babysor/MockingBird MockingBird is a PyTorch-based AI voice cloning toolbox that can clone a voice from a 5-second sample and generate arbitrary speech in real… | 51 | 36895 | maintenance |
| face-alignment A Python library built on PyTorch that detects 2D and 3D facial landmarks in images using the FAN deep learning face alignment network. It … | 64 | 7543 | active |
| hybridgroup/gocv GoCV is a Go language binding for the OpenCV 4 computer vision library, supporting Linux, macOS, Windows, and Docker. It includes support f… | 66 | 7505 | active |
| naver/dust3r DUSt3R is the official PyTorch implementation of a CVPR 2024 model that performs dense, unconstrained stereo and multi-view 3D reconstructi… | 42 | 7317 | active |
| InternLM/InternLM Official repository for the InternLM series of open-source large language models (InternLM, InternLM2, InternLM2.5, InternLM3), including p… | 38 | 7276 | active |
| liuliu/ccv ccv is a modern, minimalist computer vision library written in C/C++ with an application-driven set of state-of-the-art algorithms includin… | 77 | 7247 | active |
| Zyphra/Zonos Zonos-v0.1 is an open-weight text-to-speech model trained on over 200k hours of multilingual speech, with a Python library for inference. I… | 22 | 7236 | active |
| Background Matting V2 Official PyTorch implementation of the CVPR 2021 paper 'Real-Time High-Resolution Background Matting'. It produces state-of-the-art alpha m… | 23 | 7193 | stable |
| zai-org/GLM-4 Official repository for the GLM-4 series of open-weight, multilingual (primarily Chinese/English) multimodal chat language models from Z.ai… | 64 | 7069 | active |
| sczhou/ProPainter ProPainter is a PyTorch-based video inpainting model from ICCV 2023 that combines dual-domain propagation with a mask-guided sparse video T… | 19 | 6964 | stable |
| flashinfer-ai/flashinfer FlashInfer is a GPU kernel library and kernel generator for LLM inference, providing unified APIs for attention, GEMM, and MoE operations w… | 92 | 6736 | active |
| linkedin/Liger-Kernel Liger Kernel is a collection of Triton kernels designed to make LLM training faster and more memory-efficient, offering drop-in replacement… | 91 | 6664 | active |
| AILab-CVC/YOLO-World YOLO-World is a real-time open-vocabulary object detection model and Python toolkit from Tencent AI Lab and HUST, published at CVPR 2024. I… | 26 | 6579 | active |
| rtqichen/torchdiffeq torchdiffeq is a PyTorch library of differentiable ordinary differential equation (ODE) solvers, best known as the canonical implementation… | 35 | 6482 | stable |
| HVision-NKU/StoryDiffusion StoryDiffusion is the official implementation of a NeurIPS 2024 Spotlight paper introducing Consistent Self-Attention for character-consist… | 25 | 6475 | active |
| szad670401/HyperLPR HyperLPR3 is a high-performance open-source framework for recognizing Chinese license plates, built with deep learning and available as a P… | 23 | 6270 | active |
| ladaapp/lada Lada is an open-source tool with both GUI and CLI that restores pixelated/mosaic regions in videos, primarily targeting JAV (Japanese adult… | 59 | 6193 | active |
| bytedance/MegaTTS3 MegaTTS 3 is ByteDance's open-source PyTorch text-to-speech model with a lightweight 0.45B-parameter Diffusion Transformer backbone. It pro… | 56 | 6096 | active |
| fla-org/flash-linear-attention A PyTorch library providing hardware-efficient implementations of emerging sequence model architectures, including linear attention, sparse… | 86 | 5940 | active |
| AidLux (AidLearning) AidLux (originally AidLearning) is an AIoT development platform that runs a native Ubuntu Linux environment with GUI, deep learning tooling… | 63 | 5812 | active |
| matterport/Mask_RCNN A Python implementation of the Mask R-CNN model for object detection and instance segmentation, built on Keras and TensorFlow with a ResNet… | 23 | 25574 | maintenance |
| 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 |
| timesler/facenet-pytorch A PyTorch library providing pretrained face detection (MTCNN) and facial recognition (Inception ResNet V1) models, ported from the TensorFl… | 38 | 5169 | stable |
| PaddlePaddle/VisualDL VisualDL is a deep learning visualization toolkit for PaddlePaddle that provides charts for tracking training metrics, visualizing model st… | 23 | 4881 | stable |
| UX-Decoder/Segment-Everything-Everywhere-All-At-Once SEEM is the official PyTorch implementation of the NeurIPS 2023 paper 'Segment Everything Everywhere All at Once', a model for universal im… | 21 | 4792 | stable |