domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| MIC-DKFZ/batchgenerators A Python framework for data augmentation of 2D and 3D images, developed by the German Cancer Research Center for medical image classificati… | 71 | 1173 | stable |
| princeton-nlp/MeZO MeZO is a memory-efficient zeroth-order optimizer that fine-tunes language models using only forward passes, with the same memory footprint… | 29 | 1173 | stable |
| NVlabs/imaginaire NVIDIA's PyTorch library containing optimized implementations of image and video synthesis methods, including GAN-based image-to-image tran… | 32 | 4082 | maintenance |
| bytedance/1d-tokenizer A research repository from ByteDance containing code and pretrained model weights for 1D visual tokenizers (TiTok, TA-TiTok, FlowTok) and i… | 29 | 1172 | active |
| csguoh/MambaIR MambaIR and MambaIRv2 are PyTorch-based image restoration models built on Mamba state-space models, published at ECCV 2024 and CVPR 2025. T… | 54 | 1171 | active |
| magicleap/SuperGluePretrainedNetwork SuperGlue is a PyTorch implementation of a graph neural network with an optimal matching layer that matches sparse image features between t… | 32 | 4072 | maintenance |
| onnx/onnxmltools ONNXMLTools is a Python library that converts machine learning models from various toolkits (Keras/TensorFlow, scikit-learn, Core ML, XGBoo… | 78 | 1170 | active |
| baidu-research/warp-ctc A fast parallel implementation of the Connectionist Temporal Classification (CTC) loss function for CPU and CUDA GPU, with a simple C inter… | 32 | 4069 | maintenance |
| wladradchenko/wunjo.wladradchenko.ru Wunjo CE is an open-source, locally-run AI media suite for face swap, lip sync, voice cloning, object/text/background removal, restyling, a… | 70 | 1169 | active |
| iver56/torch-audiomentations A PyTorch library for fast audio data augmentation, inspired by audiomentations. It provides GPU-accelerated, differentiable audio transfor… | 58 | 1167 | active |
| SuperBruceJia/EEG-DL EEG-DL is a deep learning library built on TensorFlow for classifying EEG signals, supporting many architectures including CNNs, RNNs, GCNs… | 47 | 1167 | active |
| Object Detection Metrics A Python toolkit implementing the most popular metrics (AP, mAP, precision-recall curves) used to evaluate object detection algorithms, wit… | 50 | 1166 | stable |
| GaParmar/clean-fid Clean-FID is a PyTorch library for computing the Frechet Inception Distance (FID) with correct image resizing and quantization steps, fixin… | 48 | 1166 | stable |
| cure-lab/MagicDrive MagicDrive is the official PyTorch implementation of an ICLR 2024 paper for controllable street view generation using diffusion models with… | 35 | 1166 | active |
| sksq96/pytorch-summary A PyTorch library providing a Keras-style model.summary() that prints layer types, output shapes, parameter counts, and memory estimates. I… | 32 | 4053 | maintenance |
| facebookresearch/VideoPose3D A PyTorch implementation of CVPR 2019 research on 3D human pose estimation in video using temporal convolutions over 2D keypoint trajectori… | 10 | 4052 | maintenance |
| yosinski/deep-visualization-toolbox A GUI toolbox for visualizing and understanding deep neural networks, showing per-unit activations, backprop/deconv, and regularized-optimi… | 32 | 4051 | maintenance |
| CyberAgentAILab/TANGO TANGO is a research library from CyberAgent AI Lab that generates co-speech gesture videos by reenactment, using hierarchical audio-motion … | 39 | 1163 | active |
| facebookresearch/encodec EnCodec is a deep learning based neural audio codec from Meta AI that compresses mono 24 kHz and stereo 48 kHz audio to bitrates from 1.5 t… | 32 | 4041 | maintenance |
| 3D ResNets for Action Recognition A PyTorch implementation of 3D ResNet and R(2+1)D models for video action recognition, accompanying CVPR 2018 and related papers. It includ… | 23 | 4038 | maintenance |
| Linaom1214/TensorRT-For-YOLO-Series A Python and C++ toolkit for running YOLO-series object detection models (YOLOv3 through YOLOv12, YOLOX) with NVIDIA TensorRT, including ON… | 44 | 1162 | active |
| MCG-NKU/E2FGVI E2FGVI is the official PyTorch implementation of the CVPR 2022 paper 'Towards An End-to-End Framework for Flow-Guided Video Inpainting'. It… | 32 | 1161 | stable |
| meta-pytorch/torchcodec TorchCodec is a PyTorch-native library for decoding and encoding videos, audio, and images into PyTorch tensors on CPU and CUDA GPU, built … | 87 | 1160 | active |
| minivision-ai/photo2cartoon A Python deep-learning project from Minivision that converts real portrait photos into cartoon-style avatars using unpaired image translati… | 32 | 4029 | maintenance |
| DepthAnything/PromptDA Prompt Depth Anything is a Python library implementing a CVPR 2025 method for high-resolution (up to 4K) accurate metric depth estimation. … | 50 | 1159 | active |
| LibCity/Bigscity-LibCity LibCity is an open-source PyTorch library for urban spatial-temporal data mining, providing a unified pipeline for traffic prediction resea… | 23 | 1157 | active |
| fundamentalvision/Deformable-DETR Official PyTorch implementation of Deformable DETR, an efficient end-to-end object detector that uses deformable attention to fix DETR's sl… | 32 | 4015 | maintenance |
| sirius-ai/LPRNet_Pytorch A PyTorch implementation of LPRNet, a lightweight deep neural network for license plate recognition. It ships with pretrained weights focus… | 32 | 1156 | stable |
| gemelo-ai/vocos Vocos is a fast neural vocoder that synthesizes audio waveforms from acoustic features such as mel-spectrograms or EnCodec tokens. It uses … | 65 | 1155 | stable |
| deepmodeling/Uni-Mol Uni-Mol is a collection of 3D molecular representation learning frameworks and pretrained models for tasks like molecule property predictio… | 34 | 1155 | active |
| JunMa11/SegLossOdyssey A curated collection of loss functions for medical image segmentation, accompanying the 'Loss Odyssey in Medical Image Segmentation' survey… | 32 | 4007 | maintenance |
| SystemErrorWang/White-box-Cartoonization Official TensorFlow implementation of the CVPR 2020 paper 'Learning to Cartoonize Using White-box Cartoon Representations', which converts … | 61 | 4001 | maintenance |
| quark0/darts DARTS is the official PyTorch implementation of the ICLR 2019 paper 'DARTS: Differentiable Architecture Search', which performs neural arch… | 32 | 3997 | maintenance |
| PKU-Alignment/omnisafe OmniSafe is a PyTorch-based infrastructural framework for safe reinforcement learning research, providing a unified modular toolkit and com… | 27 | 1149 | active |
| JDAI-CV/fast-reid FastReID is a PyTorch-based research platform implementing state-of-the-art re-identification algorithms for persons, vehicles, and faces. … | 23 | 3981 | maintenance |
| open-gigaai/giga-world-1 GigaWorld-1 is an open-source framework providing training, inference, data processing, checkpoint conversion, and LoRA merge workflows for… | 54 | 1147 | active |
| ShiqiYu/OpenGait OpenGait is a flexible and extensible Python framework for gait recognition research, providing implementations of state-of-the-art models … | 67 | 1146 | active |
| sgl-project/SpecForge SpecForge is a Python framework from the SGLang team for training speculative decoding models such as EAGLE/EAGLE3 draft heads. Trained mod… | 64 | 1145 | active |
| extropic-ai/thrml THRML is a JAX library for building and sampling probabilistic graphical models, focused on efficient block Gibbs sampling of energy-based … | 71 | 1144 | active |
| facebookresearch/fairseq2 fairseq2 is a PyTorch-based sequence modeling toolkit from Meta FAIR for training custom models for content generation tasks such as langua… | 89 | 1143 | active |
| callous-youth/BOAT BOAT is a PyTorch-based library providing a compositional, operation-level toolbox for gradient-based bi-level optimization (BLO). It decom… | 87 | 1143 | active |
| HengyiWang/spann3r Spann3R is a transformer-based model for dense 3D reconstruction from ordered or unordered image collections, built on the DUSt3R paradigm.… | 26 | 1141 | active |
| cvg/glue-factory Glue Factory is a PyTorch-based library for training and evaluating deep neural networks that detect and match local visual features (point… | 69 | 1140 | active |
| rohitgandikota/sliders Official implementation of Concept Sliders, LoRA adaptors that enable precise, plug-and-play control of attributes in diffusion models like… | 52 | 1139 | active |
| clovaai/deep-text-recognition-benchmark Official PyTorch implementation of a four-stage scene text recognition (OCR) framework from an ICCV 2019 paper, with training and evaluatio… | 32 | 3942 | maintenance |
| andabi/deep-voice-conversion A TensorFlow implementation of deep neural networks for voice conversion (voice style transfer) that converts a source speaker's voice into… | 32 | 3938 | maintenance |
| chengtan9907/OpenSTL OpenSTL is a comprehensive benchmark and modular framework for spatio-temporal predictive learning, covering video prediction methods acros… | 54 | 1137 | active |
| horseee/LLM-Pruner LLM-Pruner is a PyTorch library implementing structural pruning of large language models based on gradient information, as published at Neu… | 29 | 1136 | active |
| IST-DASLab/marlin Marlin is a highly optimized FP16xINT4 matrix multiplication CUDA kernel for LLM inference that achieves near-ideal 4x speedups at batch si… | 26 | 1136 | active |
| Dao-AILab/quack QuACK is a collection of high-performance GPU kernels (RMSNorm, LayerNorm, softmax, cross-entropy, GEMM with epilogues) written in NVIDIA's… | 85 | 1135 | active |
| rhymes-ai/Allegro Allegro is an open-source text-to-video generation model that produces high-quality 720p videos up to 6 seconds at 15 FPS from text prompts… | 24 | 1135 | active |
| OpenGVLab/SAM-Med2D Official implementation of SAM-Med2D, a fine-tuned Segment Anything Model (SAM) for 2D medical image segmentation, trained on the SA-Med2D-… | 28 | 1134 | active |
| Janspiry/Image-Super-Resolution-via-Iterative-Refinement An unofficial PyTorch implementation of SR3 (Image Super-Resolution via Iterative Refinement), a diffusion-based model for image super-reso… | 32 | 3923 | maintenance |
| zai-org/SCAIL-2 Official implementation of SCAIL-2, an open-source model for end-to-end controlled character animation that drives character videos from re… | 58 | 1132 | active |
| mgonzs13/yolo_ros A ROS 2 wrapper for Ultralytics YOLO models (YOLOv8 through YOLO26) providing object detection, tracking, instance segmentation, human pose… | 93 | 1131 | active |
| sooftware/conformer An unofficial PyTorch implementation of the Conformer architecture (convolution-augmented Transformer) from the INTERSPEECH 2020 paper, tar… | 72 | 1131 | active |
| noahcao/OC_SORT OC-SORT is a pure motion-model-based multi-object tracker for video, improving on SORT by fixing Kalman filter limitations to handle occlus… | 67 | 1131 | stable |
| Tencent/hpc-ops HPC-Ops is a production-grade C++/CUDA operator library for high-performance LLM inference, developed by Tencent's Hunyuan AI Infra team. I… | 59 | 1131 | active |
| PriesiaMioShirakana/DragonianVoice A C++ inference library for running ONNX-based TTS, SVC (singing voice conversion), and SVS (singing voice synthesis) models, supporting ar… | 40 | 1129 | active |
| ikostrikov/pytorch-a2c-ppo-acktr-gail A PyTorch implementation of several deep reinforcement learning algorithms: A2C, PPO, ACKTR, and GAIL (imitation learning). It works with O… | 32 | 3903 | maintenance |
| Dobiasd/frugally-deep frugally-deep is a lightweight header-only C++ library for running inference (forward passes) on Keras/TensorFlow models without linking ag… | 85 | 1128 | active |
| espressif/esp-dl ESP-DL is Espressif's lightweight neural network inference framework for ESP-series chips, with a custom .espdl model format, quantization … | 72 | 1125 | active |
| OpenGVLab/VideoMamba VideoMamba is a state space model (Mamba-based) architecture for efficient video understanding, released with code and pretrained models fr… | 25 | 1125 | active |
| geopavlakos/hamer HaMeR (Hand Mesh Recovery) is a transformer-based model that reconstructs 3D hand meshes from single monocular images using the MANO parame… | 56 | 1124 | active |
| xverse-engine/XScene-UEPlugin An Unreal Engine 5 plugin for real-time visualization, management, editing, and scalable hybrid rendering of 3D Gaussian Splatting models. … | 40 | 1121 | active |
| THUDM/SwissArmyTransformer SwissArmyTransformer (sat) is a PyTorch library for developing custom Transformer model variants where models like BERT, GPT, T5, GLM, and … | 23 | 1121 | active |
| FlagAI-Open/FlagAI FlagAI is a Python toolkit for training, fine-tuning, and deploying large-scale AI models across NLP, CV, and vision-language tasks. It int… | 64 | 3869 | maintenance |
| PRIME-RL/TTRL TTRL is an open-source implementation of Test-Time Reinforcement Learning, a method for training LLMs with RL on unlabeled data using major… | 51 | 1120 | active |
| PaddlePaddle/PaddleHelix PaddleHelix is a bio-computing platform built on PaddlePaddle featuring large-scale representation learning and multi-task deep learning fo… | 57 | 1119 | active |
| alibaba-damo-academy/RynnVLA-002 RynnVLA-002 is a unified autoregressive Vision-Language-Action and world model that generates robot actions from text and image observation… | 43 | 1119 | active |
| caiyuanhao1998/MST A Python toolbox for spectral compressive imaging reconstruction that implements over 15 algorithms including MST, CST, DAUHST, BiSCI, HDNe… | 53 | 1118 | active |
| yangxue0827/RotationDetection AlphaRotate is a TensorFlow-based benchmark and toolbox for rotated (oriented) object detection, implementing detectors such as R2CNN, Reti… | 23 | 1118 | active |
| Lasagne/Lasagne Lasagne is a lightweight Python library for building and training neural networks on top of Theano. It supports feed-forward, convolutional… | 23 | 3857 | maintenance |
| HITsz-TMG/Uni-MoE Uni-MoE is a family of open-source Mixture-of-Experts (MoE) based omnimodal large language models that understand and generate across text,… | 68 | 1116 | active |
| NTMC-Community/MatchZoo MatchZoo is a Python toolkit for designing, comparing, and sharing deep text matching models. It provides a unified data pipeline, pre-buil… | 23 | 3849 | maintenance |
| yahoo/TensorFlowOnSpark TensorFlowOnSpark is a Python library that lets existing TensorFlow programs run distributed training and inference on Apache Spark and Had… | 23 | 3845 | maintenance |
| openai/improved-diffusion The official codebase for OpenAI's Improved Denoising Diffusion Probabilistic Models paper, providing a Python package for training and sam… | 32 | 3844 | maintenance |
| NVIDIA/earth2studio Earth2Studio is a Python deep-learning framework from NVIDIA for building, exploring, and deploying AI-driven weather and climate workflows… | 88 | 1112 | active |
| AdaptiveMotorControlLab/CEBRA CEBRA is a Python library for self-supervised learning of consistent latent embeddings from high-dimensional time-series recordings, using … | 86 | 1111 | active |
| baofff/U-ViT U-ViT is the official PyTorch implementation of a ViT-based backbone architecture for diffusion models from the CVPR 2023 paper 'All are Wo… | 32 | 1110 | stable |
| THU-MIG/RepViT Official PyTorch implementation of RepViT, a family of lightweight CNNs designed by integrating efficient ViT architectural designs into Mo… | 19 | 1108 | stable |
| OpenMOSS/MOVA MOVA is an open-source foundation model and toolkit for joint video-audio generation, synthesizing synchronized video and audio in a single… | 60 | 1106 | active |
| rusty1s/pytorch_sparse A PyTorch extension library providing optimized sparse matrix operations (coalesce, transpose, sparse-dense and sparse-sparse multiplicatio… | 61 | 1104 | active |
| TencentARC/T2I-Adapter Official implementation of T2I-Adapter, lightweight adapter models that add controllable conditioning (sketch, canny, lineart, depth, pose)… | 31 | 3801 | maintenance |
| yangheng95/PyABSA PyABSA is a PyTorch-based library providing state-of-the-art models for aspect-based sentiment analysis, including aspect term extraction, … | 67 | 1102 | active |
| MIT-SPARK/VGGT-SLAM VGGT-SLAM is a dense RGB SLAM system that performs real-time feed-forward 3D scene reconstruction, optimizing on the SL(4) manifold using t… | 59 | 1102 | active |
| MoonshotAI/MoonEP MoonEP is an Expert Parallelism communication library for Mixture-of-Experts training that keeps token loads perfectly balanced across rank… | 56 | 1101 | active |
| zai-org/CogView4 CogView4, CogView3-Plus and CogView3 are open-source text-to-image generation models from Zhipu AI, with CogView4 being a 6B-parameter DiT-… | 28 | 1101 | active |
| naturomics/CapsNet-Tensorflow A TensorFlow implementation of CapsNet (Capsule Networks) based on Geoffrey Hinton's paper 'Dynamic Routing Between Capsules'. It supports … | 32 | 3786 | maintenance |
| yongliang-wu/DFT DFT (Dynamic Fine-Tuning) is the official implementation of an ICLR 2026 paper that improves Supervised Fine-Tuning of LLMs by dynamically … | 61 | 1100 | active |
| BytedTsinghua-SIA/MemAgent MemAgent is a reinforcement-learning framework for training LLM agents that process arbitrarily long contexts via a memory mechanism within… | 55 | 1100 | active |
| lucidrains/stylegan2-pytorch A simple PyTorch implementation of StyleGAN2, a state-of-the-art generative adversarial network, trainable entirely from the command line w… | 23 | 3783 | maintenance |
| luchris429/purejaxrl PureJaxRL provides end-to-end reinforcement learning training pipelines implemented entirely in JAX, including environments, enabling massi… | 30 | 1099 | active |
| LAMDA-CL/PyCIL PyCIL is a PyTorch-based Python toolbox for class-incremental learning, implementing the largest collection of CIL methods for reproducible… | 52 | 1098 | active |
| WangLibo1995/GeoSeg GeoSeg is an open-source PyTorch-based semantic segmentation toolbox focused on Vision Transformers for remote sensing imagery, featuring t… | 32 | 1096 | active |
| hkchengrex/Cutie Cutie is a video object segmentation framework with object-level memory reading, a follow-up to XMem offering better consistency, robustnes… | 18 | 1095 | active |
| Eyeline-Labs/Go-with-the-Flow Official implementation of the CVPR 2025 Oral paper 'Go-with-the-Flow', which controls motion in video diffusion models by replacing i.i.d.… | 41 | 1093 | active |
| SCLBD/DeepfakeBench DeepfakeBench is a comprehensive benchmark framework for deepfake detection, providing a unified platform for data management, implementati… | 36 | 1093 | active |
| yandex/YaLM-100B YaLM-100B is a GPT-like pretrained language model with 100 billion parameters, trained by Yandex on English and Russian text using DeepSpee… | 32 | 3757 | maintenance |
| welltop-cn/ComfyUI-TeaCache A ComfyUI plugin integrating TeaCache, a training-free caching method that accelerates diffusion model inference by exploiting output diffe… | 35 | 1092 | active |