domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 | 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 | 2241 | maintenance |
| microsoft/DeBERTa Official implementation of DeBERTa and DeBERTa V3, transformer-based language models with disentangled attention and decoding-enhanced BERT… | 23 | 2239 | 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 | 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 | 2235 | maintenance |
| hustvl/YOLOP YOLOP is a multi-task deep learning network that jointly performs traffic object detection, drivable area segmentation, and lane detection … | 32 | 2234 | maintenance |
| Jongchan/attention-module Official PyTorch implementation of two attention modules for convolutional neural networks: BAM (Bottleneck Attention Module) and CBAM (Con… | 32 | 2230 | maintenance |
| x-flux XLabs AI's training scripts for fine-tuning the FLUX.1 diffusion model with LoRA, ControlNet, and IP-Adapter adapters, using DeepSpeed and … | 23 | 2229 | 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 | 2221 | 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 | 2219 | maintenance |
| zuoqing1988/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… | 61 | 2214 | 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 | 2208 | maintenance |
| allenai/longformer Longformer is a pretrained transformer model family (including the LongformerEncoderDecoder/LED variant) that processes long documents up t… | 23 | 2205 | 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 | 2205 | maintenance |
| lucidrains/reformer-pytorch A PyTorch implementation of the Reformer, an efficient Transformer architecture using LSH attention, reversible networks, and chunking to h… | 23 | 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 |
| magicleap/SuperPointPretrainedNetwork A PyTorch pre-trained implementation of the SuperPoint fully convolutional neural network for real-time interest point detection and descri… | 32 | 2185 | 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 | 2185 | maintenance |
| alibaba/EasyNLP EasyNLP is a comprehensive PyTorch-based NLP toolkit from Alibaba that provides training, inference, and deployment for pre-trained languag… | 23 | 2184 | maintenance |
| githubharald/SimpleHTR A Handwritten Text Recognition (HTR) system implemented in TensorFlow that recognizes text from images of single words or text lines, train… | 72 | 2183 | 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 | 2182 | 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 | 2163 | 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 | 2162 | 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 |
| bubbliiiing/yolov4-pytorch A PyTorch implementation of the YOLOv4 object detection model with full training, prediction, and evaluation scripts. It supports training … | 23 | 2160 | 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 | 2159 | 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 | 2141 | maintenance |
| SeanNaren/deepspeech.pytorch A PyTorch implementation of the DeepSpeech2 speech recognition model, built on PyTorch Lightning, supporting training, testing, and inferen… | 23 | 2136 | maintenance |
| Mukosame/Anime2Sketch Anime2Sketch is a PyTorch-based sketch extractor that converts anime art, illustrations, and manga into line drawings using pretrained GAN … | 32 | 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.… | 46 | 2127 | 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… | 72 | 2125 | maintenance |
| AIZOOTech/FaceMaskDetection An open-source face mask detection project providing a lightweight SSD-based model (1.01M parameters) with inference code for PyTorch, Tens… | 32 | 2125 | maintenance |
| atriumlts/subpixel A TensorFlow reimplementation of the efficient sub-pixel convolutional neural network (ESPCN) for single-image super-resolution, based on S… | 32 | 2123 | 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 | 2112 | 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 | 2109 | 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 | 2106 | maintenance |
| bgshih/crnn An implementation of the Convolutional Recurrent Neural Network (CRNN), combining CNN, RNN, and CTC loss for image-based sequence recogniti… | 32 | 2105 | maintenance |
| facebookresearch/chameleon Repository for Meta Chameleon, an early-fusion token-based mixed-modal foundation model that understands and generates interleaved images a… | 10 | 2103 | maintenance |
| qubvel/efficientnet A Keras and TensorFlow Keras reimplementation of the EfficientNet convolutional neural network family (B0-B7), including ImageNet-pretraine… | 23 | 2100 | maintenance |
| archinetai/audio-diffusion-pytorch A PyTorch library for audio generation using diffusion models, supporting unconditional and text-conditional generation, diffusion autoenco… | 23 | 2096 | maintenance |
| facebookresearch/swav PyTorch implementation of SwAV, a self-supervised method for pre-training convolutional networks without labels by swapping cluster assignm… | 10 | 2096 | 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 | 2087 | 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 | 2078 | 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 | 2069 | 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 | 2066 | 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 | 2064 | maintenance |
| cfzd/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 | 2048 | maintenance |
| zhanghang1989/PyTorch-Encoding A PyTorch computer vision toolkit with optimized CUDA backend, providing encoding layers, synchronized multi-GPU batch normalization, and p… | 23 | 2044 | maintenance |
| alibaba/AliceMind AliceMind is Alibaba's collection of pre-trained encoder-decoder language models and related NLP techniques, including StructBERT, PALM, VE… | 23 | 2041 | maintenance |
| apple/ml-fastvit Official PyTorch implementation of FastViT, a fast hybrid vision transformer architecture using structural reparameterization, published at… | 28 | 2027 | 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 | 2026 | 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 | 2025 | maintenance |
| mingyuliutw/UNIT UNIT is NVIDIA's research implementation of unsupervised image-to-image translation networks (NIPS 2017), translating images across domains… | 32 | 2025 | maintenance |
| pkhungurn/talking-head-anime-demo Demo applications for the 'Talking Head Anime from a Single Image' research project, built with PyTorch. It includes a manual poser for ani… | 32 | 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 |
| 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 | 2017 | maintenance |
| hila-chefer/Transformer-Explainability Official PyTorch implementation of the CVPR 2021 paper 'Transformer Interpretability Beyond Attention Visualization', providing relevance-b… | 32 | 2014 | 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 | 2012 | 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 | 2012 | 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 | 2009 | 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 | 2008 | 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 | 2007 | 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 | 2005 | 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 | 2003 | 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 | 2003 | 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 | 2000 | 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 | 1999 | maintenance |
| HobbitLong/PyContrast PyContrast is a PyTorch library providing reference implementations of contrastive learning methods such as InstDis, CMC, and MoCo, along w… | 32 | 1993 | maintenance |
| vsitzmann/siren Official PyTorch implementation of SIREN, a neural network architecture using periodic (sine) activation functions for implicit neural repr… | 32 | 1993 | maintenance |
| NVlabs/edm Official PyTorch implementation of the NeurIPS 2022 paper 'Elucidating the Design Space of Diffusion-Based Generative Models' (EDM) by NVID… | 32 | 1991 | 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 | 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 |
| 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 | 1985 | maintenance |
| apple/ml-cvnets CVNets is Apple's open-source PyTorch library for training computer vision networks, covering classification, detection, segmentation, vide… | 32 | 1983 | 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 | 1982 | 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… | 45 | 1980 | 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 | 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 | 1975 | 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 | 1973 | maintenance |
| JavierAntoran/Bayesian-Neural-Networks A collection of PyTorch implementations of Bayesian neural network approximate inference methods, including Bayes by Backprop, MC Dropout, … | 32 | 1969 | maintenance |
| FoundationVision/LlamaGen LlamaGen is a family of autoregressive image generation models that apply next-token prediction, as in Llama-style LLMs, to visual generati… | 24 | 1966 | 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… | 88 | 1964 | maintenance |
| ml-jku/hopfield-layers A PyTorch library implementing modern Hopfield networks with continuous states, whose update rule is equivalent to transformer attention. I… | 32 | 1959 | maintenance |
| mit-han-lab/once-for-all Once-for-All (OFA) is a PyTorch library implementing the ICLR 2020 Once-for-All network, which trains a single supernet that can be special… | 23 | 1956 | maintenance |
| guillaumegenthial/sequence_tagging A TensorFlow implementation of a Bi-LSTM + CRF model with character embeddings for named entity recognition and sequence tagging. It trains… | 32 | 1952 | maintenance |
| hukaixuan19970627/yolov5_obb A PyTorch implementation of YOLOv5 extended for oriented (rotated) object detection using Circular Smooth Label (CSL) angle encoding. It pr… | 32 | 1947 | maintenance |
| carefree0910/carefree-creator carefree-creator is a Python library and CLI that serves AI image generation endpoints built on Stable Diffusion and related models, poweri… | 32 | 1936 | maintenance |
| danieldjohnson/biaxial-rnn-music-composition A Python implementation of a biaxial recurrent neural network (LSTM-based) trained to generate classical music from MIDI data. It includes … | 32 | 1926 | maintenance |
| s9xie/hed Reference implementation of Holistically-Nested Edge Detection (HED), a deep learning edge and boundary detection model built on a modified… | 23 | 1925 | maintenance |
| S-LoRA/S-LoRA S-LoRA is a serving system for large language models that scales to thousands of concurrent LoRA adapters on one or more GPUs. It uses Unif… | 10 | 1923 | maintenance |
| Tencent/NeuralNLP-NeuralClassifier NeuralClassifier is a PyTorch-based open-source toolkit for neural text classification, with a focus on hierarchical multi-label classifica… | 56 | 1922 | maintenance |
| black0017/MedicalZooPytorch A PyTorch-based open-source library implementing state-of-the-art 3D (and some 2D) deep neural networks for multi-modal medical image segme… | 32 | 1917 | maintenance |
| appvision-ai/fast-bert Fast-Bert is a Python deep learning library for training and deploying BERT, RoBERTa, and XLNet based models for NLP tasks, starting with m… | 23 | 1917 | maintenance |
| WongKinYiu/PyTorch_YOLOv4 A PyTorch implementation of the YOLOv4 real-time object detection model, based on ultralytics/yolov3. It includes research extensions such … | 23 | 1908 | maintenance |
| astorfi/lip-reading-deeplearning A TensorFlow implementation of coupled 3D convolutional neural networks for cross audio-visual matching recognition, accompanying an IEEE A… | 23 | 1904 | maintenance |
| snap-stanford/GraphGym GraphGym is a platform for designing and evaluating Graph Neural Networks (GNNs), built around a highly modularized pipeline covering data … | 23 | 1904 | maintenance |
| ronghuaiyang/arcface-pytorch A PyTorch implementation of ArcFace, a deep metric learning approach for face recognition that adds angular margin penalties to face embedd… | 32 | 1901 | maintenance |
| whai362/PVT Official PyTorch implementation of the Pyramid Vision Transformer (PVTv1 and PVTv2) model series, providing pretrained backbone networks fo… | 23 | 1900 | maintenance |