domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| autonomousvision/occupancy_networks Official PyTorch implementation of the CVPR 2019 paper 'Occupancy Networks: Learning 3D Reconstruction in Function Space'. It learns contin… | 32 | 1663 | maintenance |
| hfawaz/dl-4-tsc Companion repository for the paper 'Deep learning for time series classification: a review', providing TensorFlow 2 implementations of nine… | 32 | 1659 | maintenance |
| google-research/pegasus PEGASUS is Google Research's implementation of transformer encoder-decoder models pre-trained with the Gap Sentences Generation objective f… | 10 | 1655 | maintenance |
| rwightman/efficientdet-pytorch A PyTorch implementation of EfficientDet object detection, faithful to the original Google TensorFlow implementation with ported pretrained… | 23 | 1654 | maintenance |
| HonglinChu/SiamTrackers A PyTorch collection of Siamese-based visual object tracking models including SiamFC, SiamRPN++, SiamMask, Ocean, LightTrack, and the light… | 32 | 1653 | maintenance |
| amir-abdi/keras_to_tensorflow A Python CLI tool that converts trained Keras models (.h5 or .h5 + .) into frozen, inference-ready TensorFlow protobuf (.pb) files. It free… | 32 | 1646 | maintenance |
| kan-bayashi/ParallelWaveGAN Unofficial PyTorch implementations of non-autoregressive neural vocoders including Parallel WaveGAN, MelGAN, Multi-band MelGAN, HiFi-GAN, a… | 23 | 1645 | maintenance |
| gnobitab/RectifiedFlow Official PyTorch implementation of Rectified Flow, an ICLR 2023 Spotlight method for learning transport maps between distributions via stra… | 32 | 1644 | maintenance |
| autonomio/talos Talos is a Python library for automating hyperparameter experiments and model evaluation with TensorFlow (tf.keras), Keras, and PyTorch mod… | 23 | 1636 | maintenance |
| lucidrains/alphafold2 An unofficial PyTorch implementation of DeepMind's AlphaFold2 protein structure prediction network, built as a translation of the official … | 23 | 1635 | maintenance |
| facebookresearch/consistent_depth A research library from Facebook AI Research implementing Consistent Video Depth Estimation (SIGGRAPH 2020). It reconstructs dense, flicker… | 10 | 1634 | maintenance |
| tensorflow/mesh Mesh TensorFlow is a Python library and embedded language for specifying distributed tensor computations, letting users define how model di… | 10 | 1630 | maintenance |
| facebookresearch/pytorch_GAN_zoo A PyTorch toolbox of GAN implementations including Progressive Growing GAN, DCGAN, and StyleGAN (beta), plus tools like GDPP, AC-GAN condit… | 10 | 1627 | maintenance |
| facebookresearch/higher higher is a PyTorch library that enables higher-order gradients over losses spanning entire training loops rather than individual steps. It… | 10 | 1626 | maintenance |
| lucidrains/PaLM-rlhf-pytorch A PyTorch library implementing Reinforcement Learning from Human Feedback (RLHF) on top of the PaLM transformer architecture, aiming to rep… | 78 | 7868 | experimental |
| allenai/bilm-tf A TensorFlow implementation of the bidirectional language model (biLM) used to compute ELMo deep contextualized word representations. It su… | 32 | 1612 | maintenance |
| divamgupta/stable-diffusion-tensorflow A TensorFlow/Keras implementation of Stable Diffusion with weights ported from the original implementation. It provides a Python API and Co… | 32 | 1608 | maintenance |
| experiencor/keras-yolo3 A Keras/TensorFlow implementation of YOLOv3 for object detection, supporting detection with pretrained weights, custom model training with … | 32 | 1608 | maintenance |
| Delta-ML/delta DELTA is a deep learning based end-to-end natural language and speech processing platform built on TensorFlow and Python 3. It provides one… | 10 | 1607 | maintenance |
| ialhashim/DenseDepth Official Keras/TensorFlow implementation (with experimental PyTorch and TF2 code) of the DenseDepth paper for high-quality monocular depth … | 32 | 1606 | maintenance |
| wy1iu/sphereface Official implementation of SphereFace (Deep Hypersphere Embedding for Face Recognition, CVPR 2017), built on Caffe with a full face recogni… | 32 | 1606 | maintenance |
| chandrikadeb7/Face-Mask-Detection A face mask detection system built with OpenCV and TensorFlow/Keras that uses deep learning (SSD MobileNetV2) to detect whether people are … | 23 | 1606 | maintenance |
| kakaobrain/fast-autoaugment Official PyTorch implementation of Fast AutoAugment (NeurIPS 2019), which learns image augmentation policies via density-matching search. I… | 32 | 1605 | maintenance |
| tensorflow/haskell Haskell bindings to the TensorFlow machine learning library, providing a typed API for building and training neural network models. It incl… | 10 | 1604 | maintenance |
| PeterWang512/FALdetector FALdetector is the official PyTorch implementation of the ICCV 2019 paper 'Detecting Photoshopped Faces by Scripting Photoshop'. It provide… | 32 | 1600 | maintenance |
| NVlabs/FUNIT FUNIT is NVIDIA's PyTorch implementation of a few-shot unsupervised image-to-image translation model (ICCV 2019) that can translate images … | 32 | 1586 | maintenance |
| PaddlePaddle/PGL Paddle Graph Learning (PGL) is an efficient and flexible graph learning framework built on PaddlePaddle. It supports heterogeneous graph le… | 23 | 1585 | maintenance |
| lufficc/SSD A high-quality, fast, modular reference implementation of the SSD (Single Shot MultiBox Detector) object detection model in PyTorch. It sup… | 23 | 1585 | maintenance |
| D-X-Y/AutoDL-Projects AutoDL-Projects is a PyTorch library implementing automated deep learning algorithms, including neural architecture search (NAS) methods li… | 32 | 1581 | maintenance |
| melodyguan/enas Authors' TensorFlow implementation of Efficient Neural Architecture Search (ENAS), which discovers neural network architectures via paramet… | 32 | 1578 | maintenance |
| Temporal Segment Networks (TSN) Official code and pretrained models for Temporal Segment Networks (TSN), a deep learning framework for video action recognition published a… | 32 | 1577 | maintenance |
| xinntao/EDVR EDVR is the winning solution of the NTIRE19 video restoration challenges, built on enhanced deformable convolutional networks. The repo is … | 32 | 1577 | maintenance |
| maxpumperla/elephas Elephas is a Python library that extends Keras to run distributed deep learning training on Apache Spark. It serializes Keras models on the… | 23 | 1577 | maintenance |
| OsciiArt/DeepAA DeepAA is a Python project that generates ASCII art from grayscale line images using convolutional neural networks built with TensorFlow an… | 32 | 1576 | maintenance |
| Kotlin/kotlindl KotlinDL is a high-level deep learning framework written in Kotlin and inspired by Keras, built on top of the TensorFlow Java API and ONNX … | 23 | 1576 | maintenance |
| vturrisi/solo-learn solo-learn is a Python library of state-of-the-art self-supervised methods for unsupervised visual representation learning, built on PyTorc… | 65 | 1573 | maintenance |
| kumar-shridhar/PyTorch-BayesianCNN A PyTorch library implementing Bayesian convolutional neural networks with variational inference via Bayes by Backprop. It provides drop-in… | 32 | 1570 | maintenance |
| HumanAIGC/EMO EMO (Emote Portrait Alive) is a research codebase from Alibaba's Institute for Intelligent Computing that generates expressive talking port… | 25 | 7594 | experimental |
| sniklaus/3d-ken-burns A PyTorch reference implementation of the 3D Ken Burns Effect from a Single Image paper, which animates a still photo with a virtual camera… | 70 | 1569 | maintenance |
| AlexHex7/Non-local_pytorch A PyTorch implementation of the Non-local Neural Block from the paper 'Non-local Neural Networks', providing multiple variants (concatenati… | 32 | 1564 | maintenance |
| genforce/interfacegan InterFaceGAN is a Python research library that interprets the latent space of pretrained GANs (PGGAN, StyleGAN) to find semantic subspaces … | 32 | 1561 | maintenance |
| msracver/FCIS FCIS is the official MXNet implementation of the CVPR 2017 paper 'Fully Convolutional Instance-aware Semantic Segmentation', which won firs… | 32 | 1561 | maintenance |
| QingyongHu/RandLA-Net Official TensorFlow implementation of RandLA-Net, a neural architecture for efficient semantic segmentation of large-scale 3D point clouds,… | 32 | 1560 | maintenance |
| V2AI/Det3D Det3D is a PyTorch-based toolbox for 3D object detection from point clouds, offering implementations of models like PointPillars, SECOND, a… | 32 | 1560 | maintenance |
| IDEA-Research/MaskDINO Official PyTorch implementation of Mask DINO, a unified transformer-based framework for object detection and segmentation, built on detectr… | 23 | 1557 | maintenance |
| bfelbo/DeepMoji DeepMoji is a deep learning model trained on 1.2 billion tweets with emojis to understand emotional language, with code for scoring texts, … | 32 | 1554 | maintenance |
| vlawhern/arl-eegmodels A collection of well-validated convolutional neural network models (EEGNet, DeepConvNet, ShallowConvNet) for EEG signal classification, imp… | 32 | 1552 | maintenance |
| vt-vl-lab/FGVC FGVC is a PyTorch implementation of the ECCV 2020 paper 'Flow-edge Guided Video Completion'. It completes missing regions in videos by extr… | 32 | 1551 | maintenance |
| Tencent/TurboTransformers A fast, user-friendly C++/Python runtime for transformer model inference (BERT, ALBERT, RoBERTa, GPT2, decoders) on CPU and GPU. It works a… | 37 | 1548 | maintenance |
| JiaRenChang/PSMNet PSMNet is the official PyTorch implementation of the CVPR 2018 paper 'Pyramid Stereo Matching Network' for stereo depth estimation. It uses… | 32 | 1548 | maintenance |
| caogang/wgan-gp A PyTorch implementation of the 'Improved Training of Wasserstein GANs' (WGAN-GP) paper, covering toy datasets, MNIST, CIFAR-10, and charac… | 32 | 1547 | maintenance |
| YoYo000/MVSNet MVSNet is a deep learning architecture for depth map inference from unstructured multi-view images, and R-MVSNet is its recurrent extension… | 32 | 1546 | maintenance |
| JingyunLiang/VRT VRT is the official PyTorch implementation of the paper 'VRT: A Video Restoration Transformer', a transformer-based model for video restora… | 23 | 1546 | maintenance |
| google-research/big_transfer Official repository for the Big Transfer (BiT) paper, providing ResNet models pre-trained on ImageNet and ImageNet-21k for transfer learnin… | 10 | 1541 | maintenance |
| gpleiss/efficient_densenet_pytorch A memory-efficient PyTorch implementation of DenseNets that uses gradient checkpointing to reduce feature map memory consumption from quadr… | 32 | 1535 | maintenance |
| poloclub/ganlab GAN Lab is an interactive, browser-based visualization tool for learning and experimenting with Generative Adversarial Networks. Built with… | 56 | 1534 | maintenance |
| lucidrains/lambda-networks A PyTorch implementation of Lambda Networks, featuring the λ layer that models long-range interactions by transforming contexts into linear… | 23 | 1528 | maintenance |
| yu4u/age-gender-estimation A Keras/TensorFlow implementation of a convolutional neural network that estimates age and gender from face images, trained on the IMDB-WIK… | 23 | 1520 | maintenance |
| shaoxiongji/federated-learning A PyTorch implementation of the FedAvg federated learning algorithm reproducing the McMahan et al. AISTATS 2017 paper. It includes experime… | 32 | 1515 | maintenance |
| tanluren/yolov3-channel-and-layer-pruning A Python toolkit built on ultralytics/yolov3 that implements channel pruning, layer pruning, and knowledge distillation for YOLOv3/v4 (incl… | 32 | 1515 | maintenance |
| junyanz/BicycleGAN A PyTorch implementation of BicycleGAN, a model for multimodal image-to-image translation that generates diverse outputs from a single inpu… | 32 | 1514 | maintenance |
| Eric-mingjie/rethinking-network-pruning A PyTorch research codebase reproducing the ICLR 2019 paper 'Rethinking the Value of Network Pruning', which shows pruned models trained fr… | 32 | 1512 | maintenance |
| FudanNLP/fitlog fitlog (fast + git + log) is a Python tool that helps deep learning practitioners record training logs and manage experiment code, combinin… | 23 | 1512 | maintenance |
| krasserm/super-resolution A TensorFlow 2.x implementation of EDSR, WDSR, and SRGAN models for single image super-resolution, with a high-level training API and DIV2K… | 32 | 1511 | maintenance |
| marsauto/europilot Europilot is a Python toolkit that bridges Euro Truck Simulator 2 with deep-learning frameworks, enabling end-to-end self-driving research … | 32 | 1511 | maintenance |
| maderix/ANE A research project demonstrating backpropagation and neural network training directly on Apple's Neural Engine using reverse-engineered pri… | 47 | 7253 | experimental |
| nlpyang/BertSum BertSum is the official PyTorch implementation of the paper 'Fine-tune BERT for Extractive Summarization', providing preprocessing pipeline… | 32 | 1506 | maintenance |
| tinyvision/SOLIDER SOLIDER is a semantic-controllable self-supervised learning framework that learns general human representations from massive unlabeled huma… | 32 | 1504 | maintenance |
| yulunzhang/RCAN PyTorch implementation of RCAN, a very deep residual channel attention network for single image super-resolution from an ECCV 2018 paper. I… | 39 | 1503 | maintenance |
| sigsep/open-unmix-pytorch Open-Unmix is a PyTorch deep learning reference implementation for music source separation, with pre-trained models that split pop music in… | 23 | 1503 | maintenance |
| jina-ai/finetuner Jina AI's Finetuner is a Python library for task-oriented fine-tuning of pretrained models like BERT and CLIP to produce better embeddings … | 10 | 1503 | maintenance |
| AITTSMD/MTCNN-Tensorflow A TensorFlow reproduction of MTCNN (Multi-task Cascaded Convolutional Networks) for joint face detection and facial landmark alignment. It … | 32 | 1502 | maintenance |
| vacancy/Synchronized-BatchNorm-PyTorch A PyTorch library implementing Synchronized Batch Normalization, which computes batch statistics across all GPUs during multi-device traini… | 32 | 1502 | maintenance |
| mpezeshki/pytorch_forward_forward A PyTorch implementation of Geoffrey Hinton's forward-forward (FF) training algorithm, an alternative to back-propagation that computes gra… | 32 | 1500 | maintenance |
| luuuyi/CBAM.PyTorch A non-official PyTorch re-implementation of the CBAM (Convolutional Block Attention Module) paper from ECCV 2018. It provides channel and s… | 32 | 1500 | maintenance |
| Swall0w/torchstat torchstat is a lightweight neural network analyzer for PyTorch that reports parameter counts, FLOPs, multiply-adds, and memory usage per la… | 23 | 1500 | maintenance |
| facebookresearch/UnsupervisedMT The original implementation of phrase-based (PBSMT) and neural (NMT) unsupervised machine translation models from the EMNLP 2018 paper by F… | 10 | 1499 | maintenance |
| NVIDIAGameWorks/kaolin-wisp NVIDIA Kaolin Wisp is a PyTorch library and engine for neural fields research, built on NVIDIA Kaolin Core. It provides differentiable rend… | 23 | 1498 | maintenance |
| filipradenovic/cnnimageretrieval-pytorch A PyTorch toolbox for training and evaluating convolutional neural networks for image retrieval, implementing the authors' TPAMI 2018 and E… | 23 | 1495 | maintenance |
| DeepVoltaire/AutoAugment An unofficial Python implementation of the AutoAugment data augmentation policies learned for ImageNet, CIFAR-10, and SVHN, built on Pillow… | 32 | 1493 | maintenance |
| thu-ml/unidiffuser UniDiffuser is a unified diffusion framework that fits marginal, conditional, and joint distributions across image and text modalities in a… | 30 | 1486 | maintenance |
| chengdazhi/Deformable-Convolution-V2-PyTorch A PyTorch implementation of Deformable Convolution V2 (DCNv2) custom CUDA operators, ported from the original MXNet implementation. It prov… | 32 | 1484 | maintenance |
| speedinghzl/CCNet Official PyTorch implementation of CCNet, a Criss-Cross Attention network for semantic segmentation published at ICCV 2019 and TPAMI 2020. … | 32 | 1484 | maintenance |
| patrick-kidger/torchtyping A Python library providing type annotations for PyTorch tensors' shape, dtype, layout, and dimension names, with optional runtime checking … | 41 | 1482 | maintenance |
| pathak22/noreward-rl TensorFlow implementation of the ICML 2017 paper 'Curiosity-driven Exploration by Self-supervised Prediction', training RL agents with an i… | 32 | 1482 | maintenance |
| Hironsan/anago anaGo is a Python library for sequence labeling tasks such as named entity recognition, part-of-speech tagging, and semantic role labeling,… | 23 | 1482 | maintenance |
| chainer/chainercv ChainerCV is a Python library providing tools to train and run neural networks for computer vision tasks on top of the Chainer framework. I… | 10 | 1481 | maintenance |
| Sharpiless/Yolov5-deepsort-inference A Python library combining YOLOv5 object detection with DeepSort multi-object tracking to detect, track, and count vehicles and pedestrians… | 66 | 1479 | maintenance |
| ruotianluo/ImageCaptioning.pytorch A PyTorch research codebase for image captioning, supporting self-critical sequence training, bottom-up features, transformer captioning mo… | 32 | 1476 | maintenance |
| google-deepmind/jraph Jraph is a lightweight Graph Neural Network library built on JAX, providing a GraphsTuple data structure, graph utilities, and a zoo of for… | 10 | 1474 | maintenance |
| YuanGongND/ast Official PyTorch implementation of the Audio Spectrogram Transformer (AST) from the Interspeech 2021 paper, which applies a Vision Transfor… | 32 | 1472 | maintenance |
| peteanderson80/bottom-up-attention A bottom-up attention model based on Faster R-CNN with ResNet-101 trained on Visual Genome, producing features for salient image regions. T… | 32 | 1469 | maintenance |
| marian-nmt/marian Marian is an efficient Neural Machine Translation framework written in pure C++ with minimal dependencies, supporting fast multi-GPU traini… | 23 | 1468 | maintenance |
| lxztju/pytorch_classification A complete PyTorch image classification codebase built on torchvision, covering training, prediction, TTA, model ensembling, knowledge dist… | 32 | 1466 | maintenance |
| CStanKonrad/long_llama LongLLaMA is a large language model handling long contexts of 256k tokens or more, built on OpenLLaMA and fine-tuned with the Focused Trans… | 29 | 1465 | maintenance |
| HRNet/HigherHRNet-Human-Pose-Estimation Official PyTorch implementation of HigherHRNet, a CVPR 2020 bottom-up multi-person human pose estimation model using scale-aware high-resol… | 32 | 1463 | maintenance |
| szagoruyko/attention-transfer PyTorch reference implementation of the ICLR 2017 paper 'Paying More Attention to Attention', which improves convolutional neural networks … | 32 | 1463 | maintenance |
| datitran/face2face-demo A pix2pix demo that learns from facial landmarks and translates them into a trained target face, with a real-time webcam application. It in… | 32 | 1461 | maintenance |
| facebookresearch/MaskFormer MaskFormer is a PyTorch/Detectron2-based implementation of the NeurIPS 2021 paper 'Per-Pixel Classification is Not All You Need for Semanti… | 10 | 1460 | maintenance |
| una-dinosauria/3d-pose-baseline A TensorFlow implementation of a simple yet effective baseline for 3D human pose estimation from 2D keypoints, published at ICCV 2017. It i… | 32 | 1459 | maintenance |