domain: deep-learning
2771 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| ucb-bar/gemmini Gemmini is Berkeley's open-source generator for parameterizable systolic-array DNN hardware accelerators, written in Chisel (Scala) and int… | 63 | 1436 | active |
| NVlabs/Fast-FoundationStereo Fast-FoundationStereo is NVIDIA's official PyTorch implementation of a real-time zero-shot stereo matching model family, accepted to CVPR 2… | 54 | 1432 | active |
| Francis-Rings/StableAnimator StableAnimator is an end-to-end ID-preserving video diffusion framework that animates a reference human image according to a sequence of po… | 41 | 1430 | active |
| YanjieZe/3D-Diffusion-Policy 3D Diffusion Policy (DP3) is a visual imitation learning algorithm that combines compact 3D point cloud representations with diffusion poli… | 46 | 1429 | active |
| tianweiy/CausVid CausVid is a research codebase implementing a fast autoregressive video diffusion model distilled from a bidirectional diffusion transforme… | 36 | 1426 | active |
| mdbloice/Augmentor Augmentor is a standalone Python library for image augmentation in machine learning, providing a pipeline of stochastic operations like rot… | 32 | 5133 | maintenance |
| numz/sd-wav2lip-uhq A Wav2Lip Studio extension for the Stable Diffusion WebUI (Automatic1111) that generates high-quality lip-synced talking-face videos from a… | 28 | 1424 | active |
| deepseek-ai/EPLB EPLB is DeepSeek's open-source Expert Parallelism Load Balancer for Mixture-of-Experts models. It computes balanced expert replication and … | 26 | 1424 | active |
| ZheC/Realtime_Multi-Person_Pose_Estimation Reference implementation of the CVPR'17 paper 'Realtime Multi-Person Pose Estimation', a bottom-up approach that detects keypoints for mult… | 32 | 5123 | maintenance |
| NVIDIA/DLSS NVIDIA DLSS is the public SDK repository for NVIDIA's RTX Deep Learning Super Sampling, a neural network that boosts game frame rates and g… | 87 | 1421 | active |
| facebookresearch/vggsfm VGGSfM is a deep learning-based Structure from Motion pipeline from Meta AI and Oxford VGG that recovers camera poses and 3D point clouds f… | 29 | 1421 | active |
| jrzaurin/pytorch-widedeep A PyTorch library for multimodal deep learning that combines tabular data with text and images using Wide and Deep model architectures. It … | 62 | 1416 | active |
| affinelayer/pix2pix-tensorflow A TensorFlow implementation of pix2pix, a conditional GAN that learns a mapping from input images to output images. It is a faithful port o… | 32 | 5081 | maintenance |
| Lyken17/pytorch-OpCounter THOP (PyTorch-OpCounter) is a Python library that counts the MACs/FLOPs and parameters of PyTorch models. It profiles arbitrary nn.Modules … | 32 | 5078 | maintenance |
| lucidrains/self-rewarding-lm-pytorch A PyTorch library implementing the Self-Rewarding Language Model training framework from MetaAI, along with the SPIN training method. It pr… | 16 | 1411 | active |
| explosion/spacy-transformers A spaCy v3 extension package that provides pipeline components for using pretrained transformer models like BERT, RoBERTa, XLNet, and GPT-2… | 71 | 1409 | stable |
| nv-tlabs/GEN3C GEN3C is NVIDIA's research codebase for a generative video model that achieves precise camera control and temporal 3D consistency using a 3… | 59 | 1409 | active |
| gnobitab/InstaFlow InstaFlow is a one-step text-to-image generation model based on Rectified Flow, enabling ultra-fast Stable Diffusion inference without iter… | 28 | 1409 | active |
| mratsim/Arraymancer Arraymancer is a fast, ergonomic N-dimensional tensor (ndarray) library written in Nim, inspired by NumPy and PyTorch. It provides CPU, CUD… | 61 | 1407 | active |
| XiaoMi/mace MACE (Mobile AI Compute Engine) is a deep learning inference framework optimized for mobile heterogeneous computing on Android, iOS, Linux … | 23 | 5046 | maintenance |
| dexmal/dexbotic Dexbotic is an open-source PyTorch-based toolbox for developing Vision-Language-Action (VLA) models for embodied intelligence. It unifies p… | 72 | 1403 | active |
| dailenson/SDT Official PyTorch implementation of the CVPR 2023 paper 'Disentangling Writer and Character Styles for Handwriting Generation' (SDT). It gen… | 43 | 1403 | active |
| fudan-generative-vision/hallo3 Hallo3 is a research model from Fudan University that animates a single portrait image into a highly dynamic and realistic talking-head vid… | 26 | 1401 | active |
| nachifur/MulimgViewer MulimgViewer is a Python-based multi-image viewer that displays many images in a single interface for side-by-side comparison, parallel sel… | 66 | 1400 | active |
| Zejun-Yang/AniPortrait AniPortrait is a Python framework from Tencent that generates photorealistic portrait animations from an audio clip and a reference image, … | 25 | 5021 | maintenance |
| open-gigaai/giga-world-policy GigaWorld-Policy is a World Action Model (WAM) for robot policy learning that jointly models actions and future visual observations during … | 59 | 1398 | active |
| yfeng95/PRNet PRNet is a Python/TensorFlow implementation of the ECCV 2018 Position Map Regression Network for joint 3D face reconstruction and dense ali… | 32 | 5013 | maintenance |
| lucidrains/transfusion-pytorch A PyTorch implementation of Transfusion, MetaAI's approach to predicting the next token and diffusing images with a single multi-modal mode… | 75 | 1395 | active |
| xl0/lovely-tensors A Python library that provides human-friendly summaries of PyTorch tensors, showing shape, statistics, histograms, and NaN/Inf detection in… | 66 | 1392 | active |
| logtd/ComfyUI-Fluxtapoz A set of ComfyUI custom nodes for editing and stylizing images with Flux models, implementing techniques like RF-Inversion, RF-Edit, Firefl… | 23 | 1392 | active |
| ARahim3/mlx-tune A Python library for fine-tuning LLMs, vision-language, audio (TTS/STT), embedding, OCR, and JEPA models natively on Apple Silicon Macs usi… | 75 | 1389 | active |
| zju3dv/street_gaussians Street Gaussians is a research implementation of the ECCV 2024 paper 'Modeling Dynamic Urban Scenes with Gaussian Splatting', which reconst… | 40 | 1388 | active |
| Junyi42/monst3r MonST3R is the official PyTorch implementation of an ICLR 2025 paper that estimates per-timestep geometry (pointmaps) from dynamic videos i… | 36 | 1386 | active |
| cszn/BSRGAN BSRGAN is a PyTorch implementation of a practical degradation model for deep blind image super-resolution, presented at ICCV 2021. It provi… | 32 | 1386 | stable |
| zhixuhao/unet A Keras implementation of the U-Net convolutional network architecture for image segmentation, based on the original biomedical segmentatio… | 66 | 4941 | maintenance |
| autonomousvision/unimatch UniMatch is a PyTorch research library implementing a unified transformer-based model for optical flow, stereo matching, and depth estimati… | 32 | 1379 | stable |
| yanx27/Pointnet_Pointnet2_pytorch A pure PyTorch implementation of the PointNet and PointNet++ deep learning architectures for point cloud processing. It includes training a… | 32 | 4936 | maintenance |
| AlmondGod/tinyworlds A minimal Python implementation of DeepMind's Genie autoregressive world model, including a video tokenizer, action tokenizer, and dynamics… | 54 | 1378 | active |
| keyu-tian/SparK SparK is the official PyTorch implementation of an ICLR 2023 Spotlight paper that applies BERT/MAE-style masked image modeling to convoluti… | 22 | 1376 | stable |
| qubvel/segmentation_models A Python library providing neural network architectures for image segmentation (Unet, FPN, Linknet, PSPNet) built on Keras and TensorFlow K… | 23 | 4923 | maintenance |
| minimaxir/textgenrnn A Python 3 library built on Keras/TensorFlow for easily training char-rnn style neural networks that generate text from any dataset in a fe… | 23 | 4922 | maintenance |
| Zefan-Cai/KVCache-Factory KVCache-Factory is a unified Python framework implementing many KV cache compression, retrieval, merging, and quantization methods for long… | 68 | 1375 | active |
| haoheliu/voicefixer VoiceFixer is a Python library and CLI tool for general speech restoration, using a pretrained neural vocoder to restore degraded human spe… | 26 | 1373 | stable |
| macanv/BERT-BiLSTM-CRF-NER A TensorFlow implementation of named entity recognition that fine-tunes Google BERT with a BiLSTM-CRF model, primarily targeting Chinese te… | 32 | 4906 | maintenance |
| pnnl/neuromancer NeuroMANCER is a PyTorch-based differentiable programming library for solving parametric constrained optimization problems, physics-informe… | 72 | 1369 | active |
| BICLab/SpikingBrain-7B SpikingBrain-7B is a brain-inspired large language model that combines hybrid efficient attention, MoE modules, and spike encoding, with a … | 54 | 1369 | active |
| ali-vilab/TeaCache TeaCache is a training-free caching approach that accelerates inference for video diffusion models by estimating output differences across … | 33 | 1369 | active |
| thunlp/OpenPrompt OpenPrompt is a PyTorch-based open-source framework for prompt-learning, providing a standard, flexible pipeline of templates and verbalize… | 23 | 4890 | maintenance |
| OpenPPL/ppl.nn PPLNN is a high-performance deep-learning inference engine written in C++ that runs ONNX models on x86 CPUs and NVIDIA GPUs, with a dedicat… | 32 | 1367 | active |
| xandergos/terrain-diffusion Terrain Diffusion is a Python framework that uses diffusion models as a learned, deterministic replacement for Perlin noise, generating inf… | 64 | 1365 | active |
| NVIDIA-AI-IOT/torch2trt torch2trt is a Python library that converts PyTorch models to TensorRT engines using the TensorRT Python API, with a simple single-function… | 23 | 4878 | maintenance |
| ImprintLab/MedSegDiff MedSegDiff is a diffusion probabilistic model framework for segmenting and reconstructing organs and tissues from medical images, with a tr… | 50 | 1363 | active |
| kengz/SLM-Lab SLM Lab is a modular deep reinforcement learning framework built in PyTorch, offering ready-to-use algorithms like PPO, SAC, DQN, and A2C v… | 95 | 1362 | active |
| bytedance/UNO UNO is a research framework from ByteDance for subject-driven image generation with diffusion transformers, supporting both single- and mul… | 38 | 1362 | active |
| MoonInTheRiver/DiffSinger Official PyTorch implementation of DiffSinger, an AAAI 2022 paper on singing voice synthesis and text-to-speech using a shallow diffusion m… | 65 | 4851 | maintenance |
| huggingface/finetrainers finetrainers is a Hugging Face library for scalable, memory-optimized training (fine-tuning) of diffusion models, including LoRA training o… | 62 | 1358 | active |
| Sense-X/Co-DETR Co-DETR is a PyTorch implementation of DETRs with Collaborative Hybrid Assignments Training, an ICCV 2023 object detection and instance seg… | 32 | 1357 | stable |
| ant-research/CoDeF CoDeF is the official PyTorch implementation of Content Deformation Fields, a video representation combining a canonical content field and … | 28 | 4846 | maintenance |
| blei-lab/edward Edward is a Python library for probabilistic modeling, inference, and criticism built on TensorFlow. It supports deep generative models, va… | 23 | 4843 | maintenance |
| reiniscimurs/DRL-robot-navigation A ROS Gazebo simulation project that trains a mobile robot to navigate to random goals while avoiding obstacles using a TD3 deep reinforcem… | 58 | 1356 | active |
| bytedance/flux Flux is a GPU kernel library from ByteDance that overlaps computation with communication for tensor and expert parallelism in dense and MoE… | 33 | 1354 | active |
| yinguobing/head-pose-estimation A Python library for realtime human head pose estimation using ONNX Runtime and OpenCV. It combines face detection (SCRFD), 68-point facial… | 23 | 1353 | stable |
| shivammehta25/Matcha-TTS Matcha-TTS is a PyTorch-based text-to-speech system that uses conditional flow matching for fast, non-autoregressive speech synthesis. It s… | 62 | 1349 | active |
| wyhuai/DDNM DDNM is a Python research codebase implementing the Denoising Diffusion Null-Space Model for zero-shot image restoration, published as an I… | 32 | 1349 | stable |
| FreedomIntelligence/HuatuoGPT-o1 HuatuoGPT-o1 is a medical large language model project for complex medical reasoning, with open-sourced models (7B-72B), fine-tuning data, … | 22 | 1349 | active |
| MegEngine/MegEngine MegEngine is a fast, scalable deep learning framework with automatic differentiation, developed in C++ with Python bindings. It unifies tra… | 23 | 4808 | maintenance |
| sjvasquez/handwriting-synthesis A Python implementation of Alex Graves' handwriting synthesis experiments using recurrent neural networks, generating realistic handwritten… | 32 | 4802 | maintenance |
| PKU-VCL-3DV/SLAM3R SLAM3R is a real-time dense 3D scene reconstruction system that regresses 3D points from monocular RGB video using feed-forward neural netw… | 42 | 1344 | active |
| hao-ai-lab/LookaheadDecoding A Python library implementing Lookahead Decoding, an exact parallel decoding algorithm that accelerates LLM inference without a draft model… | 31 | 1342 | active |
| FireRedTeam/FireRed-Image-Edit FireRed-Image-Edit is an open-source image editing foundation model built on diffusion models, released as PyTorch model weights with infer… | 49 | 1341 | active |
| alibaba/graph-learn Graph-Learn (formerly AliGraph) is a distributed framework for developing and applying large-scale graph neural networks, with a training l… | 36 | 1341 | active |
| rstudio/tensorflow An R package that provides full access to the TensorFlow API from R via reticulate, bridging R users to TensorFlow's Python implementation.… | 61 | 1339 | active |
| senguptaumd/Background-Matting Official research code for 'Background Matting: The World is Your Green Screen' (CVPR 2020), a deep network that extracts per-pixel alpha m… | 32 | 4769 | maintenance |
| ByteDance-Seed/SeedVR SeedVR/SeedVR2 are diffusion-transformer based models for generic real-world and AIGC video and image restoration, with SeedVR2 using adver… | 47 | 1334 | active |
| segment-any-text/wtpsplit wtpsplit is a Python toolkit for segmenting text into sentences or other semantic units using the SaT and WtP deep learning models. It prov… | 86 | 1333 | active |
| mapillary/inplace_abn A PyTorch extension library implementing In-Place Activated BatchNorm (InPlace-ABN), which redefines BN plus nonlinear activation as a sing… | 65 | 1333 | stable |
| jonathan-laurent/AlphaZero.jl A generic, simple, and fast Julia implementation of DeepMind's AlphaZero algorithm for training game-playing agents via self-play and MCTS.… | 64 | 1333 | active |
| wenqsun/DimensionX DimensionX is a research framework that generates photorealistic 3D and 4D scenes from a single image using controllable video diffusion mo… | 43 | 1333 | active |
| Vahe1994/AQLM Official PyTorch implementation of AQLM, an extreme LLM compression method via additive quantization, extended with PV-Tuning for finetunin… | 57 | 1329 | active |
| Tongjilibo/bert4torch bert4torch is a PyTorch library providing an elegant reimplementation of transformer models (BERT, RoBERTa, T5, GPT, ChatGLM, LLaMA, etc.) … | 82 | 1328 | active |
| facebookincubator/AITemplate AITemplate is a Python framework that compiles deep neural networks into high-performance CUDA (NVIDIA) or HIP (AMD) C++ code for fast fp16… | 66 | 4724 | maintenance |
| LLaVA-VL/LLaVA-NeXT LLaVA-NeXT is a collection of open large multimodal models (LLaVA-NeXT, LLaVA-Video, LLaVA-OneVision, LLaVA-Critic-R1) that combine vision … | 64 | 4716 | maintenance |
| agemagician/ProtTrans ProtTrans provides state-of-the-art pre-trained Transformer language models for protein sequences, trained on thousands of GPUs and hundred… | 33 | 1324 | active |
| sjtuytc/UnboundedNeRFPytorch A PyTorch implementation benchmarking state-of-the-art unbounded (large-scale) neural radiance field methods like NeRF++, DVGO, and Block-N… | 23 | 1324 | active |
| tensorflow/lucid Lucid is a collection of infrastructure and tools for research in neural network interpretability, built on TensorFlow 1.x. It provides fea… | 10 | 4704 | maintenance |
| galilai-group/lejepa LeJEPA is a Python framework for scalable, theoretically grounded self-supervised representation learning based on Joint-Embedding Predicti… | 45 | 1322 | active |
| ImprintLab/Medical-SAM-Adapter Medical SAM Adapter (MSA) is a Python framework that fine-tunes Meta's Segment Anything Model for medical image segmentation using lightwei… | 39 | 1322 | active |
| meta-pytorch/segment-anything-fast A fast, batched offline inference-oriented fork of Meta's Segment Anything (SAM) image segmentation model. It applies optimizations like bf… | 45 | 1321 | active |
| stared/livelossplot A Python library that draws live training loss and metric plots inside Jupyter Notebooks for Keras, PyTorch, and other deep learning framew… | 91 | 1319 | stable |
| sanchit-gandhi/whisper-jax An optimized JAX implementation of OpenAI's Whisper speech recognition model, built on Hugging Face Transformers, offering up to 70x faster… | 30 | 4682 | maintenance |
| open-edge-platform/geti Geti is an open-source, end-to-end Vision AI application from Intel that takes users from raw images to deployed computer vision models, ru… | 98 | 1317 | active |
| sicara/easy-few-shot-learning A Python library (easyfsl) with ready-to-use code and tutorial notebooks for few-shot image classification and meta-learning, built on PyTo… | 23 | 1313 | stable |
| yeyupiaoling/VoiceprintRecognition-Pytorch A PyTorch-based voiceprint recognition (speaker recognition) framework implementing models such as ECAPA-TDNN, ResNetSE, ERes2Net, and CAM+… | 58 | 1312 | active |
| albermax/innvestigate iNNvestigate is a Python toolbox providing a common interface and out-of-the-box implementations of many neural network explanation methods… | 30 | 1309 | active |
| ChenmienTan/RL2 RL2 (Ray Less Reinforcement Learning) is a concise Python library for post-training large language models with reinforcement learning, SFT,… | 57 | 1307 | active |
| huawei-noah/Efficient-Computing A collection of efficient deep learning methods from Huawei Noah's Ark Lab, covering model compression, knowledge distillation, pruning, qu… | 32 | 1307 | active |
| inference-labs-inc/JSTprove JSTprove is a Rust CLI toolkit that generates zero-knowledge proofs of machine learning inference on ONNX models, built on Polyhedra Networ… | 70 | 1306 | active |
| Vincentqyw/image-matching-webui A Gradio-based web UI that matches keypoints between two images using many state-of-the-art image matching algorithms (LoFTR, SuperGlue, Li… | 91 | 1302 | active |
| luosiallen/latent-consistency-model Official implementation of Latent Consistency Models (LCM), a diffusion-based approach for synthesizing high-resolution images with few-ste… | 27 | 4615 | maintenance |
| STVIR/pysot PySOT is a Python research platform by SenseTime for single object visual tracking, implementing algorithms such as SiamRPN, SiamRPN++, DaS… | 45 | 4600 | maintenance |