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 |
|---|---|---|---|
| facebookresearch/multimodal TorchMultimodal is a PyTorch library from Meta for training state-of-the-art multimodal multi-task models at scale, covering both content u… | 77 | 1740 | active |
| openai/consistency_models Official PyTorch implementation of Consistency Models, a generative image model family from OpenAI supporting consistency distillation, con… | 10 | 6483 | maintenance |
| google/automl Google Brain's AutoML repository containing implementations of AutoML models and libraries such as EfficientNet, EfficientNetV2, and Effici… | 10 | 6474 | maintenance |
| NVIDIA/FasterTransformer NVIDIA's highly optimized C++/CUDA library for fast inference of Transformer-based models such as BERT, GPT, and encoder-decoder models, wi… | 23 | 6450 | maintenance |
| birdnet-team/BirdNET-Analyzer BirdNET-Analyzer is an open-source Python tool from the Cornell Lab of Ornithology that uses deep learning to identify bird species from au… | 77 | 1728 | active |
| meta-pytorch/tnt TNT (torchtnt) is a lightweight library from Meta providing tools and utilities for PyTorch training workflows. It offers building blocks f… | 76 | 1724 | active |
| facebookresearch/ConvNeXt Official PyTorch implementation of ConvNeXt, a pure convolutional neural network architecture from the CVPR 2022 paper 'A ConvNet for the 2… | 10 | 6408 | maintenance |
| One-2-3-45/One-2-3-45 One-2-3-45 is the official PyTorch implementation of a NeurIPS 2023 paper that converts any single image into a full 360-degree 3D textured… | 29 | 1717 | stable |
| tenstorrent/tt-metal TT-Metal is Tenstorrent's open-source software stack containing TT-NN, a Python and C++ neural network operator library, and TT-Metalium, a… | 99 | 1716 | active |
| lightly-ai/lightly-train LightlyTrain is a Python framework for training computer vision models, covering pretraining of vision foundation models (DINOv2/v3) on unl… | 83 | 1716 | active |
| sihyun-yu/REPA REPA is the official PyTorch implementation of the ICLR 2025 paper 'Representation Alignment for Generation', a regularization technique th… | 24 | 1716 | active |
| kingoflolz/mesh-transformer-jax A JAX/Haiku library implementing model-parallel training and inference of transformer models using xmap/pjit operators, similar to Megatron… | 32 | 6384 | maintenance |
| lsdefine/simple_GRPO A minimal (~200 lines) Python implementation of GRPO reinforcement learning for training LLMs to develop r1-like reasoning, built on PyTorc… | 41 | 1714 | active |
| feizc/FluxMusic FluxMusic is the official PyTorch implementation of a rectified flow Transformer model for text-to-music generation, from the paper 'Flux t… | 24 | 1709 | active |
| zai-org/ImageReward ImageReward is the first general-purpose human preference reward model for text-to-image generation, trained on 137k expert comparison pair… | 49 | 1705 | stable |
| BindsNET/bindsnet BindsNET is a Python package for simulating spiking neural networks (SNNs) built on PyTorch tensor functionality, running on CPUs or GPUs. … | 81 | 1704 | active |
| PaddlePaddle/PaddleVideo PaddleVideo is a video understanding toolkit built on PaddlePaddle, offering state-of-the-art models for action recognition, temporal actio… | 23 | 1702 | active |
| lucidrains/alphafold3-pytorch A PyTorch implementation of Google DeepMind's Alphafold 3 model for protein and molecular structure prediction. It is a research-oriented r… | 67 | 1699 | active |
| scvi-tools scvi-tools is a Python library for deep probabilistic modeling and analysis of single-cell and spatial omics data, built on PyTorch, PyTorc… | 92 | 1697 | stable |
| shubham-goel/4D-Humans 4DHumans is a Python research codebase implementing HMR 2.0, a transformer-based model for 3D human mesh recovery from single images, plus … | 56 | 1695 | active |
| TensorPack Tensorpack is a high-level neural network training interface built on graph-mode TensorFlow, focused on training speed and flexibility for … | 23 | 6284 | maintenance |
| elixir-nx/axon Axon is a neural network library for Elixir built on top of Nx numerical definitions. It provides a functional API, a high-level model crea… | 89 | 1692 | active |
| intelligent-machine-learning/dlrover DLRover is an automatic distributed deep learning system that manages training of large AI models on Kubernetes and Ray clusters. It provid… | 80 | 1689 | active |
| pytorch-tabular/pytorch_tabular PyTorch Tabular is a Python library providing a unified, high-level interface for deep learning models on tabular data, built on PyTorch an… | 73 | 1686 | active |
| takuseno/d3rlpy d3rlpy is a Python library for offline and online deep reinforcement learning built on PyTorch, offering state-of-the-art algorithms throug… | 45 | 1684 | active |
| franciszzj/Leffa Leffa is a diffusion-based framework for controllable person image generation, supporting virtual try-on and pose transfer via a regulariza… | 37 | 1683 | active |
| bytedance/Sa2VA Sa2VA is a family of research models and codebases from ByteDance that combine SAM-2 with multimodal LLMs for pixel-level grounded understa… | 67 | 1682 | active |
| williamyang1991/DualStyleGAN Official PyTorch implementation of DualStyleGAN, a CVPR 2022 model for exemplar-based high-resolution (1024px) portrait style transfer. It … | 32 | 1682 | stable |
| sunlabuiuc/PyHealth PyHealth is an open-source Python toolkit for clinical deep learning, unifying healthcare datasets (EHRs, physiological signals, medical im… | 95 | 1673 | active |
| Gen-Verse/MMaDA MMaDA is an open-source family of multimodal large diffusion language models that unify textual reasoning, multimodal understanding, and te… | 46 | 1673 | active |
| zihangdai/xlnet XLNet is a generalized autoregressive pretraining method for language representation learning, built on Transformer-XL. This repo provides … | 32 | 6189 | maintenance |
| mit-han-lab/torchquantum TorchQuantum is a PyTorch-based framework for quantum computing simulation, supporting statevector and pulse-level simulation on GPUs with … | 60 | 1669 | active |
| nwojke/deep_sort A Python implementation of Deep SORT, a multi-object tracking algorithm that extends SORT with a deep appearance descriptor for robust data… | 33 | 6179 | maintenance |
| tkarras/progressive_growing_of_gans Official TensorFlow implementation of the ICLR 2018 NVIDIA paper 'Progressive Growing of GANs', which trains generators and discriminators … | 32 | 6176 | maintenance |
| JiuhaiChen/BLIP3o Official implementation of the BLIP3o-Series, a unified autoregressive-plus-diffusion model for text-to-image generation and editing. It co… | 41 | 1667 | active |
| apple/ml-mobileclip Official implementation of Apple's MobileCLIP and MobileCLIP2, fast image-text contrastive (CLIP-style) models optimized for mobile latency… | 68 | 1664 | active |
| maziarraissi/PINNs The original reference implementation of Physics-Informed Neural Networks (PINNs), which train neural networks to solve and discover nonlin… | 59 | 6160 | maintenance |
| ZJU4HealthCare/HealthGPT HealthGPT is a medical multimodal large language model family unifying medical image comprehension and generation via heterogeneous knowled… | 60 | 1662 | active |
| thtrieu/darkflow Darkflow is a Python library that translates Darknet's YOLO neural network definitions to TensorFlow, enabling real-time object detection a… | 32 | 6139 | maintenance |
| Data-Centric-AI-Community/fg-data-synthetic A Python library for generating synthetic tabular and time-series data using state-of-the-art generative models such as GANs (including CTG… | 87 | 1659 | active |
| thunil/TecoGAN TecoGAN is the official source code for a temporally coherent GAN for video super-resolution, published at SIGGRAPH/ACM TOG. It includes in… | 32 | 6133 | maintenance |
| Robbyant/lingbot-depth LingBot-Depth is a PyTorch-based model and toolkit for masked depth modeling that transforms incomplete, noisy depth sensor data into metri… | 53 | 1655 | active |
| mhamilton723/FeatUp FeatUp is a model-agnostic framework that upsamples the spatial resolution of deep neural network features by 16-32x without changing their… | 17 | 1655 | active |
| XueZeyue/DanceGRPO Official implementation of DanceGRPO, a framework applying Group Relative Policy Optimization (GRPO) to fine-tune visual generation models … | 37 | 1654 | active |
| taki0112/UGATIT Official TensorFlow implementation of U-GAT-IT, an unsupervised image-to-image translation model using attention modules and adaptive layer… | 32 | 6113 | maintenance |
| GoMLX GoMLX is an accelerated machine learning and math framework for Go, comparable to PyTorch/JAX/TensorFlow. It offers differentiable operator… | 96 | 1651 | active |
| microsoft/Mage Mage is a family of lightweight 4B-parameter multimodal models from Microsoft, including Mage-VL for image and video understanding and Mage… | 55 | 1649 | active |
| InterDigitalInc/CompressAI CompressAI is a PyTorch library and evaluation platform for end-to-end learned data compression research. It provides custom layers, entrop… | 76 | 1648 | active |
| bowang-lab/scGPT scGPT is a foundation model for single-cell multi-omics built with a generative transformer, providing pretrained checkpoints and a Python … | 64 | 1646 | active |
| bytedance/DreamO DreamO is the official PyTorch implementation of a unified framework for image customization, built on FLUX diffusion models. It supports t… | 33 | 1646 | active |
| JIA-Lab-research/ControlNeXt ControlNeXt is the official implementation of a controllable generation method for images and videos, built on Stable Diffusion XL, Stable … | 24 | 1646 | active |
| Lucchetto/SuperImage SuperImage is an Android application that upscales low-resolution images using a Real-ESRGAN neural network running on-device via the MNN d… | 10 | 1641 | active |
| yenchenlin/nerf-pytorch A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the original paper's results for synthesizing novel views of comp… | 32 | 6052 | maintenance |
| CoinCheung/BiSeNet A PyTorch implementation of the BiSeNet V1 and V2 real-time semantic segmentation models, with pretrained weights for Cityscapes, COCO-Stuf… | 67 | 1639 | active |
| luanfujun/deep-painterly-harmonization Research code implementing the 'Deep Painterly Harmonization' algorithm, which seamlessly blends a pasted object into a painting's style us… | 32 | 6033 | maintenance |
| yoshitomo-matsubara/torchdistill torchdistill is a modular, configuration-driven PyTorch framework for knowledge distillation and general deep learning experiments, requiri… | 84 | 1633 | active |
| facebookresearch/DomainBed DomainBed is a PyTorch suite containing benchmark datasets and algorithms for domain generalization research. It implements many algorithms… | 10 | 1632 | active |
| Fantasy-AMAP/fantasy-talking FantasyTalking is a research codebase and model for generating realistic talking portrait videos from a single image and an audio clip, bui… | 45 | 1631 | active |
| SyncTalk SyncTalk is the official PyTorch implementation of a CVPR 2024 paper that synthesizes speech-driven, synchronized talking head videos using… | 43 | 1628 | active |
| Picsart-AI-Research/StreamingT2V StreamingT2V is a research codebase (CVPR 2025) implementing an autoregressive technique that turns short text-to-video diffusion models li… | 28 | 1628 | active |
| guochengqian/Magic123 Magic123 is the official PyTorch implementation of an ICLR 2024 paper that generates high-quality textured 3D meshes from a single unposed … | 84 | 1623 | stable |
| UbiquitousLearning/mllm MLLM is a fast, lightweight multimodal LLM inference engine written in C++ for mobile and edge devices, with backends for ARM CPU, Qualcomm… | 71 | 1617 | active |
| PKU-Alignment/safe-rlhf Beaver is a modular open-source framework from Peking University for training large language models with SFT, RLHF, and Safe RLHF (constrai… | 50 | 1617 | active |
| open-gigaai/giga-world-0 GigaWorld-0 is a unified world model framework that acts as a data engine for Vision-Language-Action (VLA) learning in embodied AI. It comb… | 38 | 1616 | active |
| NVlabs/sionna Sionna is an open-source, GPU-accelerated, differentiable Python library from NVIDIA for research on communication systems. It comprises Si… | 91 | 1615 | active |
| PaddlePaddle/PaddleSlim PaddleSlim is an open-source library built on PaddlePaddle for deep learning model compression and architecture search. It provides low-bit… | 46 | 1612 | active |
| gorgonia/gorgonia Gorgonia is a Go library for machine learning that lets you define and evaluate mathematical equations over multidimensional arrays using a… | 23 | 5927 | maintenance |
| Chainer Chainer is a Python-based deep learning framework that pioneered the define-by-run approach with dynamic computational graphs and automatic… | 23 | 5919 | maintenance |
| dmlc/gluon-cv GluonCV is a deep learning toolkit providing state-of-the-art computer vision model implementations with 170+ pre-trained models. It suppor… | 23 | 5914 | maintenance |
| MoonshotAI/Kimi-Linear Kimi Linear is a hybrid linear attention architecture (Kimi Delta Attention, based on Gated DeltaNet) released by Moonshot AI with 48B-para… | 37 | 1607 | active |
| autonomousvision/transfuser Official PyTorch implementation of TransFuser, a transformer-based multi-modal sensor fusion model for end-to-end autonomous driving, publi… | 75 | 1606 | stable |
| AlibabaResearch/DAMO-ConvAI The official codebase for Alibaba DAMO Academy's Conversational AI research, containing implementations of models from their published pape… | 75 | 1605 | active |
| Tencent-Hunyuan/HY-WorldPlay HY-WorldPlay (HY-World 1.5) is Tencent Hunyuan's open-source framework for interactive 3D world modeling, generating explorable 3D scenes f… | 52 | 1604 | active |
| ml4a/ml4a ml4a is a Python library and collection of Jupyter notebooks for making art with machine learning. It wraps popular deep learning models li… | 32 | 1604 | active |
| meituan/YOLOv6 YOLOv6 is a single-stage object detection framework implemented in PyTorch, designed for industrial applications with a family of pretraine… | 23 | 5894 | maintenance |
| Drexubery/ViewCrafter ViewCrafter is a research codebase that uses video diffusion models to synthesize high-fidelity novel views of scenes from a single or spar… | 45 | 1602 | active |
| google/uncertainty-baselines A library of high-quality, minimal-dependency implementations of standard and state-of-the-art uncertainty and robustness methods for deep … | 77 | 1601 | active |
| time-series-foundation-models/lag-llama Lag-Llama is the first open-source foundation model for probabilistic time series forecasting, built on a transformer architecture. It prov… | 34 | 1600 | active |
| SalesforceAIResearch/uni2ts Uni2TS is a PyTorch library for unified pre-training, fine-tuning, inference, and evaluation of universal time series forecasting transform… | 54 | 1599 | active |
| alibaba/Pai-Megatron-Patch Pai-Megatron-Patch is an open-source deep learning training toolkit from Alibaba Cloud for large-scale training and inference of LLMs and V… | 50 | 1599 | active |
| Tencent-Hunyuan/HunyuanWorld-Voyager HunyuanWorld-Voyager is a video diffusion framework from Tencent Hunyuan that generates world-consistent RGBD video and 3D point-cloud sequ… | 49 | 1599 | active |
| bilibili/ailab Bilibili's AI lab repository, best known for Real-CUGAN, a deep learning model for anime image super-resolution (upscaling). It provides pr… | 23 | 5877 | maintenance |
| XPixelGroup/HAT HAT (Hybrid Attention Transformer) is a PyTorch implementation of a state-of-the-art transformer model for image super-resolution and resto… | 32 | 1598 | stable |
| X-LANCE/AniTalker AniTalker is the official PyTorch implementation of an ACM MM 2024 paper that animates a single static portrait into a vivid talking-face v… | 25 | 1598 | active |
| SonyResearch/micro_diffusion Official implementation of Sony Research's micro-budget approach to training large-scale text-to-image diffusion transformer models from sc… | 24 | 1598 | active |
| pytorch/FBGEMM FBGEMM is a collection of highly optimized low-precision matrix multiplication and convolution kernels for server-side deep learning infere… | 91 | 1595 | active |
| zju3dv/EasyVolcap EasyVolcap is a PyTorch-based library for accelerating neural volumetric video research, covering volumetric video capture, reconstruction,… | 70 | 1594 | active |
| microsoft/Semi-supervised-learning USB (Unified Semi-supervised learning Benchmark) is a PyTorch-based codebase from Microsoft for semi-supervised learning across computer vi… | 62 | 1590 | active |
| Tencent/DepthCrafter DepthCrafter is a diffusion-based video depth estimation model from Tencent AI Lab that generates temporally consistent long depth sequence… | 34 | 1590 | active |
| FoundationVision/Infinity Infinity is a bitwise autoregressive text-to-image generation model (CVPR 2025 Oral) with released training and inference code, checkpoints… | 53 | 1589 | active |
| DeepGraphLearning/torchdrug TorchDrug is a PyTorch-based machine learning library for drug discovery, covering graph neural networks, deep generative models, and reinf… | 23 | 1589 | active |
| gcorso/DiffDock DiffDock is a deep learning implementation of a diffusion generative model for molecular docking, predicting how small molecule ligands bin… | 28 | 1585 | active |
| meta-pytorch/torchtune Torchtune is a PyTorch-native library for authoring, post-training, and experimenting with large language models. It provides hackable trai… | 68 | 5812 | maintenance |
| RWKV/rwkv.cpp A C++ port of the RWKV language model to the ggml tensor library, providing FP32, FP16, and quantized INT4/INT5/INT8 inference focused on C… | 34 | 1583 | active |
| Xilinx/brevitas Brevitas is a PyTorch library for neural network quantization supporting both post-training quantization (PTQ) and quantization-aware train… | 93 | 1582 | active |
| microsoft/MMdnn MMdnn is a Microsoft toolkit for converting, visualizing, and diagnosing deep learning models across frameworks such as TensorFlow, PyTorch… | 35 | 5802 | maintenance |
| DiffEqML/torchdyn Torchdyn is a PyTorch library dedicated to numerical deep learning, providing tools for neural differential equations, implicit models, and… | 23 | 1580 | active |
| tensorflow/model-optimization The TensorFlow Model Optimization Toolkit (tfmot) is a Python library providing tools to optimize machine learning models for deployment, i… | 79 | 1579 | stable |
| evo-design/evo Evo is a 7-billion-parameter DNA foundation model built on the StripedHyena architecture, trained on ~300 billion tokens of prokaryotic who… | 65 | 1576 | active |
| BloodAxe/pytorch-toolbelt A Python library of PyTorch extensions providing building blocks for fast R&D prototyping, including encoder-decoder architectures, special… | 40 | 1575 | active |