Open-source deep-learning software
2663 open-source libraries, tools and applications tagged deep-learning, ranked by how closely deep-learning is their primary purpose, then by adoption and maintenance health. Each entry links to a page with sources, use cases and when to choose or avoid it.
2663 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| zhaoweicai/cascade-rcnn A C++/Caffe implementation of Cascade R-CNN and other popular two-stage object detection frameworks such as Faster R-CNN, R-FCN, and FPN. I… | 32 | 1059 | maintenance |
| BurguerJohn/Dain-App Dain-App is a video frame interpolation application built on the DAIN (Depth-Aware Video Frame Interpolation) deep learning model. It provi… | 23 | 1058 | maintenance |
| chenyuntc/PyTorchText A PyTorch implementation of multiple text classification models (TextCNN, TextRNN/LSTM, RCNN, FastText, inception CNN) that won 1st place i… | 32 | 1057 | maintenance |
| piergiaj/pytorch-i3d A PyTorch port of DeepMind's I3D (Inflated 3D ConvNet) models pretrained on the Kinetics dataset for video action recognition. It includes … | 32 | 1057 | maintenance |
| HRNet/HRNet-Image-Classification Official PyTorch implementation and training code for HRNet (High-Resolution Network) image classification models on ImageNet. It provides … | 23 | 1057 | maintenance |
| damo-cv/TransReID Official PyTorch implementation of TransReID, an ICCV 2021 paper applying vision transformers to object re-identification. It provides trai… | 32 | 1056 | maintenance |
| raminmh/CfC Reference implementations of Closed-form Continuous-time (CfC) neural networks, a fast closed-form approximation of liquid time-constant ne… | 23 | 1056 | maintenance |
| ashkamath/mdetr MDETR (Modulated Detection) is a PyTorch research codebase for end-to-end multi-modal object detection that grounds free-form text queries … | 32 | 1054 | maintenance |
| microsoft/Oscar Oscar is Microsoft's research code for object-semantics aligned cross-modal pre-training of vision-language models, with VinVL providing im… | 10 | 1054 | maintenance |
| 4uiiurz1/pytorch-nested-unet A PyTorch implementation of the UNet++ (Nested U-Net) architecture for image segmentation, based on the paper 'UNet++: A Nested U-Net Archi… | 32 | 1051 | maintenance |
| thunlp/OpenDelta OpenDelta is a Python library for parameter-efficient tuning (delta tuning) of pretrained language models, letting users attach small train… | 23 | 1048 | maintenance |
| microsoft/SimMIM Official PyTorch implementation of SimMIM, a simple framework for masked image modeling (self-supervised visual pre-training) from Microsof… | 10 | 1048 | maintenance |
| yangxudong/deeplearning A collection of deep learning model training, evaluation, and prediction code implemented with TensorFlow's high-level Estimator API, with … | 32 | 1047 | maintenance |
| yuval-alaluf/restyle-encoder Official PyTorch implementation of ReStyle, a residual-based StyleGAN encoder that inverts real images into GAN latent codes via iterative … | 32 | 1046 | maintenance |
| YuwenXiong/py-R-FCN A Python implementation of R-FCN (Region-based Fully Convolutional Networks) for object detection, modified from the official MATLAB code a… | 32 | 1044 | maintenance |
| JIA-Lab-research/SNR-Aware-Low-Light-Enhance Official PyTorch implementation of the CVPR 2022 paper 'SNR-aware Low-Light Image Enhancement'. It combines SNR-aware transformers and conv… | 32 | 1044 | maintenance |
| zepen/predict_Lottery_ticket A Python application that scrapes historical draw data for Chinese lotteries (双色球/Double Color Ball and 大乐透/Super Lotto) and trains deep le… | 75 | 1043 | maintenance |
| CalciferZh/minimal-hand A Python library for real-time 3D hand motion capture from a single RGB camera, running at over 100fps. It combines a deep network (DetNet)… | 23 | 1043 | maintenance |
| facebookresearch/FixRes FixRes is a PyTorch implementation of the NeurIPS 2019 paper 'Fixing the train-test resolution discrepancy', providing training and fine-tu… | 10 | 1043 | maintenance |
| google-research/morph-net MorphNet is a TensorFlow library for resource-constrained learning of deep network structure. It adds regularizers during training that ind… | 76 | 1042 | maintenance |
| thuml/Anomaly-Transformer Official PyTorch implementation of the Anomaly Transformer model from the ICLR 2022 Spotlight paper on unsupervised time series anomaly det… | 32 | 1042 | maintenance |
| descriptinc/melgan-neurips Official PyTorch implementation of MelGAN, a GAN-based non-autoregressive vocoder that inverts mel-spectrograms into raw audio waveforms fo… | 32 | 1041 | maintenance |
| huggingface/pytorch-pretrained-BigGAN A PyTorch reimplementation of DeepMind's BigGAN generator with pretrained weights at 128, 256, and 512 pixel resolutions, plus scripts to c… | 10 | 1041 | maintenance |
| google-research/deeplab2 DeepLab2 is a TensorFlow library from Google Research providing a unified, state-of-the-art codebase for dense pixel labeling tasks such as… | 10 | 1038 | maintenance |
| Xwin-LM/Xwin-LM Xwin-LM is an open-source project for LLM alignment technologies including supervised fine-tuning, reward models, reject sampling, and RLHF… | 28 | 1037 | maintenance |
| keijiro/Pix2Pix A Unity library that runs pix2pix image-to-image translation neural networks in real time using compute shaders. It includes its own infere… | 23 | 1037 | maintenance |
| sicara/tf-explain tf-explain is a Python library implementing interpretability methods (Grad CAM, occlusion sensitivity, vanilla gradients, etc.) for tf.kera… | 23 | 1035 | maintenance |
| google-deepmind/scalable_agent A TensorFlow implementation of IMPALA (Importance Weighted Actor-Learner Architectures) for scalable distributed deep reinforcement learnin… | 32 | 1034 | maintenance |
| pesser/stable-diffusion The development repository for Stable Diffusion and Latent Diffusion Models, containing research code, training scripts, and pretrained mod… | 32 | 1032 | maintenance |
| goberoi/faceit A Python script that simplifies swapping faces in videos using the deepfakes/faceswap library, with training data sourced from YouTube vide… | 32 | 1032 | maintenance |
| ClementPinard/SfmLearner-Pytorch A PyTorch reimplementation of SfMLearner (Zhou et al., CVPR 2017), which jointly learns monocular depth estimation and camera ego-motion fr… | 63 | 1031 | maintenance |
| qubvel/ttach TTAch is a Python library for image test time augmentation (TTA) with PyTorch. It wraps existing models to apply augmentations like flips, … | 23 | 1031 | maintenance |
| graphdeeplearning/graphtransformer Reference implementation of the Graph Transformer architecture from the paper 'A Generalization of Transformer Networks to Graphs' (DLG-AAA… | 32 | 1030 | maintenance |
| vincentherrmann/pytorch-wavenet A PyTorch implementation of the WaveNet neural audio generation architecture with fast generation support. It includes dataset creation fro… | 32 | 1029 | maintenance |
| yuval-alaluf/hyperstyle Official PyTorch implementation of HyperStyle (CVPR 2022), a hypernetwork that inverts real images into editable regions of StyleGAN's late… | 32 | 1029 | maintenance |
| kakaobrain/rq-vae-transformer The official PyTorch implementation of 'Autoregressive Image Generation using Residual Quantization' (CVPR 2022), implementing RQ-VAE and R… | 32 | 1028 | maintenance |
| lmb-freiburg/flownet2 A Caffe fork implementing FlowNet 2.0, a deep CNN for optical flow estimation from image pairs, released with the CVPR 2017 paper. It inclu… | 32 | 1026 | maintenance |
| dwofk/fast-depth FastDepth is the official PyTorch implementation of the ICRA 2019 paper 'FastDepth: Fast Monocular Depth Estimation on Embedded Systems' fr… | 32 | 1026 | maintenance |
| edvardHua/PoseEstimationForMobile A TensorFlow-based library implementing CPM and Hourglass models with MobileNetV2 inverted residual modules for real-time single-person hum… | 32 | 1025 | maintenance |
| microsoft/Llama-2-Onnx Microsoft's optimized ONNX export of Meta's Llama 2 models (7B and 13B, pretrained and fine-tuned, float16/float32), distributed via Git su… | 29 | 1025 | maintenance |
| sniklaus/sepconv-slomo A reference PyTorch implementation of Video Frame Interpolation via Adaptive Separable Convolution, which generates intermediate frames bet… | 39 | 1021 | maintenance |
| nflows nflows is a comprehensive PyTorch library of normalizing flows for density estimation and generative modeling. It provides invertible trans… | 23 | 1021 | maintenance |
| Gumpest/YOLOv5-Multibackbone-Compression A YOLOv5-based toolbox for swapping in lightweight or high-accuracy backbones (TPH-YOLOv5, GhostNet, ShuffleNetV2, MobileNetV3-Small, Effic… | 32 | 1020 | maintenance |
| wywu/LAB Official C++/Caffe implementation of the CVPR 2018 paper 'Look at Boundary: A Boundary-Aware Face Alignment Algorithm', which localizes fac… | 32 | 1018 | maintenance |
| EvgenyKashin/stylegan2-distillation A research implementation of the ECCV 2020 paper 'StyleGAN2 Distillation for Feed-forward Image Manipulation', distilling StyleGAN2 latent-… | 32 | 1018 | maintenance |
| tensorflow/similarity TensorFlow Similarity is a Python library built on TensorFlow for training and serving similarity and metric learning models, including sel… | 10 | 1018 | maintenance |
| brain-research/self-attention-gan A TensorFlow implementation of Self-Attention GANs for reproducing results from the paper 'Self-Attention Generative Adversarial Networks' … | 10 | 1018 | maintenance |
| facebookresearch/barlowtwins A PyTorch implementation of Barlow Twins, a self-supervised learning method that trains visual representations by reducing redundancy betwe… | 10 | 1015 | maintenance |
| AMLab-Amsterdam/AttentionDeepMIL A PyTorch implementation of the paper 'Attention-based Deep Multiple Instance Learning' (Ilse, Tomczak & Welling, 2018). It provides code t… | 32 | 1013 | maintenance |
| wpeebles/gangealing Official PyTorch implementation of GANgealing, a CVPR 2022 method that trains a Spatial Transformer to densely align images using GAN-gener… | 32 | 1013 | maintenance |
| snap-research/NeROIC Official PyTorch implementation of NeROIC, a neural method for capturing 3D object geometry and material from online image collections and … | 32 | 1012 | maintenance |
| tensorflow/neural-structured-learning Neural Structured Learning (NSL) is a TensorFlow framework for training neural networks with structured signals, either explicit graphs or … | 61 | 1011 | maintenance |
| allegro/allRank allRank is a PyTorch-based framework for training neural learning-to-rank (LTR) models. It provides pointwise, pairwise, and listwise loss … | 32 | 1011 | maintenance |
| kakaobrain/kogpt KakaoBrain's KoGPT, a Korean Generative Pre-trained Transformer (GPT) model with 6B parameters, distributed via Hugging Face with inference… | 23 | 1011 | maintenance |
| bubbliiiing/yolov8-pytorch A PyTorch implementation of the YOLOv8 object detection model with training, prediction, and evaluation scripts. It supports training on cu… | 22 | 1011 | maintenance |
| xiaoyufenfei/Efficient-Segmentation-Networks A PyTorch reference implementation collection of lightweight, real-time semantic segmentation models such as ENet, ERFNet, LEDNet, Fast-SCN… | 32 | 1009 | maintenance |
| zhanghang1989/PyTorch-Multi-Style-Transfer A PyTorch implementation of MSG-Net and Gatys et al. neural style transfer for applying artistic styles to images in real time. It includes… | 23 | 1009 | maintenance |
| mbrossar/ai-imu-dr Python/PyTorch implementation of the AI-IMU dead reckoning algorithm from the authors' IEEE/ArXiv paper, combining an invariant Kalman filt… | 32 | 1008 | maintenance |
| kang205/SASRec Official TensorFlow implementation of SASRec (Self-Attentive Sequential Recommendation), a Transformer-based sequential recommendation mode… | 32 | 1007 | maintenance |
| mileyan/pseudo_lidar Research code implementing Pseudo-LiDAR, a CVPR 2019 method that converts image-based depth maps into pseudo-LiDAR point clouds for 3D obje… | 32 | 1007 | maintenance |
| alex04072000/ObstructionRemoval The official TensorFlow implementation of the CVPR 2020 paper 'Learning to See Through Obstructions', which removes obstructions like windo… | 32 | 1005 | maintenance |
| microsoft/nnfusion NNFusion is a flexible and efficient deep neural network (DNN) compiler that generates high-performance executables from model descriptions… | 10 | 1005 | maintenance |
| kevinzakka/spatial-transformer-network A TensorFlow implementation of Spatial Transformer Networks, a differentiable module that can be inserted into ConvNet architectures to add… | 32 | 1004 | maintenance |
| NATSpeech/NATSpeech A PyTorch framework for non-autoregressive text-to-speech (NAR-TTS), containing official implementations of PortaSpeech (NeurIPS 2021) and … | 23 | 1004 | maintenance |
| LeeSureman/Flat-Lattice-Transformer Reference implementation of the ACL 2020 paper FLAT: Chinese NER Using Flat-Lattice Transformer, built on PyTorch and FastNLP. It trains fl… | 32 | 1002 | maintenance |
| google-research/magvit Official JAX implementation of MAGVIT, a masked generative video transformer from a CVPR 2023 paper by Google Research and CMU. It provides… | 10 | 1002 | maintenance |
| Plug-and-Play Diffusion Features Official PyTorch implementation of Plug-and-Play (PnP) diffusion features for text-driven image-to-image translation, presented at CVPR 202… | 31 | 1001 | maintenance |
| omertov/encoder4editing Official PyTorch implementation of the SIGGRAPH 2021 paper 'Designing an Encoder for StyleGAN Image Manipulation' (e4e). It encodes real im… | 32 | 1000 | maintenance |
| openai/grok Research code accompanying the paper 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' from OpenAI. It provides t… | 10 | 4259 | experimental |
| apple/ml-mgie MGIE (MLLM-Guided Image Editing) is Apple's research implementation of instruction-based image editing guided by multimodal large language … | 69 | 3872 | experimental |
| 0hq/WebGPT WebGPT is a vanilla JavaScript and HTML implementation of GPT transformer inference running in the browser via WebGPU, in under ~1500 lines… | 30 | 3787 | experimental |
| MoonshotAI/Attention-Residuals Official implementation of Attention Residuals (AttnRes), a drop-in replacement for standard residual connections in Transformers that lets… | 44 | 3516 | experimental |
| lucidrains/musiclm-pytorch A PyTorch library implementing MusicLM, Google's text-to-music generation model, by combining text-conditioned AudioLM with MuLan, a text-a… | 22 | 3292 | experimental |
| enhuiz/vall-e An unofficial PyTorch implementation of the VALL-E text-to-speech audio language model, built on the EnCodec tokenizer. It provides trainin… | 32 | 2967 | experimental |
| guoqincode/Open-AnimateAnyone An unofficial PyTorch implementation of Animate Anyone, a diffusion-based method that animates a static character image using pose sequence… | 27 | 2917 | experimental |
| Everlyn-Labs/Everlyn-1 Everlyn-1 is an open autoregressive foundational video AI model from Everlyn Labs, accompanied by research on video compression/tokenizatio… | 67 | 2893 | experimental |
| openai/weak-to-strong OpenAI's research codebase implementing weak-to-strong generalization experiments from their alignment paper, where a strong pretrained mod… | 10 | 2548 | experimental |
| test-time-training/ttt-video-dit Official PyTorch implementation of 'One-Minute Video Generation with Test-Time Training', which finetunes the CogVideoX 5B diffusion transf… | 48 | 2451 | experimental |
| AIGCDesignGroup/ReplaceAnything ReplaceAnything is a research project from Alibaba's Institute for Intelligent Computing for ultra-high quality content replacement in imag… | 27 | 2423 | experimental |
| facebookresearch/large_concept_model Official PyTorch implementation of Meta's Large Concept Models (LCM), which perform language modeling by autoregressively predicting senten… | 22 | 2381 | experimental |
| dvmazur/mixtral-offloading A Python library enabling efficient inference of Mixtral-8x7B mixture-of-experts language models on limited hardware like Google Colab or c… | 27 | 2334 | experimental |
| Sumandora/remove-refusals-with-transformers A proof-of-concept Python script that removes refusal behavior from LLMs by ablating the refusal direction using pure Hugging Face Transfor… | 46 | 2230 | experimental |
| lllyasviel/LayerDiffuse LayerDiffuse is a research project that generates transparent images and image layers using diffusion models with latent transparency. It p… | 26 | 2223 | experimental |
| MineDojo/NitroGen NitroGen is an open foundation model for generalist gaming agents that takes pixel input from games and predicts gamepad actions, trained v… | 42 | 2216 | experimental |
| etched-ai/open-oasis Inference code and model weights for Oasis 500M, an interactive world model from Decart and Etched that generates gameplay video autoregres… | 22 | 2145 | experimental |
| JiauZhang/DragGAN A Python implementation of DragGAN, a research method for interactively manipulating generated images by dragging points on the generative … | 30 | 2123 | experimental |
| QwenLM/Qwen-Image-Layered Qwen-Image-Layered is a diffusion-based model and pipeline that decomposes an input image into multiple independently editable RGBA layers.… | 39 | 2117 | experimental |
| lucidrains/make-a-video-pytorch A PyTorch library implementing Make-A-Video, Meta AI's text-to-video generation approach, built around pseudo-3d (axial) convolutions and s… | 23 | 1989 | experimental |
| lucidrains/gigagan-pytorch A PyTorch implementation of GigaGAN, Adobe's state-of-the-art generative adversarial network for text-to-image and unconditional image synt… | 22 | 1945 | experimental |
| chelsea0x3b/dfdx A Rust deep learning library providing shape-checked tensors and neural network building blocks with compile-time verification of tensor op… | 23 | 1934 | experimental |
| matthewearl/deep-anpr An experimental Python project that uses neural networks (TensorFlow) to build an automatic number plate recognition (ANPR) system. It incl… | 32 | 1860 | experimental |
| rentruewang/aioway Aioway is an optimizing compiler for deep learning algorithms that treats ML models as instructions and builds declarative pipelines using … | 77 | 1823 | experimental |
| deepseek-ai/DeepSeek-V3.2-Exp DeepSeek-V3.2-Exp is an experimental open-weight large language model release that introduces DeepSeek Sparse Attention for more efficient … | 37 | 1654 | experimental |
| EvolvingLMMs-Lab/open-r1-multimodal A fork of Hugging Face's open-r1 that extends the R1 GRPO reinforcement learning training paradigm to multimodal (vision-language) models l… | 21 | 1606 | experimental |
| lyuchenyang/Macaw-LLM Macaw-LLM is a multi-modal language modeling framework that integrates image, video, audio, and text data, built on CLIP, Whisper, and LLaM… | 30 | 1591 | experimental |
| ali-vilab/composer Official implementation of Composer, a 5-billion-parameter controllable diffusion model for creative image synthesis using composable condi… | 31 | 1559 | experimental |
| KUR-creative/SickZil-Machine SickZil-Machine is a desktop application that automates text removal from manga and comic pages during the scanlation (translation) process… | 23 | 1526 | experimental |
| Learning to (Learn at Test Time) Language Model Official PyTorch implementation of Test-Time Training (TTT) layers, a sequence modeling layer with linear complexity whose hidden state is … | 24 | 1405 | experimental |
| lucidrains/video-diffusion-pytorch A Pytorch implementation of the Video Diffusion Models paper by Jonathan Ho, extending denoising diffusion probabilistic models (DDPMs) to … | 23 | 1391 | experimental |
| Orion Orion is a Python machine learning library for unsupervised time series anomaly detection, offering verified ML pipelines that flag rare pa… | 73 | 1370 | experimental |