domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| bowang-lab/U-Mamba U-Mamba is a hybrid CNN-state-space-model (Mamba) network for biomedical image segmentation, built on top of the nnU-Net framework. It comb… | 26 | 1019 | active |
| google/prompt-to-prompt Google's official implementation of the Prompt-to-Prompt paper, which enables text-driven image editing in Latent Diffusion and Stable Diff… | 10 | 3456 | maintenance |
| towhee-io/towhee Towhee is a Python framework for building ETL pipelines that process unstructured data (images, video, text, audio) into embeddings using s… | 23 | 3454 | maintenance |
| tensorflow/adanet AdaNet is a lightweight TensorFlow-based AutoML framework that automatically learns high-quality neural network architectures and ensembles… | 10 | 3454 | maintenance |
| HobbitLong/SupContrast A PyTorch reference implementation of the Supervised Contrastive Learning paper (SupCon loss) that also supports SimCLR when labels are omi… | 32 | 3449 | maintenance |
| ShangtongZhang/DeepRL A modularized PyTorch implementation of popular deep reinforcement learning algorithms including DQN variants, PPO, DDPG, TD3, A2C, and Opt… | 23 | 3448 | maintenance |
| microsoft/Tutel Tutel is Microsoft's optimized Mixture-of-Experts (MoE) library for efficient training and inference of large language models, featuring dy… | 79 | 1016 | active |
| MeshAnything MeshAnything is an autoregressive transformer model that generates artist-created 3D meshes (up to 1600 faces in V2) aligned with a given s… | 31 | 1015 | active |
| deepseek-ai/DeepSeek-Math DeepSeekMath is a 7B open language model specialized in mathematical reasoning, initialized from DeepSeek-Coder and trained on 500B math-re… | 26 | 3436 | maintenance |
| clab/dynet DyNet is a C++ neural network toolkit with Python bindings, designed for efficient CPU/GPU training of networks with dynamic per-instance s… | 23 | 3436 | maintenance |
| EvolvingLMMs-Lab/Otter Otter is a multi-modal vision-language model built on OpenFlamingo, instruction-tuned on the MIMIC-IT dataset with image and video understa… | 21 | 3436 | maintenance |
| Alpha-VLLM/Lumina-Image-2.0 Lumina-Image 2.0 is an open-source 2.6B-parameter text-to-image generation framework built on a unified Next-DiT architecture with a unifie… | 58 | 1013 | active |
| Soul-AILab/SoulX-FlashHead SoulX-FlashHead is a 1.3B-parameter framework for high-fidelity, infinite-length, real-time streaming talking-head portrait video generatio… | 53 | 1011 | active |
| TensorSpeech/TensorFlowASR TensorFlowASR is a Python library implementing automatic speech recognition architectures such as DeepSpeech2, Jasper, RNN Transducer, Cont… | 66 | 1010 | active |
| sail-sg/EditAnything Edit Anything is a Python application for text-guided image editing and generation, combining Segment Anything, ControlNet, BLIP2, and Stab… | 33 | 3422 | maintenance |
| waifu2x (nunif) waifu2x is an image super-resolution and noise-reduction tool for anime-style art (and photos) using deep convolutional neural networks, or… | 78 | 3418 | maintenance |
| shaoanlu/faceswap-GAN A Jupyter Notebook-based implementation of face swapping using a denoising autoencoder architecture enhanced with adversarial losses, VGGFa… | 32 | 3416 | maintenance |
| facebookresearch/Mask2Former Mask2Former is the official PyTorch implementation of the CVPR 2022 paper 'Masked-attention Mask Transformer for Universal Image Segmentati… | 10 | 3416 | maintenance |
| davidtvs/pytorch-lr-finder A PyTorch library implementing the learning rate range test from Leslie Smith's cyclical learning rates paper, including the fastai-tweaked… | 35 | 1008 | active |
| graspnet/graspnet-baseline The official baseline deep learning model for the GraspNet-1Billion benchmark, detecting dense 6-DoF grasp poses from point clouds of clutt… | 35 | 1008 | stable |
| makegirlsmoe/makegirlsmoe_web The React front-end for MakeGirlsMoe, a web app that generates anime character portraits using a GAN model. It provides the interactive UI … | 32 | 3414 | maintenance |
| AutoArk/open-audio-opd An industrial training stack for online policy distillation (OPD) of audio models, distilling compact ASR (and planned TTS) student models … | 52 | 1007 | active |
| facebookresearch/fairscale FairScale is a PyTorch extension library providing composable modules and APIs for high-performance, large-scale distributed training, incl… | 10 | 3407 | maintenance |
| meetps/pytorch-semseg A PyTorch library implementing popular semantic segmentation architectures such as FCN, U-Net, SegNet, PSPNet, ICNet, FRRN, and LinkNet, wi… | 23 | 3402 | maintenance |
| minimaxir/gpt-2-simple A Python package that simplifies fine-tuning OpenAI's GPT-2 text-generation model (124M/355M) on custom text and generating text from the r… | 23 | 3400 | maintenance |
| RIFE RIFE is a deep learning model for real-time video frame interpolation, estimating intermediate flow between frames to generate smooth slow-… | 77 | 1004 | active |
| huawei-noah/noah-research A collection of research code subprojects released by Huawei Noah's Ark Lab, each in its own directory. It is not an official Huawei produc… | 76 | 1004 | active |
| SciSharp/TensorFlow.NET TensorFlow.NET provides .NET Standard bindings for Google's TensorFlow, aiming to implement the complete TensorFlow API in C# and F#. It in… | 24 | 3398 | maintenance |
| siyuanchen0214/Scam-AI-Multi-modal-Evaluation-System A Python-based multi-modal AI system for detecting fraudulent content across text, image, audio, and video, with provenance tracing and cro… | 39 | 1001 | active |
| pjlab-sys4nlp/llama-moe LLaMA-MoE is a Python toolkit and model series for building Mixture-of-Experts (MoE) language models from dense LLaMA models via expert con… | 19 | 1001 | active |
| catalyst-team/catalyst Catalyst is a high-level PyTorch framework for deep learning research and development, focused on reproducibility, rapid experimentation, a… | 64 | 3382 | maintenance |
| mit-han-lab/efficientvit A collection of efficient vision foundation models from MIT Han Lab, including EfficientViT backbones for perception, EfficientViT-SAM for … | 48 | 3354 | maintenance |
| eladrich/pixel2style2pixel Official PyTorch implementation of pixel2style2pixel (pSp), a StyleGAN encoder from CVPR 2021 that maps real images directly into the W+ la… | 32 | 3350 | maintenance |
| shelhamer/fcn.berkeleyvision.org Reference implementation of Fully Convolutional Networks (FCN) for semantic segmentation from the CVPR 2015 / PAMI 2016 papers, built on Ca… | 32 | 3350 | maintenance |
| tianzhi0549/FCOS Official PyTorch implementation of FCOS, a fully convolutional one-stage, anchor-free object detector published at ICCV 2019. It provides t… | 32 | 3345 | maintenance |
| tamarott/SinGAN Official PyTorch implementation of SinGAN, an ICCV 2019 best-paper generative model trained on a single natural image. It learns patch stat… | 32 | 3344 | maintenance |
| NVlabs/eg3d Official PyTorch implementation of EG3D, an efficient geometry-aware 3D generative adversarial network from NVIDIA Research (CVPR 2022). It… | 32 | 3338 | maintenance |
| HRNet/HRNet-Semantic-Segmentation Official PyTorch implementation of HRNet (High-Resolution Network) and the Segmentation Transformer (OCR) approach for semantic segmentatio… | 32 | 3331 | maintenance |
| run-youngjoo/SC-FEGAN SC-FEGAN is a GUI application that uses a generative adversarial network (SN-patchGAN discriminator with a U-Net generator) to edit face im… | 32 | 3329 | maintenance |
| JIA-Lab-research/MGM Official PyTorch implementation of Mini-Gemini, a multimodal vision-language model framework built on LLaVA that supports dense and MoE LLM… | 25 | 3327 | maintenance |
| pytorch-yolo-v3 A minimal PyTorch implementation of the YOLO v3 object detection algorithm, supporting detection on images and video with configurable reso… | 32 | 3312 | maintenance |
| PixArt-alpha/PixArt-alpha PixArt-α is a Transformer-based text-to-image diffusion model with PyTorch model definitions, pre-trained weights, and inference/training c… | 27 | 3304 | maintenance |
| open-mmlab/mmselfsup OpenMMLab's PyTorch-based toolbox and benchmark for self-supervised and unsupervised visual representation learning. It provides implementa… | 23 | 3302 | maintenance |
| bytedance/lightseq LightSeq is a high-performance CUDA-based library for training and inference of sequence models like Transformer, BERT, GPT, and BART, with… | 10 | 3295 | maintenance |
| facebookresearch/vissl VISSL is Facebook AI Research's extensible, modular and scalable PyTorch library for state-of-the-art self-supervised learning with images.… | 10 | 3293 | maintenance |
| NVIDIA/flownet2-pytorch A PyTorch implementation of FlowNet 2.0 for deep-learning-based optical flow estimation, released by NVIDIA. It provides multiple network a… | 66 | 3289 | maintenance |
| google-research/albert Official TensorFlow implementation and pretrained checkpoints of ALBERT, a lite version of BERT for self-supervised learning of language re… | 10 | 3278 | maintenance |
| google-deepmind/dm-haiku Haiku is a simple neural network library for JAX from Google DeepMind, providing an object-oriented module abstraction (hk.Module) and a fu… | 89 | 3275 | maintenance |
| zhanghang1989/ResNeSt ResNeSt is a PyTorch implementation of the Split-Attention Network, a ResNet variant that applies channel-wise attention across network bra… | 23 | 3261 | maintenance |
| anandpawara/Real_Time_Image_Animation A real-time Python application that animates a still image (e.g., a portrait) using facial motion from a live camera or video file, built o… | 32 | 3248 | maintenance |
| martinarjovsky/WassersteinGAN Reference PyTorch implementation of the Wasserstein GAN paper, providing training scripts for DCGAN and MLP architectures on datasets like … | 32 | 3243 | maintenance |
| google/model_search Model Search is a Google AutoML framework that implements neural architecture search algorithms at scale to find optimal DNN architectures … | 10 | 3238 | maintenance |
| mkocabas/VIBE Official PyTorch implementation of VIBE (CVPR 2020), a video-based method for 3D human body pose and shape estimation that predicts SMPL bo… | 23 | 3211 | maintenance |
| tomgoldstein/loss-landscape PyTorch code for visualizing the loss landscape of neural networks, from the NIPS 2018 paper 'Visualizing the Loss Landscape of Neural Nets… | 32 | 3199 | maintenance |
| tinyvision/DAMO-YOLO DAMO-YOLO is a fast and accurate object detection framework built on PyTorch, featuring NAS-searched backbones, RepGFPN, a lightweight Zero… | 32 | 3183 | maintenance |
| yusugomori/DeepLearning A multi-language (Python, C, C++, Java, Scala, Go) educational implementation of classic deep learning algorithms such as Deep Belief Nets,… | 32 | 3177 | maintenance |
| adambielski/siamese-triplet A PyTorch library implementing siamese and triplet networks for learning image embeddings, with online pair/triplet mining strategies. It p… | 32 | 3174 | maintenance |
| jettify/pytorch-optimizer torch-optimizer is a Python library providing a collection of additional optimization algorithms for PyTorch, drop-in compatible with the t… | 23 | 3171 | maintenance |
| farizrahman4u/seq2seq A sequence-to-sequence learning add-on library for Keras, providing modular encoder-decoder layers and ready-made Seq2Seq models. It suppor… | 32 | 3169 | maintenance |
| huawei-noah/Pretrained-Language-Model A collection of pretrained language models and optimization techniques from Huawei Noah's Ark Lab, including PanGu-α (200B-parameter Chines… | 32 | 3165 | maintenance |
| migueldeicaza/TensorFlowSharp TensorFlowSharp provides .NET bindings to the TensorFlow C API, exposing a strongly-typed low-level API for C# and F#. It is designed mainl… | 10 | 3147 | maintenance |
| FudanNLP/fastNLP fastNLP is a lightweight, modularized and extensible NLP framework in Python that reduces engineering boilerplate such as data processing l… | 23 | 3141 | maintenance |
| microsoft/torchscale A PyTorch library from Microsoft implementing foundation Transformer architectures such as DeepNet, Magneto, RetNet, LongNet, BitNet, and X… | 32 | 3138 | maintenance |
| open-mmlab/mmdeploy MMDeploy is the OpenMMLab model deployment framework that converts PyTorch-based OpenMMLab models (mmdetection, mmsegmentation, etc.) into … | 23 | 3137 | maintenance |
| google-deepmind/trfl TRFL is a Python library built on TensorFlow that provides building-block loss operations (e.g., Q-learning, TD learning, distributional RL… | 32 | 3131 | maintenance |
| open-mmlab/mmskeleton MMSkeleton is an OpenMMLAB toolbox for skeleton-based human understanding, built on PyTorch. It supports 2D pose estimation, skeleton-based… | 23 | 3127 | maintenance |
| diegoantognini/pyGAT A PyTorch implementation of the Graph Attention Network (GAT) model from Veličković et al. (2017), including a sparse-matrix variant. It re… | 32 | 3123 | maintenance |
| dbiir/UER-py UER-py is a PyTorch framework for pre-training transformer language models (BERT, GPT-2, T5, ELMo, etc.) and fine-tuning them on downstream… | 32 | 3112 | maintenance |
| lucasjinreal/yolov7_d2 A detectron2-based implementation of YOLOv7 that extends YOLO-style detection to instance segmentation, keypoint detection, and multi-head … | 23 | 3109 | maintenance |
| cysmith/neural-style-tf A TensorFlow implementation of neural style transfer based on Gatys et al.'s convolutional neural network approach, with support for video … | 32 | 3104 | maintenance |
| tusen-ai/simpledet SimpleDet is a Python framework built on MXNet for object detection and instance recognition. It provides state-of-the-art detection models… | 32 | 3085 | maintenance |
| Tramac/awesome-semantic-segmentation-pytorch A PyTorch library providing concise, modifiable reference implementations of many semantic segmentation models such as FCN, PSPNet, DeepLab… | 32 | 3069 | maintenance |
| stellargraph/stellargraph StellarGraph is a Python library for machine learning on graphs and networks, offering state-of-the-art graph neural network algorithms suc… | 23 | 3060 | maintenance |
| argman/EAST A TensorFlow re-implementation of the EAST (Efficient and Accurate Scene Text Detector) deep learning model for detecting text in natural s… | 32 | 3059 | maintenance |
| cvlab-columbia/zero123 Zero-1-to-3 is a research codebase and pretrained diffusion model from Columbia CVLab that changes the camera viewpoint of an object from a… | 30 | 3058 | maintenance |
| rinongal/textual_inversion Official implementation of the Textual Inversion paper, which learns new word embeddings in a frozen text-to-image (Latent Diffusion) model… | 32 | 3055 | maintenance |
| avinashpaliwal/Super-SloMo A PyTorch implementation of the Super SloMo paper for high-quality video frame interpolation, generating multiple intermediate frames to co… | 10 | 3025 | maintenance |
| microsoft/human-pose-estimation.pytorch Official PyTorch implementation of the ECCV 2018 paper 'Simple Baselines for Human Pose Estimation and Tracking' from Microsoft. It provide… | 10 | 3008 | maintenance |
| divamgupta/image-segmentation-keras A Keras library implementing popular deep learning semantic image segmentation models including SegNet, FCN, U-Net, and PSPNet. It provides… | 23 | 3003 | maintenance |
| jfzhang95/pytorch-deeplab-xception A PyTorch implementation of the DeepLab v3+ semantic segmentation model with support for multiple backbones (Xception, ResNet, MobileNet, D… | 32 | 3000 | maintenance |
| google-research/t5x T5X is a modular, composable framework built on JAX and Flax for high-performance training, evaluation, and inference of sequence models at… | 75 | 2998 | maintenance |
| yangjianxin1/GPT2-chitchat A GPT2-based Chinese chitchat dialogue model project built on HuggingFace transformers, including training, preprocessing, and interactive … | 32 | 2996 | maintenance |
| keithito/tacotron An unofficial open-source TensorFlow implementation of Google's Tacotron end-to-end neural text-to-speech model, with a pre-trained model a… | 23 | 2995 | maintenance |
| libffcv/ffcv FFCV is a fast data loading system for PyTorch that dramatically increases data throughput in model training by replacing standard data loa… | 23 | 2993 | maintenance |
| biubug6/Pytorch_Retinaface A PyTorch implementation of the RetinaFace single-stage face detection model, supporting mobilenet0.25 and resnet50 backbones with pretrain… | 32 | 2976 | maintenance |
| Tencent/FaceDetection-DSFD DSFD (Dual Shot Face Detector) is Tencent Youtu's high-accuracy face detection network, released with PyTorch inference code and pretrained… | 56 | 2969 | maintenance |
| Cartucho/mAP A Python library and script that computes mean Average Precision (mAP) for object detection models, adapted from the official PASCAL VOC 20… | 23 | 2966 | maintenance |
| johnma2006/mamba-minimal A single-file, readable PyTorch implementation of the Mamba selective state space model architecture. It produces numerically equivalent ou… | 26 | 2965 | maintenance |
| NVIDIA/MinkowskiEngine Minkowski Engine is an auto-differentiation neural network library for high-dimensional sparse tensors, built on PyTorch with CUDA accelera… | 23 | 2956 | maintenance |
| xiaofengShi/CHINESE-OCR An end-to-end Chinese scene-text OCR pipeline combining CTPN for text detection, a VGG16-based orientation classifier, and CRNN with CTC fo… | 76 | 2955 | maintenance |
| pytorch/audio TorchAudio is PyTorch's audio library providing data manipulation, transforms, and dataset loaders for audio and speech machine learning. I… | 87 | 2927 | maintenance |
| ajbrock/BigGAN-PyTorch A PyTorch implementation of BigGAN for large-scale high-fidelity image synthesis, written by the paper's first author. It supports multi-GP… | 32 | 2923 | maintenance |
| facebookresearch/XLM PyTorch implementation of Cross-lingual Language Model Pretraining (XLM) from Facebook AI Research, covering MLM, CLM, and TLM objectives p… | 10 | 2920 | maintenance |
| Conchylicultor/DeepQA DeepQA is a TensorFlow implementation of Google's 'A Neural Conversational Model', a seq2seq RNN-based deep learning chatbot. It supports t… | 32 | 2910 | maintenance |
| Tencent/PocketFlow PocketFlow is an open-source AutoML framework from Tencent AI Lab for automatically compressing and accelerating deep learning models. Deve… | 32 | 2909 | maintenance |
| haoheliu/AudioLDM AudioLDM is a Python library and research model for text-to-audio generation using latent diffusion models conditioned on CLAP embeddings. … | 43 | 2907 | maintenance |
| Luolc/AdaBound AdaBound is a PyTorch implementation of the AdaBound optimizer from an ICLR 2019 paper, which behaves like Adam early in training and gradu… | 23 | 2902 | maintenance |
| alibaba/euler Euler is a distributed graph deep learning framework developed by Alibaba for running graph neural network algorithms on large-scale graphs… | 23 | 2897 | maintenance |
| shouxieai/tensorRT_Pro A C++ library built on NVIDIA TensorRT that provides a high-level interface for high-performance deep learning inference, with Python bindi… | 23 | 2892 | maintenance |
| tensorflow/lingvo Lingvo is a TensorFlow-based framework for building neural networks, particularly sequence models, with a focus on speech recognition, mach… | 72 | 2864 | maintenance |