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 |
|---|---|---|---|
| hzwer/ICCV2019-LearningToPaint A PyTorch research implementation of the ICCV 2019 paper 'Learning to Paint With Model-based Deep Reinforcement Learning'. It trains agents… | 38 | 2315 | maintenance |
| Dynamic 3D Gaussians Official PyTorch implementation of 'Dynamic 3D Gaussians: Tracking by Persistent Dynamic View Synthesis' (3DV 2024), which models dynamic 3… | 29 | 2303 | maintenance |
| Flow Forecast Flow Forecast is a PyTorch-based deep learning framework for time series forecasting, classification, and anomaly detection. It offers stat… | 66 | 2302 | maintenance |
| titu1994/Neural-Style-Transfer A Keras 2.0+ implementation of Neural Style Transfer from the paper 'A Neural Algorithm of Artistic Style', with improvements like INetwork… | 23 | 2287 | maintenance |
| jsbroks/coco-annotator COCO Annotator is a web-based image annotation tool for labeling images with segments, bounding boxes, keypoints, and object tracking to cr… | 23 | 2286 | maintenance |
| zju3dv/NeuralRecon NeuralRecon is a deep learning framework for real-time 3D scene reconstruction from monocular video with known camera poses. It reconstruct… | 32 | 2284 | maintenance |
| astooke/rlpyt A PyTorch library providing modular, high-throughput implementations of deep reinforcement learning algorithms, covering policy gradient, d… | 32 | 2276 | maintenance |
| andreas128/RePaint Official PyTorch implementation of RePaint, a CVPR 2022 paper on image inpainting using denoising diffusion probabilistic models. It provid… | 32 | 2273 | maintenance |
| ndrplz/ConvLSTM_pytorch A PyTorch implementation of the Convolutional LSTM (ConvLSTM) neural network module, provided as a reusable nn.Module supporting arbitrary … | 32 | 2271 | maintenance |
| Tencent/MedicalNet MedicalNet provides a series of 3D-ResNet pre-trained models trained on 23 diverse medical imaging datasets, with PyTorch transfer-learning… | 53 | 2269 | maintenance |
| qianqianwang68/omnimotion OmniMotion is a PyTorch implementation of the ICCV 2023 paper 'Tracking Everything Everywhere All at Once', which tracks every point in a v… | 30 | 2268 | maintenance |
| mrharicot/monodepth A TensorFlow implementation of unsupervised monocular depth estimation from single images using convolutional neural networks, based on the… | 32 | 2263 | maintenance |
| microsoft/DeBERTa Official implementation of DeBERTa and DeBERTa V3, transformer-based language models with disentangled attention and decoding-enhanced BERT… | 23 | 2260 | maintenance |
| ShoufaChen/DiffusionDet PyTorch implementation of DiffusionDet, the first diffusion-model-based object detection framework (ICCV 2023 Best Paper Finalist). It prov… | 23 | 2259 | maintenance |
| namisan/mt-dnn A PyTorch library implementing Multi-Task Deep Neural Networks (MT-DNN) for natural language understanding, based on Microsoft research pap… | 32 | 2256 | maintenance |
| hunglc007/tensorflow-yolov4-tflite A TensorFlow 2.x implementation of YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny object detection models, with scripts that convert original… | 32 | 2255 | maintenance |
| bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets A PyTorch implementation of several U-Net variants for image segmentation, including UNet, R2U-Net, Attention U-Net, Attention R2U-Net, and… | 32 | 2255 | maintenance |
| TorchCV TorchCV is a PyTorch-based framework providing reimplementations of deep learning models for major computer vision tasks. It covers image c… | 32 | 2252 | maintenance |
| hustvl/YOLOP YOLOP is a multi-task deep learning network that jointly performs traffic object detection, drivable area segmentation, and lane detection … | 32 | 2249 | maintenance |
| Daniil-Osokin/lightweight-human-pose-estimation.pytorch A PyTorch implementation of Lightweight OpenPose for real-time 2D multi-person human pose estimation on CPU. It detects up to 18 body keypo… | 32 | 2237 | maintenance |
| Jongchan/attention-module Official PyTorch implementation of two attention modules for convolutional neural networks: BAM (Bottleneck Attention Module) and CBAM (Con… | 32 | 2237 | maintenance |
| lonePatient/BERT-NER-Pytorch A PyTorch library for Chinese Named Entity Recognition using BERT-based models with Softmax, CRF, and Span decoding heads. It also supports… | 32 | 2236 | maintenance |
| machrisaa/tensorflow-vgg A Python library implementing the VGG16 and VGG19 convolutional neural networks in TensorFlow, loading pretrained weights from npy files fo… | 32 | 2235 | maintenance |
| XLabs AI Flux Toolkit XLabs AI's training scripts for fine-tuning the FLUX.1 diffusion model with LoRA, ControlNet, and IP-Adapter adapters, using DeepSpeed and … | 24 | 2227 | maintenance |
| mit-han-lab/temporal-shift-module PyTorch implementation of the Temporal Shift Module (TSM), an ICCV 2019 technique that adds temporal modeling to 2D CNNs at zero extra comp… | 32 | 2223 | maintenance |
| thu-ml/zhusuan ZhuSuan is a Python probabilistic programming library for Bayesian deep learning built on TensorFlow. It provides primitives for building p… | 32 | 2216 | maintenance |
| ZQCNN ZQCNN is a lightweight deep learning inference framework written in C/C++ that runs on Windows, Linux, and ARM-Linux. It ships with demos f… | 57 | 2212 | maintenance |
| epfLLM/meditron Meditron is a suite of open-source medical large language models (7B and 70B) adapted from Llama-2 via continued pretraining on a curated m… | 27 | 2211 | maintenance |
| allenai/longformer Longformer is a pretrained transformer model family (including the LongformerEncoderDecoder/LED variant) that processes long documents up t… | 23 | 2208 | maintenance |
| yhenon/pytorch-retinanet A PyTorch implementation of the RetinaNet object detection model with focal loss, designed for readability and easy modification. It includ… | 10 | 2204 | maintenance |
| ming024/FastSpeech2 A PyTorch implementation of Microsoft's FastSpeech 2 text-to-speech model, supporting English and Mandarin with single- and multi-speaker s… | 32 | 2199 | maintenance |
| magicleap/SuperPointPretrainedNetwork A PyTorch pre-trained implementation of the SuperPoint fully convolutional neural network for real-time interest point detection and descri… | 32 | 2196 | maintenance |
| lucidrains/reformer-pytorch A PyTorch implementation of the Reformer, an efficient Transformer architecture using LSH attention, reversible networks, and chunking to h… | 23 | 2195 | maintenance |
| tianweiy/CenterPoint Official PyTorch implementation of CenterPoint, a CVPR 2021 method that performs 3D object detection and tracking from LiDAR point clouds b… | 23 | 2195 | maintenance |
| githubharald/SimpleHTR A Handwritten Text Recognition (HTR) system implemented in TensorFlow that recognizes text from images of single words or text lines, train… | 69 | 2191 | maintenance |
| fatchord/WaveRNN A PyTorch implementation of DeepMind's WaveRNN neural vocoder plus a Tacotron text-to-speech system, trained on LJSpeech. It supports train… | 32 | 2190 | maintenance |
| alibaba/EasyNLP EasyNLP is a comprehensive PyTorch-based NLP toolkit from Alibaba that provides training, inference, and deployment for pre-trained languag… | 23 | 2186 | maintenance |
| maxpumperla/hyperas Hyperas is a Python library that wraps Hyperopt to provide convenient hyperparameter optimization for Keras models. Users annotate model pa… | 32 | 2173 | maintenance |
| open-mmlab/mmrotate MMRotate is an open-source PyTorch toolbox for rotated object detection, part of the OpenMMLab project. It provides modular components, mul… | 23 | 2168 | maintenance |
| nvdla/hw The hardware release of NVIDIA's open-source Deep Learning Accelerator (NVDLA), containing Verilog RTL, a C-model, testbenches, and synthes… | 32 | 2166 | maintenance |
| bubbliiiing/yolov4-pytorch A PyTorch implementation of the YOLOv4 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2163 | maintenance |
| facebookresearch/pycls pycls is a PyTorch-based codebase for image classification research developed by Facebook AI Research. It provides implementations of stand… | 10 | 2161 | maintenance |
| kingyiusuen/image-to-latex A PyTorch application that converts images of LaTeX math equations into LaTeX code using a ResNet-18 encoder and Transformer decoder traine… | 32 | 2160 | maintenance |
| jeonsworld/ViT-pytorch A PyTorch reimplementation of Google's Vision Transformer (ViT) for image classification, based on the paper 'An Image is Worth 16x16 Words… | 32 | 2159 | maintenance |
| SeanNaren/deepspeech.pytorch A PyTorch implementation of the DeepSpeech2 speech recognition model, built on PyTorch Lightning, supporting training, testing, and inferen… | 23 | 2134 | maintenance |
| magenta/magenta-js Magenta.js is a collection of TypeScript libraries for running inference with pre-trained Magenta machine learning models directly in the b… | 68 | 2133 | maintenance |
| Anime2Sketch Anime2Sketch is a PyTorch-based sketch extractor that converts anime art, illustrations, and manga into line drawings using pretrained GAN … | 74 | 2128 | maintenance |
| chuanqi305/MobileNet-SSD A Caffe implementation of the MobileNet-SSD object detection network with pretrained weights on the VOC0712 dataset achieving mAP of 0.727.… | 43 | 2126 | maintenance |
| atriumlts/subpixel A TensorFlow reimplementation of the efficient sub-pixel convolutional neural network (ESPCN) for single-image super-resolution, based on S… | 32 | 2124 | maintenance |
| AIZOO FaceMaskDetection An open-source face mask detection project providing a lightweight SSD-based model (1.01M parameters) with inference code for PyTorch, Tens… | 32 | 2122 | maintenance |
| songyouwei/ABSA-PyTorch PyTorch implementations of aspect-based sentiment analysis (ABSA) models, including attention-based and BERT-based approaches for aspect-te… | 10 | 2114 | maintenance |
| bubbliiiing/yolo3-pytorch A PyTorch implementation of the YOLOv3 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2111 | maintenance |
| asappresearch/sru SRU is a PyTorch library implementing the Simple Recurrent Unit, a highly parallelizable RNN that runs 10-16x faster than cuDNN LSTM with c… | 23 | 2108 | maintenance |
| graphific/DeepDreamVideo A Python-based command-line tool that applies Google's DeepDream algorithm to videos by extracting frames, running a pretrained deep neural… | 32 | 2107 | maintenance |
| bgshih/crnn An implementation of the Convolutional Recurrent Neural Network (CRNN), combining CNN, RNN, and CTC loss for image-based sequence recogniti… | 32 | 2101 | maintenance |
| facebookresearch/chameleon Repository for Meta Chameleon, an early-fusion token-based mixed-modal foundation model that understands and generates interleaved images a… | 10 | 2101 | maintenance |
| qubvel/efficientnet A Keras and TensorFlow Keras reimplementation of the EfficientNet convolutional neural network family (B0-B7), including ImageNet-pretraine… | 23 | 2098 | maintenance |
| audio-diffusion-pytorch A PyTorch library for audio generation using diffusion models, supporting unconditional and text-conditional generation, diffusion autoenco… | 23 | 2095 | maintenance |
| facebookresearch/swav PyTorch implementation of SwAV, a self-supervised method for pre-training convolutional networks without labels by swapping cluster assignm… | 10 | 2093 | maintenance |
| ai-forever/ru-gpts A repository of Russian GPT-3 language models (ruGPT3XL/Large/Medium/Small and ruGPT2Large) with usage and fine-tuning examples. It provide… | 32 | 2089 | maintenance |
| THUDM/P-tuning-v2 P-tuning v2 is a Python implementation of deep prompt tuning, applying trainable continuous prompts at every transformer layer so prompt tu… | 32 | 2079 | maintenance |
| koth/kcws A deep learning Chinese word segmentation toolkit implemented in C++ with TensorFlow, supporting BiLSTM+CRF and IDCNN+CRF models, plus part… | 23 | 2068 | maintenance |
| ozan-oktay/Attention-Gated-Networks A PyTorch implementation of attention gates for convolutional neural networks, applied to U-Net and VGG-16 architectures. It targets medica… | 32 | 2065 | maintenance |
| ChenglongChen/tensorflow-DeepFM A TensorFlow implementation of the DeepFM model combining factorization machines and deep neural networks for click-through-rate (CTR) pred… | 32 | 2063 | maintenance |
| Ultra-Fast-Lane-Detection A PyTorch implementation of the ECCV 2020 paper 'Ultra Fast Structure-aware Deep Lane Detection', providing training and evaluation code fo… | 32 | 2061 | maintenance |
| PyTorch-Encoding A PyTorch computer vision toolkit with optimized CUDA backend, providing encoding layers, synchronized multi-GPU batch normalization, and p… | 65 | 2045 | maintenance |
| alibaba/AliceMind AliceMind is Alibaba's collection of pre-trained encoder-decoder language models and related NLP techniques, including StructBERT, PALM, VE… | 23 | 2040 | maintenance |
| apple/ml-fastvit Official PyTorch implementation of FastViT, a fast hybrid vision transformer architecture using structural reparameterization, published at… | 71 | 2039 | maintenance |
| jiupinjia/SkyAR SkyAR is the official PyTorch implementation of the paper 'Castle in the Sky: Dynamic Sky Replacement and Harmonization in Videos'. It perf… | 32 | 2029 | maintenance |
| Talking Head Anime from a Single Image Demo applications for the 'Talking Head Anime from a Single Image' research project, built with PyTorch. It includes a manual poser for ani… | 32 | 2028 | maintenance |
| darglein/ADOP ADOP is a point-based differentiable neural rendering pipeline for scene refinement and novel view synthesis, implemented in C++/CUDA with … | 23 | 2028 | maintenance |
| mingyuliutw/UNIT UNIT is NVIDIA's research implementation of unsupervised image-to-image translation networks (NIPS 2017), translating images across domains… | 32 | 2026 | maintenance |
| NVlabs/alpamayo NVIDIA Alpamayo 1 is an open 10B-parameter reasoning vision-language-action (VLA) model for autonomous vehicles that pairs driving trajecto… | 59 | 2025 | maintenance |
| WongKinYiu/ScaledYOLOv4 PyTorch implementation of Scaled-YOLOv4, which scales the Cross Stage Partial Network (CSP) across model sizes from tiny to P7 for object d… | 23 | 2024 | maintenance |
| hila-chefer/Transformer-Explainability Official PyTorch implementation of the CVPR 2021 paper 'Transformer Interpretability Beyond Attention Visualization', providing relevance-b… | 32 | 2019 | maintenance |
| tinghuiz/SfMLearner SfMLearner is a TensorFlow implementation of the CVPR 2017 paper 'Unsupervised Learning of Depth and Ego-Motion from Video'. It trains mode… | 32 | 2016 | maintenance |
| brannondorsey/PassGAN A Python command-line implementation of the PassGAN paper that uses a Wasserstein GAN to generate password guesses, with a pretrained model… | 23 | 2015 | maintenance |
| facebookresearch/Detic Detic is the official code release for the ECCV 2022 paper 'Detecting Twenty-thousand Classes using Image-level Supervision'. It is an open… | 32 | 2013 | maintenance |
| Zz-ww/SadTalker-Video-Lip-Sync A Python tool built on SadTalker that generates lip-synced video from an audio file and a source video, with configurable face/lip region e… | 30 | 2012 | maintenance |
| anthropics/jacobian-lens A Python library implementing the Jacobian lens, an interpretability technique that linearly transports residual-stream activations at any … | 56 | 2011 | maintenance |
| open-mmlab/mmgeneration MMGeneration is a PyTorch-based toolkit from OpenMMLab for training and applying generative models, with a focus on GANs and support for di… | 23 | 2011 | maintenance |
| divelab/DIG DIG (Dive into Graphs) is a Python library built on PyTorch Geometric that provides a unified testbed for advanced graph deep learning rese… | 23 | 2010 | maintenance |
| WongKinYiu/yolor PyTorch implementation of the YOLOR paper 'You Only Learn One Representation: Unified Network for Multiple Tasks', a real-time object detec… | 23 | 2006 | maintenance |
| robbiebarrat/art-DCGAN A modified Torch implementation of DCGAN focused on generating artworks, with support for 128x128 images and resuming training from checkpo… | 32 | 2005 | maintenance |
| JDAI-CV/FaceX-Zoo FaceX-Zoo is a PyTorch toolbox for face recognition that provides training modules with various state-of-the-art supervisory heads and back… | 32 | 2000 | maintenance |
| vsitzmann/siren Official PyTorch implementation of SIREN, a neural network architecture using periodic (sine) activation functions for implicit neural repr… | 32 | 2000 | maintenance |
| haitongli/knowledge-distillation-pytorch A PyTorch framework for running knowledge distillation experiments, supporting both shallow (teacher-to-small-CNN) and deep distillation on… | 32 | 1999 | maintenance |
| NVlabs/edm Official PyTorch implementation of the NeurIPS 2022 paper 'Elucidating the Design Space of Diffusion-Based Generative Models' (EDM) by NVID… | 32 | 1999 | maintenance |
| apple/ml-cvnets CVNets is Apple's open-source PyTorch library for training computer vision networks, covering classification, detection, segmentation, vide… | 75 | 1993 | maintenance |
| HobbitLong/PyContrast PyContrast is a PyTorch library providing reference implementations of contrastive learning methods such as InstDis, CMC, and MoCo, along w… | 32 | 1992 | maintenance |
| salesforce/awd-lstm-lm A PyTorch toolkit for training LSTM and QRNN language models, implementing Salesforce Research's AWD-LSTM papers. It supports word-level an… | 10 | 1990 | maintenance |
| albertpumarola/GANimation Official PyTorch implementation of GANimation, an ECCV'18 research paper that animates facial expressions in a single image using a GAN con… | 32 | 1989 | maintenance |
| databricks/spark-deep-learning Deep Learning Pipelines for Apache Spark, now reduced to the HorovodRunner component for distributed deep learning training via Horovod on … | 23 | 1987 | maintenance |
| Music-and-Culture-Technology-Lab/omnizart Omnizart is a Python library and CLI for automatic music transcription, transcribing pitched instruments, vocal melody, chords, drum events… | 82 | 1979 | maintenance |
| guanshuicheng/invoice A Flask-based OCR microservice that recognizes Chinese VAT invoices (electronic, regular, and special) using a YOLOv3 + CRNN + CTC deep lea… | 32 | 1979 | maintenance |
| WuJie1010/Facial-Expression-Recognition.Pytorch A PyTorch implementation of CNN-based facial expression recognition achieving state-of-the-art accuracy on FER2013 (73.112%) and CK+ (94.64… | 32 | 1979 | maintenance |
| Tencent/ObjectDetection-OneStageDet OneStageDet (OSD) is a PyTorch-based single-stage object detection framework from Tencent implementing YOLOv2 and YOLOv3 with plans to unif… | 42 | 1976 | maintenance |
| JavierAntoran/Bayesian-Neural-Networks A collection of PyTorch implementations of Bayesian neural network approximate inference methods, including Bayes by Backprop, MC Dropout, … | 32 | 1976 | maintenance |
| r9y9/deepvoice3_pytorch A PyTorch implementation of Deep Voice 3 and related convolutional neural network-based text-to-speech synthesis models. It includes traini… | 23 | 1976 | maintenance |
| SummitKwan/transparent_latent_gan TL-GAN is a Python/TensorFlow project that makes a GAN's latent space transparent by discovering feature axes, enabling controlled image sy… | 32 | 1974 | maintenance |