function: deep-learning
2653 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| unitaryai/detoxify Detoxify is a Python library providing pre-trained transformer models that classify toxic comments, built on PyTorch Lightning and Hugging … | 76 | 1291 | maintenance |
| HumanAIGC/OutfitAnyone OutfitAnyone is an Alibaba research project providing ultra-high quality AI virtual try-on, letting users visualize any clothing on any per… | 26 | 5984 | experimental |
| NVlabs/alias-free-gan The official project page and code for Alias-Free GAN, an NVIDIA research method for aliasing-free generative adversarial networks. The imp… | 32 | 1289 | maintenance |
| Khrylx/PyTorch-RL A PyTorch library implementing deep reinforcement learning policy gradient algorithms (TRPO, PPO, A2C) and Generative Adversarial Imitation… | 32 | 1286 | maintenance |
| apple/ml-neuman Official reference implementation of NeuMan (ECCV 2022), which reconstructs an animatable human and the background scene from a single vide… | 32 | 1286 | maintenance |
| kwea123/ngp_pl A PyTorch + CUDA implementation of NVIDIA's Instant-NGP focused on NeRF, trained with PyTorch Lightning for fast, high-quality neural radia… | 23 | 1286 | maintenance |
| weihua916/powerful-gnns Official PyTorch implementation of the experiments from the ICLR 2019 paper 'How Powerful are Graph Neural Networks?', introducing the Grap… | 32 | 1285 | maintenance |
| oscarknagg/few-shot A PyTorch repository with clean, tested implementations that reproduce few-shot learning and meta-learning research papers such as Prototyp… | 32 | 1284 | maintenance |
| YuanxunLu/LiveSpeechPortraits A PyTorch implementation of the SIGGRAPH Asia 2021 paper 'Live Speech Portraits', which generates photorealistic personalized talking-head … | 32 | 1283 | maintenance |
| NVIDIA-Merlin/Transformers4Rec Transformers4Rec is a Python library by NVIDIA for sequential and session-based recommendation using transformer architectures, bridging NL… | 67 | 1282 | maintenance |
| harvardnlp/seq2seq-attn A Torch (Lua) implementation of sequence-to-sequence models with LSTM encoder-decoders and optional attention for neural machine translatio… | 32 | 1281 | maintenance |
| benedekrozemberczki/CapsGNN A PyTorch implementation of the Capsule Graph Neural Network (CapsGNN) from ICLR 2019, which uses capsule networks and routing mechanisms t… | 23 | 1278 | maintenance |
| snap-research/articulated-animation Official research code for the CVPR 2021 paper 'Motion Representations for Articulated Animation' by Snap Research. It animates a static so… | 43 | 1277 | maintenance |
| kaiyinzhou/BERT-NER A Python implementation that fine-tunes Google's BERT for named entity recognition using the CoNLL-2003 dataset, built on TensorFlow. It pr… | 32 | 1277 | maintenance |
| SullyChen/Autopilot-TensorFlow A TensorFlow implementation of Nvidia's end-to-end self-driving steering angle prediction paper (arXiv 1604.07316) with some modifications.… | 32 | 1276 | maintenance |
| j96w/DenseFusion DenseFusion is the official PyTorch implementation of the paper '6D Object Pose Estimation by Iterative Dense Fusion', which estimates the … | 32 | 1276 | maintenance |
| TRI-ML/packnet-sfm Official PyTorch implementation of PackNet and related self-supervised monocular depth estimation methods from Toyota Research Institute's … | 23 | 1274 | maintenance |
| marcbelmont/cnn-watermark-removal A TensorFlow implementation of a fully convolutional neural network that removes transparent watermark overlays from images. It trains on s… | 32 | 1269 | maintenance |
| linkedin/detext DeText is a deep neural text understanding framework from LinkedIn for NLP ranking, classification, and language generation tasks. It uses … | 23 | 1263 | maintenance |
| salesforce/pytorch-qrnn A PyTorch implementation of the Quasi-Recurrent Neural Network (QRNN) from Salesforce Research, offering LSTM-like accuracy with 2-17x fast… | 10 | 1263 | maintenance |
| kennymckormick/pyskl PYSKL is a PyTorch-based toolbox for skeleton-based human action recognition, built on MMAction2. It provides official implementations of P… | 53 | 1261 | maintenance |
| openpifpaf/openpifpaf OpenPifPaf is a PyTorch library implementing Composite Fields for semantic keypoint detection and spatio-temporal association, primarily fo… | 23 | 1261 | maintenance |
| kuleshov/audio-super-res A Python/TensorFlow implementation of neural network models for audio super resolution, upsampling low-resolution audio to higher sampling … | 32 | 1260 | maintenance |
| harvardnlp/im2markup A deep learning system (built on Torch) that converts images of rendered text into presentational markup such as LaTeX or HTML, using a CNN… | 32 | 1259 | maintenance |
| DrSleep/tensorflow-deeplab-resnet A TensorFlow re-implementation of the DeepLab-ResNet model for semantic image segmentation, trained and evaluated on the PASCAL VOC dataset… | 10 | 1258 | maintenance |
| uclaml/SPIN Official implementation of Self-Play Fine-Tuning (SPIN), a method that improves language models by having them iteratively generate and dis… | 26 | 1254 | maintenance |
| Fantasy-Studio/Paint-by-Example Paint by Example is a PyTorch implementation of exemplar-based image editing with diffusion models, letting users fill masked regions of an… | 32 | 1252 | maintenance |
| davidmrau/mixture-of-experts A PyTorch re-implementation of the sparsely-gated Mixture-of-Experts layer from the paper 'Outrageously Large Neural Networks' by Shazeer e… | 32 | 1252 | maintenance |
| clovaai/CutMix-PyTorch Official PyTorch implementation of CutMix, an image data augmentation regularizer that cuts and pastes patches between training images whil… | 32 | 1251 | maintenance |
| ARM-software/ML-KWS-for-MCU TensorFlow models and training scripts for keyword spotting (wake-word detection) on Arm Cortex-M microcontrollers, accompanying the 'Hello… | 32 | 1249 | maintenance |
| CQFIO/PhotographicImageSynthesis A TensorFlow implementation of the ICCV 2017 paper 'Photographic Image Synthesis with Cascaded Refinement Networks', which synthesizes phot… | 32 | 1248 | maintenance |
| leftthomas/SRGAN A PyTorch implementation of SRGAN, the CVPR 2017 generative adversarial network for photo-realistic single-image super-resolution. It inclu… | 32 | 1248 | maintenance |
| RedaOps/ann-visualizer A Python library that generates presentable graph diagrams of Keras neural network architectures using Graphviz. It supports visualizing bo… | 10 | 1248 | maintenance |
| XiangLi1999/Diffusion-LM Diffusion-LM is the official research code for the paper 'Diffusion-LM Improves Controllable Text Generation', implementing a diffusion-bas… | 32 | 1245 | maintenance |
| KMnP/vpt Official PyTorch implementation of Visual Prompt Tuning (VPT), an ECCV 2022 method for parameter-efficient fine-tuning of vision transforme… | 32 | 1244 | maintenance |
| mravanelli/SincNet SincNet is a PyTorch neural architecture that processes raw audio waveforms using parametrized sinc band-pass filters in the first convolut… | 32 | 1243 | maintenance |
| tf-encrypted/tf-encrypted TF Encrypted is a Python framework for privacy-preserving machine learning in TensorFlow, enabling training and prediction on encrypted dat… | 32 | 1243 | maintenance |
| NeymarL/ChineseChess-AlphaZero An open-source implementation of AlphaZero/AlphaGo Zero reinforcement learning methods applied to Chinese chess (Xiangqi). It includes self… | 23 | 1242 | maintenance |
| lfz/DSB2017 The winning solution of team 'grt123' for the 2017 Data Science Bowl (DSB2017), a deep learning pipeline for detecting lung cancer from CT … | 32 | 1241 | maintenance |
| ranahanocka/point2mesh Point2Mesh is a PyTorch implementation of a SIGGRAPH 2020 technique that reconstructs watertight surface meshes from input point clouds by … | 32 | 1240 | maintenance |
| AlgoTraders/stock-analysis-engine A distributed stock analysis and backtesting framework that ingests automated pricing data from IEX Cloud, Tradier, and FinViz and runs tho… | 32 | 1238 | maintenance |
| autonomousvision/giraffe Official PyTorch implementation of GIRAFFE, a CVPR 2021 (oral, best paper award) generative model that represents scenes as compositional g… | 32 | 1238 | maintenance |
| BR-IDL/PaddleViT PaddleViT is a collection of state-of-the-art Vision Transformer and MLP model implementations for PaddlePaddle 2.1+, covering image classi… | 23 | 1238 | maintenance |
| VeritasYin/STGCN_IJCAI-18 Reference implementation of Spatio-Temporal Graph Convolutional Networks (STGCN) from the IJCAI 2018 paper, a deep learning framework for t… | 71 | 1237 | maintenance |
| openseg-group/openseg.pytorch Official PyTorch implementations of semantic segmentation models OCNet, OCRNet, and SegFix, achieving state-of-the-art results on benchmark… | 23 | 1237 | maintenance |
| facebookresearch/simsiam A PyTorch implementation of SimSiam, a simple Siamese-based self-supervised representation learning method from Facebook AI Research (CVPR … | 10 | 1237 | maintenance |
| jfzhang95/pytorch-video-recognition A PyTorch library implementing C3D, R3D, and R2Plus1D models for video action recognition, with training scripts for UCF101 and HMDB51 data… | 32 | 1236 | maintenance |
| jakesnell/prototypical-networks Reference PyTorch implementation of Prototypical Networks for few-shot classification from the NeurIPS 2017 paper. It includes training and… | 32 | 1235 | maintenance |
| qubvel/classification_models A Keras/TensorFlow Keras library providing ImageNet-pretrained image classification model architectures such as ResNet, SE-ResNet, DenseNet… | 23 | 1231 | maintenance |
| PlayVoice/vits_chinese A Chinese text-to-speech library combining VITS with BERT-based prosody embeddings and NaturalSpeech infer-loss features, supporting ONNX c… | 23 | 1227 | maintenance |
| iqiyi/FASPell FASPell is a Python-based Chinese spell checker based on the DAE-Decoder paradigm, published at the EMNLP 2019 W-NUT workshop. It detects a… | 32 | 1223 | maintenance |
| google-research/fixmatch Official research code for FixMatch, a semi-supervised learning method that combines consistency regularization with confidence-based pseud… | 10 | 1223 | maintenance |
| xvjiarui/GCNet Official PyTorch implementation of GCNet (Global Context Networks), a paper combining ideas from Non-Local Networks and Squeeze-Excitation … | 32 | 1220 | maintenance |
| Jhy1993/HAN Reference implementation of the Heterogeneous Graph Attention Network (HAN) model from the WWW 2019 paper, built on top of GAT in Python. I… | 32 | 1219 | maintenance |
| xingyizhou/CenterNet2 CenterNet2 is a research implementation of probabilistic two-stage object detection built on detectron2, where a class-agnostic one-stage C… | 32 | 1219 | maintenance |
| awslabs/sockeye Sockeye is an open-source sequence-to-sequence framework for Neural Machine Translation built on PyTorch, with distributed training and opt… | 23 | 1215 | maintenance |
| The-AI-Summer/self-attention-cv A PyTorch library implementing various self-attention mechanisms and transformer building blocks for computer vision, including multi-head … | 23 | 1214 | maintenance |
| mouna99/dien A TensorFlow implementation of the Deep Interest Evolution Network (DIEN) for click-through rate prediction, based on the Alibaba research … | 32 | 1211 | maintenance |
| mit-han-lab/tinyml MIT Han Lab's TinyML research repository containing projects like TinyTL and NetAug for memory-efficient deep learning on microcontrollers … | 32 | 1211 | maintenance |
| Alexander-H-Liu/End-to-end-ASR-Pytorch A PyTorch implementation of end-to-end automatic speech recognition (ASR), formerly known as Listen, Attend and Spell. It supports seq2seq … | 32 | 1208 | maintenance |
| NVlabs/VoxFormer Official PyTorch implementation of VoxFormer, a CVPR 2023 highlight paper presenting a sparse voxel transformer for camera-based 3D semanti… | 31 | 1208 | maintenance |
| lessw2020/Ranger-Deep-Learning-Optimizer Ranger is a PyTorch deep learning optimizer that synergistically combines RAdam (Rectified Adam), LookAhead, and Gradient Centralization in… | 32 | 1207 | maintenance |
| facebookresearch/ToMe ToMe (Token Merging) is a PyTorch library from Meta AI that speeds up existing Vision Transformers by merging similar tokens inside the net… | 10 | 1207 | maintenance |
| bckenstler/CLR A Keras callback implementing cyclical learning rate (CLR) policies from Leslie Smith's paper, with built-in triangular, triangular2, and e… | 32 | 1206 | maintenance |
| reiinakano/arbitrary-image-stylization-tfjs A browser-based implementation of arbitrary neural style transfer built with TensorFlow.js, letting users stylize any photo in the style of… | 32 | 1205 | maintenance |
| aigamedev/scikit-neuralnetwork A Python library providing deep neural network classifiers and regressors with a scikit-learn-compatible API, wrapping Lasagne and Theano. … | 10 | 1204 | maintenance |
| twitter-research/tgn Official PyTorch implementation of Temporal Graph Networks (TGN), a framework for deep learning on dynamic graphs represented as sequences … | 32 | 1202 | maintenance |
| google-research/tapas TAPAS is Google Research's implementation of transformer-based models for question answering over tables, including pretrained checkpoints … | 10 | 1202 | maintenance |
| henrywoo/chatllama An open-source Python library implementing RLHF-based fine-tuning of Meta's LLaMA models to build ChatGPT-style chat assistants, runnable o… | 22 | 1201 | maintenance |
| chainer/chainerrl ChainerRL is a deep reinforcement learning library implementing state-of-the-art algorithms like DQN, Rainbow, PPO, SAC, and A3C in Python … | 23 | 1198 | maintenance |
| lucidrains/CoCa-pytorch A Pytorch implementation of CoCa (Contrastive Captioners), an image-text foundation model that combines contrastive learning with an encode… | 23 | 1198 | maintenance |
| rotemtzaban/STIT STIT (Stitch it in Time) is a research implementation of a GAN-based framework for semantic editing of faces in real videos, based on the p… | 32 | 1197 | maintenance |
| facebookresearch/mixup-cifar10 Official PyTorch implementation of mixup, a data augmentation technique that trains neural networks on convex combinations of image pairs a… | 10 | 1197 | maintenance |
| DeepWisdom/AutoDL AutoDL is a fully automated deep learning framework from DeepWisdom that won the NeurIPS AutoDL challenge. It performs automatic feature en… | 23 | 1196 | maintenance |
| rinongal/StyleGAN-nada Official PyTorch implementation of StyleGAN-NADA, a CLIP-guided method for adapting pre-trained StyleGAN image generators to new domains us… | 32 | 1195 | maintenance |
| philipperemy/yolo-9000 A packaging of YOLO9000 (YOLOv2), a real-time object detection model that can detect over 9,000 object classes, built around the Darknet fr… | 32 | 1194 | maintenance |
| ROCm/MIOpen MIOpen is AMD's open-source library providing high-performance GPU kernels for machine learning primitives (convolutions, normalization, ac… | 93 | 1193 | maintenance |
| VITA-Group/DeblurGANv2 Official PyTorch implementation of DeblurGAN-v2, an ICCV 2019 relativistic conditional GAN for single-image motion deblurring with a Featur… | 32 | 1192 | maintenance |
| altosaar/variational-autoencoder A reference implementation of a variational autoencoder (VAE) in PyTorch, TensorFlow, and JAX, including an inverse autoregressive flow var… | 23 | 1192 | maintenance |
| yitu-opensource/T2T-ViT Official PyTorch implementation of Tokens-to-Token ViT (T2T-ViT), an ICCV 2021 vision transformer architecture trained from scratch on Imag… | 23 | 1191 | maintenance |
| elliottwu/unsup3d Official PyTorch implementation of the CVPR 2020 (Oral, Best Paper Award) research paper 'Unsupervised Learning of Probably Symmetric Defor… | 32 | 1189 | maintenance |
| vanhuyz/CycleGAN-TensorFlow A TensorFlow implementation of CycleGAN for unpaired image-to-image translation, based on the original CycleGAN paper. It includes scripts … | 23 | 1189 | maintenance |
| khundman/telemanom Telemanom is a Python framework that uses vanilla LSTMs (Keras/TensorFlow) to detect anomalies in multivariate time series sensor data, wit… | 33 | 1187 | maintenance |
| lightaime/deep_gcns_torch A PyTorch library implementing DeepGCNs, DeeperGCN, and GNN1000 - graph convolutional network architectures that borrow residual/dense conn… | 32 | 1187 | maintenance |
| zju3dv/snake Official research code for 'Deep Snake for Real-Time Instance Segmentation' (CVPR 2020 oral), implementing a deep contour-based instance se… | 32 | 1185 | maintenance |
| kexinhuang12345/DeepPurpose DeepPurpose is a PyTorch-based deep learning library for molecular modeling, supporting drug-target interaction (DTI), drug property, drug-… | 23 | 1184 | maintenance |
| PyRetri/PyRetri PyRetri is a unified deep learning based unsupervised image retrieval toolbox built on PyTorch. It provides a modular pipeline of pre-proce… | 32 | 1183 | maintenance |
| HyperGAN/HyperGAN HyperGAN is a composable GAN (generative adversarial network) framework built on PyTorch, offering both a Python API and a CLI with a user … | 23 | 1183 | maintenance |
| jankrepl/deepdow deepdow is a Python library built on PyTorch that connects deep learning with portfolio optimization, allowing networks that output asset w… | 32 | 1181 | maintenance |
| lucidrains/performer-pytorch A PyTorch implementation of the Performer transformer, which uses linear attention via the FAVOR+ (Fast Attention Via positive Orthogonal R… | 23 | 1181 | maintenance |
| naoto0804/pytorch-AdaIN An unofficial PyTorch implementation of the AdaIN paper 'Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization' (Huang… | 10 | 1181 | maintenance |
| Smerity/sha-rnn A PyTorch implementation of the Single Headed Attention RNN (SHA-LSTM) from Stephen Merity's paper, combining an LSTM with a single attenti… | 32 | 1180 | maintenance |
| VoltaML/voltaML VoltaML is a lightweight Python library that compiles and optimizes machine learning and deep learning models for high-performance inferenc… | 32 | 1176 | maintenance |
| clovaai/voxceleb_trainer A PyTorch framework for training and evaluating speaker recognition and verification models on the VoxCeleb datasets. It implements multipl… | 67 | 1175 | maintenance |
| fbcotter/pytorch_wavelets A PyTorch library implementing 2D discrete wavelet transforms (DWT), dual-tree complex wavelet transforms (DTCWT), their inverses, and a DT… | 23 | 1175 | maintenance |
| ermongroup/SDEdit SDEdit is a PyTorch implementation of the ICLR 2022 paper 'Guided Image Synthesis and Editing with Stochastic Differential Equations'. It p… | 32 | 1168 | maintenance |
| alexandre01/deepsvg Official PyTorch code for the NeurIPS 2020 paper 'DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation'. It provides a … | 32 | 1164 | maintenance |
| xiangwang1223/knowledge_graph_attention_network TensorFlow implementation of KGAT (Knowledge Graph Attention Network), a KDD 2019 paper on knowledge-aware personalized recommendation. It … | 32 | 1164 | maintenance |
| facebookresearch/meshrcnn Mesh R-CNN is Facebook AI Research's official implementation of the ICCV 2019 paper, a model that detects objects in images and predicts th… | 60 | 1161 | maintenance |
| shadow2496/VITON-HD Official PyTorch implementation of VITON-HD (CVPR 2021), a high-resolution image-based virtual try-on model that transfers a clothing item … | 40 | 1161 | maintenance |