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 |
|---|---|---|---|
| ARISE-Initiative/robomimic robomimic is a modular Python framework for robot learning from demonstration, providing standardized demonstration datasets and offline le… | 65 | 1572 | active |
| ZFTurbo/Music-Source-Separation-Training A Python training framework for music source separation models, supporting many architectures such as MDX23C, Demucs, Band Split RoFormer, … | 87 | 1569 | active |
| sdv-dev/CTGAN CTGAN is a Python library of deep learning based synthetic data generators for single-table tabular data, implementing the CTGAN conditiona… | 76 | 1569 | active |
| Chinese-Text-Classification-Pytorch A PyTorch-based collection of ready-to-run Chinese text classification models including TextCNN, TextRNN, FastText, TextRCNN, BiLSTM with a… | 32 | 5728 | maintenance |
| baaivision/Emu3.5 Emu3.5 is BAAI's native multimodal foundation model that jointly predicts next states across vision and language, trained on 10T+ interleav… | 40 | 1561 | active |
| kritiksoman/GIMP-ML GIMP-ML is a set of Python plugins that bring computer vision and deep learning models into the GNU Image Manipulation Program (GIMP). It p… | 23 | 1560 | active |
| hustvl/MapTR MapTR is an end-to-end transformer-based framework for online vectorized HD map construction from camera imagery in autonomous driving. It … | 23 | 1558 | active |
| decoderesearch/SAELens SAELens is a Python library for training sparse autoencoders (SAEs) on language model activations and analyzing them for mechanistic interp… | 90 | 1555 | active |
| tianrun-chen/SAM-Adapter-PyTorch A PyTorch library that adapts Meta AI's Segment Anything Model (SAM, SAM2, SAM3) to underperforming downstream segmentation tasks using lig… | 64 | 1553 | active |
| ZeyueT/AudioX AudioX is a unified multimodal framework for anything-to-audio generation, producing audio and music conditioned on text, video, image, or … | 49 | 1552 | active |
| MST (Spectral Compressive Imaging Toolbox) Retinexformer is a one-stage Retinex-based Transformer model and toolbox for low-light image enhancement, published at ICCV 2023. It suppor… | 63 | 1548 | active |
| ShineChen1024/MagicClothing Official PyTorch implementation of Magic Clothing, a diffusion-based model for controllable garment-driven image synthesis (virtual try-on)… | 26 | 1548 | active |
| yuantianyuan01/FastWAM Official PyTorch codebase for Fast-WAM, a World Action Model for robot manipulation that skips test-time future video imagination and gener… | 56 | 1547 | active |
| hustvl/LightningDiT LightningDiT is a research codebase for latent diffusion models implementing VA-VAE and LightningDiT, achieving FID 1.35 on ImageNet-256 wi… | 44 | 1547 | active |
| NirAharon/BoT-SORT BoT-SORT is a state-of-the-art multi-object tracker that combines motion and appearance information with camera motion compensation and an … | 32 | 1547 | active |
| tensorflow/gnn TensorFlow GNN is a Python library for building Graph Neural Networks on TensorFlow, including a GraphTensor type for heterogeneous graphs,… | 66 | 1546 | active |
| MoonshotAI/Moonlight Moonlight is a 3B/16B Mixture-of-Experts LLM trained with the Muon optimizer, released by Moonshot AI along with a memory- and communicatio… | 33 | 1546 | active |
| RLHFlow/RLHF-Reward-Modeling A collection of training recipes for reward models used in RLHF, covering Bradley-Terry reward models, pairwise preference models, ArmoRM, … | 30 | 1546 | active |
| lucidrains/soundstorm-pytorch A PyTorch implementation of SoundStorm, Google DeepMind's efficient parallel audio generation model that applies MaskGiT-style masked gener… | 32 | 1545 | active |
| facebookresearch/mmf MMF is a modular PyTorch framework for vision and language multimodal research from Facebook AI Research. It ships reference implementation… | 67 | 5637 | maintenance |
| ROMP ROMP is a PyTorch-based library and pip-installable API (simple-romp) for real-time monocular multi-person 3D human mesh recovery, implemen… | 23 | 1543 | stable |
| lucidrains/DALLE-pytorch A PyTorch implementation/replication of OpenAI's DALL-E, a text-to-image transformer, including a discrete VAE and optional CLIP for rankin… | 23 | 5627 | maintenance |
| JingyunLiang/SwinIR Official PyTorch implementation of SwinIR, a Swin Transformer-based model for image restoration tasks including super-resolution, denoising… | 23 | 5611 | maintenance |
| allenzren/open-pi-zero An open-source re-implementation of the pi0 vision-language-action (VLA) model from Physical Intelligence, built on a pre-trained PaliGemma… | 24 | 1533 | active |
| Leela Zero Leela Zero is an open-source Go engine that reimplements AlphaGo Zero, combining Monte Carlo Tree Search with a deep residual convolutional… | 23 | 5591 | maintenance |
| luciddreamer-cvlab/LucidDreamer LucidDreamer is the official implementation of a research method that generates 3D Gaussian Splatting scenes from text prompts, published i… | 66 | 1528 | active |
| ATH-MaaS/Ovis Ovis is an open-source Multimodal Large Language Model (MLLM) architecture that structurally aligns visual and textual embeddings, with rel… | 62 | 1528 | active |
| WenmuZhou/PytorchOCR A PyTorch-based OCR toolkit that ports PaddleOCR models to PyTorch, supporting common text detection and recognition algorithms like the PP… | 56 | 1527 | active |
| cchen156/Learning-to-See-in-the-Dark TensorFlow implementation of 'Learning to See in the Dark' (CVPR 2018), a deep learning model that brightens very dark, short-exposure RAW … | 48 | 5567 | maintenance |
| Tencent/TFace TFace is a research platform from Tencent Youtu Lab for trusty face analysis, covering face recognition, face security (anti-spoofing), fac… | 54 | 1524 | active |
| geomstats/geomstats Geomstats is an open-source Python package for computations, statistics, and machine learning on manifolds with geometric structures. It pr… | 67 | 1523 | active |
| keras-rl/keras-rl A Python library implementing state-of-the-art deep reinforcement learning algorithms (DQN, DDPG, SARSA, and more) that integrates seamless… | 23 | 5549 | maintenance |
| anishathalye/neural-style A Python command-line tool implementing the neural style transfer algorithm (Gatys et al.) in TensorFlow, applying the style of one image t… | 64 | 5547 | maintenance |
| Yuliang-Liu/MonkeyOCRv2 MonkeyOCRv2 is a document-native vision encoder and visual-text foundation model for Document AI, pretrained on the 113M-image MonkeyDoc v2… | 56 | 1517 | active |
| Phantom-video/Phantom Phantom is a subject-consistent video generation model from ByteDance that preserves reference subject identity via cross-modal alignment. … | 36 | 1517 | active |
| hustvl/DiffusionDrive DiffusionDrive is a truncated diffusion model for real-time end-to-end autonomous driving, released as the official PyTorch implementation … | 41 | 1513 | active |
| k2-fsa/icefall Icefall is a collection of speech recognition (ASR) and TTS training recipes built on the k2 and lhotse libraries, implemented in Python wi… | 68 | 1512 | active |
| charlesq34/pointnet Reference implementation of PointNet, a neural network architecture that directly consumes unordered 3D point clouds for classification and… | 32 | 5482 | maintenance |
| wb14123/seq2seq-couplet A deep learning project that generates Chinese couplets (对联) using a seq2seq model built with TensorFlow. It includes training scripts, a w… | 32 | 5480 | maintenance |
| Flashlight wav2letter++ is Facebook AI Research's end-to-end automatic speech recognition (ASR) toolkit written in C++. It has been consolidated into … | 67 | 5476 | maintenance |
| mattmireles/gemma-tuner-multimodal A Python tool for LoRA fine-tuning of Gemma 4 and 3n models on text, images, and audio using Apple Silicon's Metal Performance Shaders. It … | 61 | 1504 | active |
| piddnad/DDColor DDColor is the official PyTorch implementation of an ICCV 2023 paper on photo-realistic automatic image colorization using dual decoders an… | 56 | 1502 | active |
| TencentARC/MotionCtrl MotionCtrl is the official implementation of a SIGGRAPH 2024 paper providing a unified and flexible motion controller for video generation … | 27 | 1500 | active |
| CUT3R/CUT3R CUT3R is the official PyTorch implementation of 'Continuous 3D Perception Model with Persistent State' (CVPR 2025 Oral), a stateful recurre… | 35 | 1499 | active |
| NJU-PCALab/STAR STAR is a research implementation of an ICCV 2025 paper performing real-world video super-resolution using spatial-temporal augmentation wi… | 32 | 1498 | active |
| open-mmlab/mmengine MMEngine is the foundational training engine library for OpenMMLab projects, providing a unified training loop, config system, registry, ho… | 68 | 1496 | 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… | 59 | 1496 | active |
| ucb-bar/gemmini Gemmini is Berkeley's open-source generator for parameterizable systolic-array DNN hardware accelerators, written in Chisel (Scala) and int… | 66 | 1493 | active |
| ibab/tensorflow-wavenet A TensorFlow implementation of DeepMind's WaveNet generative neural network architecture for raw audio waveform generation. It provides tra… | 32 | 5422 | maintenance |
| bert4keras A lightweight, clean reimplementation of BERT and other transformer models (RoBERTa, ALBERT, T5, GPT, ELECTRA, NEZHA) for Keras/tf.keras. I… | 23 | 5409 | maintenance |
| google-deepmind/graph_nets DeepMind's library for building graph networks (graph neural networks) in TensorFlow and Sonnet, based on the 'Relational inductive biases,… | 32 | 5407 | maintenance |
| reczoo/FuxiCTR FuxiCTR is an open-source Python library for click-through rate (CTR) prediction built on PyTorch and TensorFlow. It offers a configurable,… | 75 | 1481 | active |
| alxndrTL/mamba.py A simple, readable pure-PyTorch (plus MLX) implementation of the Mamba state-space model architecture with a parallel scan for efficient tr… | 50 | 1480 | active |
| sczhou/Upscale-A-Video Upscale-A-Video is a diffusion-based model for real-world video super-resolution that takes low-resolution videos and text prompts as input… | 27 | 1477 | active |
| Lightning-AI/lightning-thunder Lightning Thunder is a source-to-source deep learning compiler for PyTorch that optimizes models for training and inference. It provides a … | 70 | 1475 | active |
| hojonathanho/diffusion The official reference implementation of Denoising Diffusion Probabilistic Models (DDPM) from the 2020 paper by Jonathan Ho et al., written… | 32 | 5341 | maintenance |
| mit-han-lab/torchsparse TorchSparse is a high-performance PyTorch library for sparse convolution on 3D point clouds, with optimized GPU kernels for both training a… | 23 | 1473 | active |
| facebookresearch/MobileLLM Meta's training code for MobileLLM, a family of sub-billion parameter language models optimized for on-device use, published at ICML 2024. … | 56 | 1465 | active |
| Topdu/OpenOCR OpenOCR is an open-source Python toolkit for general OCR research and applications, covering text detection and recognition, formula and ta… | 58 | 1464 | active |
| tianweiy/DMD2 DMD2 is the official PyTorch implementation of Improved Distribution Matching Distillation, a NeurIPS 2024 method that distills diffusion m… | 25 | 1464 | active |
| NVIDIA/tacotron2 NVIDIA's PyTorch implementation of the Tacotron 2 text-to-speech model, which synthesizes mel spectrograms from text for vocoder-based audi… | 32 | 5295 | maintenance |
| yunjey/stargan Official PyTorch implementation of StarGAN, a unified generative adversarial network for multi-domain image-to-image translation (CVPR 2018… | 32 | 5295 | maintenance |
| FeiYull/TensorRT-Alpha A C++/CUDA library providing TensorRT-accelerated deployment for 30+ popular computer vision models including YOLOv3-v8, YOLOv8-Pose/Seg/Cl… | 77 | 1462 | active |
| tensorflow/tpu A collection of reference models and tools for training machine learning models on Google Cloud TPUs, maintained as a public mirror by the … | 68 | 5278 | maintenance |
| zsyOAOA/InvSR InvSR is a Python research library implementing arbitrary-steps image super-resolution via diffusion inversion, leveraging pre-trained diff… | 48 | 1459 | active |
| aeon-toolkit/aeon aeon is a scikit-learn compatible Python toolkit for machine learning on time series, covering classification, regression, clustering, fore… | 93 | 1457 | active |
| Walter0807/MotionBERT Official PyTorch implementation of MotionBERT (ICCV 2023), a unified pretrained model for learning human motion representations from 2D ske… | 61 | 1454 | 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… | 43 | 1453 | active |
| dbolya/yolact YOLACT is a PyTorch implementation of a fully convolutional model for real-time instance segmentation, accompanying the YOLACT and YOLACT++… | 47 | 5242 | maintenance |
| zylo117/Yet-Another-EfficientDet-Pytorch A PyTorch re-implementation of Google's EfficientDet object detection model with pretrained weights matching near-SOTA accuracy at real-tim… | 23 | 5233 | maintenance |
| SakanaAI/evolutionary-model-merge Official repository for SakanaAI's Evolutionary Model Merge research, providing code and resources to reproduce paper evaluations of models… | 16 | 1442 | active |
| tianweiy/CausVid CausVid is a research codebase implementing a fast autoregressive video diffusion model distilled from a bidirectional diffusion transforme… | 33 | 1441 | active |
| tensorspace-team/tensorspace TensorSpace is a neural network 3D visualization framework built on TensorFlow.js, Three.js, and Tween.js. It provides Keras-like APIs to b… | 23 | 5198 | maintenance |
| voice-cloning-app/Voice-Cloning-App A Python/PyTorch desktop application for cloning and synthesizing human voices from audio datasets. It handles the full pipeline from autom… | 23 | 1438 | active |
| zuruoke/watermark-removal A machine learning tool that removes watermarks from images using deep learning image inpainting, based on Contextual Attention and Gated C… | 77 | 5181 | maintenance |
| salesforce/CodeGen CodeGen is a family of open-source large language models (350M to 16B parameters) from Salesforce AI Research for program synthesis, genera… | 67 | 5181 | maintenance |
| 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… | 38 | 1433 | active |
| zsyOAOA/ResShift ResShift is an efficient diffusion model for image super-resolution that transfers between low- and high-resolution images by shifting resi… | 59 | 1430 | active |
| apple/ml-aim Apple's official repository for AIM (Autoregressive Image Models), providing code and pretrained checkpoints for AIMv1 and AIMv2 large visi… | 69 | 1426 | active |
| 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… | 56 | 1425 | 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 | 5126 | maintenance |
| SagiPolaczek/NeuralSVG Official PyTorch implementation of NeuralSVG, an ICCV 2025 paper that generates layered, editable SVG vector graphics from text prompts. It… | 44 | 1421 | active |
| oracle/tribuo Tribuo is a Java machine learning library from Oracle Labs providing classification, regression, clustering, anomaly detection, and multi-l… | 56 | 1418 | 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… | 70 | 1416 | 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 … | 56 | 1416 | active |
| Terrain Diffusion Terrain Diffusion is a Python framework that uses diffusion models as a learned, deterministic replacement for Perlin noise, generating inf… | 58 | 1414 | 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… | 23 | 1413 | 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 | 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… | 17 | 1412 | active |
| dailenson/SDT Official PyTorch implementation of the CVPR 2023 paper 'Disentangling Writer and Character Styles for Handwriting Generation' (SDT). It gen… | 40 | 1411 | active |
| 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… | 70 | 1408 | 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… | 57 | 1407 | active |
| 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 … | 53 | 1406 | active |
| XiaoMi/mace MACE (Mobile AI Compute Engine) is a deep learning inference framework optimized for mobile heterogeneous computing on Android, iOS, Linux … | 23 | 5052 | maintenance |
| gnobitab/InstaFlow InstaFlow is a one-step text-to-image generation model based on Rectified Flow, enabling ultra-fast Stable Diffusion inference without iter… | 29 | 1403 | active |
| Irodori-TTS Irodori-TTS is a Flow Matching-based text-to-speech model with training and inference code, built on a Rectified Flow Diffusion Transformer… | 58 | 1401 | active |
| Denys88/rl_games RL Games is a high-performance reinforcement learning library built on PyTorch, focused on training agents in massively parallel GPU-based … | 75 | 1397 | active |
| Zejun-Yang/AniPortrait AniPortrait is a Python framework from Tencent that generates photorealistic portrait animations from an audio clip and a reference image, … | 26 | 5014 | maintenance |
| galilai-group/lejepa LeJEPA is a Python framework for scalable, theoretically grounded self-supervised representation learning based on Joint-Embedding Predicti… | 42 | 1393 | 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 | 1390 | stable |