domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| 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 |
| 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 |
| uber/petastorm Petastorm is a Python data access library from Uber that enables single-machine or distributed training and evaluation of deep learning mod… | 58 | 1891 | 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 |
| Ucas-HaoranWei/Vary Official ECCV 2024 implementation of Vary, a method for scaling up the vision vocabulary of large vision-language models. It provides train… | 26 | 1889 | 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 |
| zhoubolei/CAM Reference implementation of Class Activation Mapping (CAM), a CVPR'16 technique that visualizes which image regions a CNN attends to when p… | 32 | 1883 | maintenance |
| salesforce/ctrl CTRL is a 1.63 billion-parameter conditional Transformer language model from Salesforce that generates text controllable via control codes … | 10 | 1879 | 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 |
| 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 |
| 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 |
| 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 |
| MrNothing/AI-Blocks AI-Blocks is a WYSIWYG desktop application for visually building machine learning models by dragging objects with attached scripts in a sce… | 23 | 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 |
| google/gemmlowp gemmlowp is a small, self-contained, headers-only C++ library for low-precision (quantized) general matrix multiplication (GEMM). It provid… | 32 | 1843 | maintenance |
| AlexeyAB/Yolo_mark A Windows and Linux GUI application for drawing bounding boxes around objects in images to create labeled training data for YOLO v2/v3 obje… | 32 | 1840 | maintenance |
| microsoft/Cream A collection of Microsoft research projects on neural architecture search (NAS) and efficient Vision Transformers, including AutoFormer, Ti… | 23 | 1839 | maintenance |
| zetane/viewer Zetane Viewer is a free desktop application that visualizes machine learning models and their internal tensors in 3D, showing architecture,… | 23 | 1839 | maintenance |
| anthropics/jacobian-lens A Python library implementing the Jacobian lens, an interpretability technique that linearly transports residual-stream activations at any … | 56 | 1836 | maintenance |
| cubiq/ComfyUI_InstantID A ComfyUI custom node providing native InstantID support for identity-preserving image generation with SDXL, implemented without diffusers.… | 33 | 1834 | 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 |
| 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 |
| jkjung-avt/tensorrt_demos A collection of Python demo programs showing how to optimize and run deep learning models (YOLOv4, YOLOv3, SSD, MTCNN, GoogLeNet, MODNet) w… | 50 | 1792 | 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 |
| 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 |
| facebookresearch/TensorComprehensions Tensor Comprehensions is a C++ library with a Python API that provides a domain-specific language for expressing machine learning workloads… | 10 | 1767 | 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 |
| huggingface/transfer-learning-conv-ai A clean, commented PyTorch codebase for training a dialog/chatbot agent by transfer learning from OpenAI GPT/GPT-2 language models. It repr… | 32 | 1754 | 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 |
| deepset-ai/FARM FARM is a Python framework for fine-tuning and evaluating transformer-based language models for NLP tasks, with a focus on question answeri… | 10 | 1752 | 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 |
| 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 |
| 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 |
| natethegreate/hent-AI A Python application that automatically detects censor bars and mosaic blurs in illustrated adult content using deep learning (Mask R-CNN) … | 23 | 1730 | 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 |
| 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 |
| Maratyszcza/NNPACK NNPACK is a C99 acceleration package providing high-performance SIMD and multi-core CPU implementations of neural network layers, especiall… | 32 | 1710 | maintenance |
| airaria/TextBrewer TextBrewer is a PyTorch-based knowledge distillation toolkit for NLP models. It provides an easy-to-use framework implementing state-of-the… | 23 | 1708 | 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 |
| 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 |
| ELS-RD/transformer-deploy A Python tool that optimizes Hugging Face transformer models for production inference using ONNX Runtime and NVIDIA TensorRT, deploying the… | 23 | 1689 | maintenance |
| LoSealL/VideoSuperResolution A Python library (pip-installable as VSR) collecting reimplementation of state-of-the-art single-image and video super-resolution neural ne… | 23 | 1687 | maintenance |
| adobe/antialiased-cnns A PyTorch library providing antialiased CNN models and a BlurPool layer from the ICML 2019 paper 'Making Convolutional Networks Shift-Invar… | 23 | 1683 | 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 |
| 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 |
| akanazawa/hmr HMR (Human Mesh Recovery) is a TensorFlow implementation of the CVPR 2018 paper 'End-to-end Recovery of Human Shape and Pose', which regres… | 32 | 1666 | 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 |