function: deep-learning
2653 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 |
| 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 |
| jiesutd/NCRFpp NCRF++ is a PyTorch-based toolkit for neural sequence labeling tasks such as NER, POS tagging, and word segmentation. Model architectures c… | 32 | 1899 | maintenance |
| ChenyangSi/FreeU FreeU is a research codebase implementing a training-free method that improves diffusion model sample quality by re-weighting U-Net backbon… | 28 | 1899 | maintenance |
| facebookresearch/denoiser A PyTorch library implementing a causal, real-time speech enhancement model that operates on raw waveforms to remove background noise and r… | 10 | 1899 | maintenance |
| zjhuang22/maskscoring_rcnn Official PyTorch implementation of Mask Scoring R-CNN (CVPR 2019), built on maskrcnn-benchmark. It adds a network block that learns the qua… | 32 | 1895 | maintenance |
| TreB1eN/InsightFace_Pytorch A PyTorch reimplementation of InsightFace/ArcFace for face recognition, including backbone models (IR-SE50, MobileFacenet) and pretrained w… | 32 | 1894 | maintenance |
| junshutang/Make-It-3D Make-It-3D is a research codebase from an ICCV 2023 paper that creates high-fidelity 3D content from a single image using a 2D diffusion mo… | 30 | 1891 | maintenance |
| Duankaiwen/CenterNet Official PyTorch implementation of CenterNet, a one-stage keypoint-triplet-based object detector built on CornerNet. It trains from scratch… | 32 | 1884 | maintenance |
| hexiangnan/neural_collaborative_filtering Reference implementation of the WWW'17 Neural Collaborative Filtering paper, providing GMF, MLP, and NeuMF models for implicit-feedback rec… | 32 | 1883 | maintenance |
| open-mmlab/mmaction MMAction is an open-source PyTorch toolbox for video action understanding, covering action recognition, temporal action detection, and spat… | 32 | 1876 | maintenance |
| Sierkinhane/CRNN_Chinese_Characters_Rec A PyTorch implementation of a CRNN (convolutional recurrent neural network) model for recognizing Chinese characters in images. It includes… | 32 | 1875 | maintenance |
| pierluigiferrari/ssd_keras A Keras implementation of the Single Shot MultiBox Detector (SSD) object detection architecture, with ports of the original trained weights… | 23 | 1869 | maintenance |
| waleedka/hiddenlayer A lightweight Python library for rendering readable neural network architecture graphs and plotting training metrics for PyTorch, TensorFlo… | 23 | 1867 | maintenance |
| minimaxir/automl-gs automl-gs is a Python AutoML tool that takes an input CSV and a target prediction field and automatically generates a trained machine learn… | 23 | 1866 | maintenance |
| xiaolai-sqlai/mobilenetv3 A PyTorch implementation of the MobileNetV3 architecture with retrained pre-trained weights that outperform the original paper and torchvis… | 32 | 1864 | maintenance |
| facebookresearch/TimeSformer Official PyTorch implementation of TimeSformer, a space-time attention transformer model for video understanding from an ICML 2021 paper. I… | 10 | 1863 | maintenance |
| hanzhanggit/StackGAN TensorFlow implementation of StackGAN, a stacked generative adversarial network that synthesizes photo-realistic images from text descripti… | 32 | 1860 | maintenance |
| raminmh/liquid_time_constant_networks Official code repository for Liquid Time-Constant Networks (LTCs), a class of continuous-time recurrent neural networks with varying time-c… | 32 | 1857 | maintenance |
| magicleap/Atlas Atlas is a PyTorch-based deep learning model from Magic Leap that performs end-to-end 3D scene reconstruction from posed RGB images, produc… | 32 | 1857 | maintenance |
| nlpodyssey/spago Spago is a self-contained machine learning and NLP library written in pure Go, featuring an asynchronous computation graph with automatic d… | 29 | 1850 | maintenance |
| WangYueFt/dgcnn Author's reference implementation of Dynamic Graph CNN (DGCNN), a neural network module called EdgeConv for learning on 3D point clouds. It… | 32 | 1846 | maintenance |
| ermongroup/ddim Official PyTorch implementation of Denoising Diffusion Implicit Models (DDIM), a faster sampling method for diffusion probabilistic models.… | 32 | 1845 | maintenance |
| Score-Based Generative Modeling through SDEs Official PyTorch implementation of the ICLR 2021 paper 'Score-Based Generative Modeling through Stochastic Differential Equations', unifyin… | 32 | 1844 | maintenance |
| microsoft/Cream A collection of Microsoft research projects on neural architecture search (NAS) and efficient Vision Transformers, including AutoFormer, Ti… | 23 | 1839 | maintenance |
| google-deepmind/kinetics-i3d A repository of pre-trained Inflated 3D Convnet (I3D) models for video action classification, trained on the Kinetics dataset, released alo… | 32 | 1838 | maintenance |
| Janspiry/Palette-Image-to-Image-Diffusion-Models An unofficial PyTorch implementation of the Palette paper on image-to-image diffusion models, supporting tasks like inpainting, uncropping,… | 32 | 1833 | maintenance |
| bubbliiiing/faster-rcnn-pytorch A PyTorch implementation of the Faster R-CNN two-stage object detection model, supporting training on VOC-format datasets with ResNet or VG… | 23 | 1833 | maintenance |
| Kyubyong/tacotron A heavily documented TensorFlow implementation of Tacotron, a fully end-to-end text-to-speech synthesis model. It includes training, prepro… | 32 | 1832 | maintenance |
| NVlabs/stylegan2-ada NVIDIA's official TensorFlow implementation of StyleGAN2 with adaptive discriminator augmentation (ADA), enabling high-quality GAN training… | 32 | 1829 | maintenance |
| NVIDIA/semantic-segmentation NVIDIA's PyTorch monorepo implementing the paper 'Hierarchical Multi-Scale Attention for Semantic Segmentation', with pretrained weights an… | 32 | 1828 | maintenance |
| THUDM/CogDL CogDL is a Python/PyTorch toolkit for graph deep learning that lets researchers train and compare GNN and graph embedding models on tasks l… | 23 | 1821 | maintenance |
| yassouali/pytorch-segmentation A PyTorch library implementing multiple semantic segmentation models (DeepLab V3+, PSPNet, U-Net, SegNet, FCN, ENet, and others) with datas… | 26 | 1818 | maintenance |
| shunsukesaito/PIFu PyTorch implementation of PIFu (Pixel-Aligned Implicit Function), a deep learning method that reconstructs high-resolution 3D surfaces and … | 32 | 1817 | maintenance |
| sshaoshuai/PointRCNN PyTorch implementation of PointRCNN, a two-stage 3D object detector that generates 3D box proposals directly from raw LiDAR point clouds. I… | 32 | 1817 | maintenance |
| OpenPPL/ppq PPQ (PPL Quantization Tool) is an extensible, high-performance offline neural network quantization framework written in Python. It parses a… | 23 | 1816 | maintenance |
| google/compare_gan A TensorFlow library from Google Research providing configurable implementations of GAN components: losses, penalties, normalization techni… | 10 | 1815 | maintenance |
| yongzhuo/Keras-TextClassification A Keras-based library for Chinese text classification supporting long/short text, multi-label classification, and sentence pair similarity.… | 23 | 1808 | maintenance |
| rosinality/vq-vae-2-pytorch A PyTorch implementation of VQ-VAE-2 for generating diverse high-fidelity images, including training scripts for the VQ-VAE and PixelSNAIL … | 32 | 1803 | maintenance |
| xmartlabs/Bender Bender is a Swift library that provides an abstraction layer over Apple's MetalPerformanceShaders for defining and running neural networks … | 23 | 1802 | maintenance |
| zai-org/CogView CogView is a 4-billion-parameter pretrained transformer model for text-to-image image generation, released alongside a NeurIPS 2021 paper. … | 32 | 1799 | maintenance |
| minivision-ai/Silent-Face-Anti-Spoofing An open-source silent face anti-spoofing (liveness detection) project by MiniVision, providing model training code, data preprocessing, tes… | 32 | 1796 | maintenance |
| harskish/ganspace GANSpace is a research codebase implementing the NeurIPS 2020 paper on discovering interpretable controls for GANs via PCA in activation sp… | 32 | 1794 | maintenance |
| dreamoving/dreamoving-project DreaMoving is the official implementation of a diffusion-based controllable video generation framework from Alibaba that produces high-qual… | 26 | 1788 | maintenance |
| ali-vilab/dreamtalk DreamTalk is the official implementation of a diffusion-based framework for generating expressive, audio-driven talking head videos from a … | 26 | 1787 | maintenance |
| MCG-NJU/VideoMAE Official PyTorch implementation of VideoMAE, a masked autoencoder method for data-efficient self-supervised video pre-training with video t… | 32 | 1784 | maintenance |
| idiap/fast-transformers A PyTorch library providing fast transformer implementations with efficient attention mechanisms such as linear attention, designed to scal… | 32 | 1779 | maintenance |
| jquesnelle/yarn Reference implementation of YaRN, an efficient method for extending the context window of large language models, published as an ICLR 2024 … | 29 | 1777 | maintenance |
| wkentaro/pytorch-fcn A PyTorch implementation of Fully Convolutional Networks (FCN) for semantic segmentation, including training code that reproduces the origi… | 10 | 1777 | maintenance |
| qiuqiangkong/audioset_tagging_cnn PANNs is a collection of large-scale pretrained CNN audio neural networks trained on AudioSet (5000 hours, 527 classes) for audio pattern r… | 32 | 1772 | maintenance |
| princeton-vl/CornerNet-Lite CornerNet-Lite is the official PyTorch implementation of the paper 'CornerNet-Lite: Efficient Keypoint Based Object Detection', providing t… | 32 | 1771 | maintenance |
| CSAILVision/gandissect GANDissect is a PyTorch-based toolkit for visualizing and understanding the internal neurons of generative adversarial networks, showing ho… | 32 | 1765 | maintenance |
| nywang16/Pixel2Mesh Pixel2Mesh is the official TensorFlow implementation of the ECCV 2018 paper that generates 3D mesh models from a single RGB image using a g… | 32 | 1765 | maintenance |
| facebookresearch/votenet VoteNet is the official PyTorch implementation of the ICCV 2019 paper 'Deep Hough Voting for 3D Object Detection in Point Clouds'. It provi… | 10 | 1761 | maintenance |
| salesforce/ALBEF Official PyTorch implementation of ALBEF, a vision-and-language pre-training method that aligns image and text representations before fusin… | 10 | 1755 | maintenance |
| manuelruder/artistic-videos A Torch (Lua) implementation of the paper 'Artistic style transfer for videos', which applies the style of a painting to an entire video wh… | 10 | 1754 | maintenance |
| Lightning-Universe/lightning-bolts Lightning Bolts is a toolbox of pre-built models, callbacks, and datasets that extend PyTorch Lightning for AI/ML research and production. … | 10 | 1751 | maintenance |
| libfann/fann Fast Artificial Neural Network (FANN) is a free open-source C library implementing multilayer neural networks with fully and sparsely conne… | 73 | 1750 | maintenance |
| magenta/mt3 MT3 is a multi-instrument automatic music transcription model built on the T5X framework, converting audio recordings into multitrack MIDI.… | 73 | 1744 | maintenance |
| facebookresearch/deepcluster DeepCluster is a research codebase from Facebook AI Research implementing the 'Deep Clustering for Unsupervised Learning of Visual Features… | 10 | 1743 | maintenance |
| google-research/robotics_transformer A Python library from Google Research implementing RT-1 (Robotics Transformer), a transformer model for end-to-end robotic control with an … | 10 | 1743 | maintenance |
| tkipf/gae A TensorFlow implementation of (Variational) Graph Auto-Encoders from Kipf & Welling's 2016 paper, supporting unsupervised learning, cluste… | 32 | 1742 | maintenance |
| bearpaw/pytorch-classification A PyTorch library providing a unified interface for training image classification models on CIFAR-10/100 and ImageNet. It includes implemen… | 32 | 1738 | maintenance |
| openai/Video-Pre-Training OpenAI's Video PreTraining (VPT) codebase for learning Minecraft agents by watching unlabeled online videos, including behavioral cloning a… | 41 | 1737 | maintenance |
| NVlabs/PWC-Net Official NVIDIA implementation of PWC-Net, a CNN for optical flow estimation using pyramid, warping, and cost volume, released with CVPR 20… | 32 | 1736 | maintenance |
| zijundeng/pytorch-semantic-segmentation A PyTorch library providing implementations of popular semantic segmentation models such as FCN, U-Net, SegNet, PSPNet, GCN, and DUC/HDC. I… | 32 | 1735 | maintenance |
| zhougr1993/DeepInterestNetwork A TensorFlow demo implementation of the Deep Interest Network (DIN) paper for click-through rate prediction, run on the Amazon electronics … | 32 | 1733 | maintenance |
| experiencor/keras-yolo2 A Keras/TensorFlow implementation of the YOLOv2 real-time object detection model with support for training on custom datasets. It offers mu… | 23 | 1733 | maintenance |
| ranahanocka/MeshCNN MeshCNN is a PyTorch library implementing a convolutional neural network that operates directly on 3D triangular meshes, with mesh-specific… | 32 | 1731 | maintenance |
| NVIDIA/Cosmos-Tokenizer NVIDIA Cosmos Tokenizer is a suite of neural tokenizers for images and videos that convert visual data into continuous latents or discrete … | 10 | 1731 | maintenance |
| Qihoo360/hbox Hbox (formerly XLearning) is a scheduling platform that runs machine learning and deep learning frameworks like TensorFlow, MXNet, Caffe, a… | 47 | 1727 | maintenance |
| cszn/DnCNN DnCNN is the official implementation of the TIP 2017 paper 'Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising',… | 32 | 1727 | maintenance |
| google-research/torchsde A PyTorch library providing differentiable stochastic differential equation (SDE) solvers with GPU support and efficient backpropagation vi… | 10 | 1726 | maintenance |
| bubbliiiing/unet-pytorch A PyTorch implementation of the U-Net semantic segmentation model with training, prediction, and mIoU evaluation scripts. It supports multi… | 23 | 1725 | maintenance |
| Franck-Dernoncourt/NeuroNER NeuroNER is a Python/TensorFlow program that performs named-entity recognition (NER) using neural networks, with pretrained models and an e… | 23 | 1723 | maintenance |
| DeepMotionEditing/deep-motion-editing A PyTorch-based end-to-end library for editing and rendering 3D character motion, built on SIGGRAPH 2020 research. It provides deep learnin… | 32 | 1722 | maintenance |
| Lightning-Universe/lightning-flash Lightning Flash is a high-level PyTorch library built on PyTorch Lightning that provides ready-made 'recipes' for over 15 AI tasks across 7… | 10 | 1722 | maintenance |
| svip-lab/impersonator A PyTorch implementation of Liquid Gating GAN (ICCV 2019) that performs human motion imitation, appearance transfer, and novel view synthes… | 32 | 1717 | maintenance |
| dunbar12138/pix2pix3D pix2pix3D is the official PyTorch implementation of a CVPR 2023 paper on 3D-aware conditional image synthesis. It generates 3D objects (neu… | 31 | 1716 | maintenance |
| aiff22/DPED DPED is a Python/TensorFlow implementation of a deep convolutional network approach that translates ordinary smartphone photos into DSLR-qu… | 49 | 1710 | maintenance |
| Fyusion/LLFF LLFF (Local Light Field Fusion) is a TensorFlow implementation of the SIGGRAPH 2019 paper for novel view synthesis from sparse input images… | 32 | 1703 | maintenance |
| microsoft/i-Code Microsoft's i-Code is a collection of research models and frameworks for integrative, composable multimodal AI spanning vision, language, a… | 32 | 1703 | maintenance |
| imcaspar/gpt2-ml GPT2-ML is a TensorFlow-based GPT-2 training and inference project supporting multiple languages, with a focus on Chinese. It provides 1.5B… | 23 | 1701 | maintenance |
| NVIDIA/DeepRecommender A research library implementing deep autoencoders for collaborative filtering in recommender systems, based on the paper 'Training Deep Aut… | 10 | 1701 | maintenance |
| tensorflow/addons TensorFlow Addons is a collection of extra layers, optimizers, losses, metrics, activations, callbacks, and image ops that conform to Tenso… | 10 | 1699 | maintenance |
| VITA-Group/TransGAN Official PyTorch implementation of TransGAN, a NeurIPS 2021 paper that builds a GAN whose generator and discriminator are both pure transfo… | 32 | 1695 | maintenance |
| graph4ai/graph4nlp Graph4NLP is a Python library for applying Graph Neural Networks to Natural Language Processing tasks, built on DGL and PyTorch. It provide… | 23 | 1690 | maintenance |
| lucidrains/lightweight-gan A PyTorch implementation of the 'lightweight' GAN from ICLR 2021, featuring skip-layer excitation in the generator and autoencoding self-su… | 34 | 1682 | maintenance |
| open-mmlab/mmrazor MMRazor is OpenMMLab's model compression toolbox and benchmark built on PyTorch. It provides implementations of neural architecture search,… | 23 | 1682 | maintenance |
| microsoft/EdgeML A library from Microsoft Research India implementing resource-efficient machine learning algorithms (Bonsai, ProtoNN, FastGRNN, EMI-RNN, DR… | 23 | 1681 | maintenance |
| CuriousAI/mean-teacher Reference implementations (TensorFlow and PyTorch) of the Mean Teacher semi-supervised learning method from the NIPS 2017 paper by Tarvaine… | 32 | 1678 | maintenance |
| mit-han-lab/smoothquant SmoothQuant is a training-free post-training quantization (PTQ) method that enables W8A8 (8-bit weight, 8-bit activation) quantization for … | 32 | 1677 | maintenance |
| Lam1360/YOLOv3-model-pruning A PyTorch implementation of YOLOv3 channel pruning (network slimming) applied to hand detection on the Oxford Hand dataset. It provides spa… | 32 | 1676 | maintenance |
| argusswift/YOLOv4-pytorch A PyTorch re-implementation of YOLOv4 object detection with variants including attentive YOLOv4 (SEnet, CBAM, CoordAttention) and MobileNet… | 23 | 1676 | maintenance |
| Kaixhin/Rainbow A PyTorch implementation of the Rainbow deep reinforcement learning agent, combining DQN improvements like prioritized experience replay, d… | 23 | 1673 | maintenance |
| charlesq34/frustum-pointnets Official TensorFlow code release for the CVPR 2018 paper 'Frustum PointNets for 3D Object Detection from RGB-D Data' by Stanford and Nuro r… | 32 | 1668 | maintenance |
| natanielruiz/deep-head-pose Hopenet is a PyTorch deep learning model for fine-grained head pose estimation from images and video, without requiring facial keypoints. I… | 32 | 1666 | maintenance |