Open-source deep-learning software
2785 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.
2785 products, primary matches first, then adoption-weighted; health v2 shown.
| Product | Health v2 | Stars | Maturity |
|---|---|---|---|
| LAMDA-CL/PyCIL PyCIL is a PyTorch-based Python toolbox for class-incremental learning, implementing the largest collection of CIL methods for reproducible… | 48 | 1112 | active |
| BytedTsinghua-SIA/MemAgent MemAgent is a reinforcement-learning framework for training LLM agents that process arbitrarily long contexts via a memory mechanism within… | 52 | 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 | 1109 | stable |
| WangLibo1995/GeoSeg GeoSeg is an open-source PyTorch-based semantic segmentation toolbox focused on Vision Transformers for remote sensing imagery, featuring t… | 32 | 1109 | active |
| google-ai-edge/litert-torch LiteRT Torch is a Python library that converts PyTorch models into .tflite format for running with LiteRT on Android, iOS, and IoT devices.… | 86 | 1107 | active |
| rusty1s/pytorch_sparse A PyTorch extension library providing optimized sparse matrix operations (coalesce, transpose, sparse-dense and sparse-sparse multiplicatio… | 57 | 1106 | active |
| PyABSA PyABSA is a PyTorch-based library providing state-of-the-art models for aspect-based sentiment analysis, including aspect term extraction, … | 66 | 1105 | active |
| TencentARC/T2I-Adapter Official implementation of T2I-Adapter, lightweight adapter models that add controllable conditioning (sketch, canny, lineart, depth, pose)… | 31 | 3799 | maintenance |
| 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 |
| geoopt/geoopt Geoopt is a PyTorch library providing manifold-aware optimizers and manifold tensor containers for Riemannian optimization. It implements R… | 67 | 1100 | active |
| lucidrains/tab-transformer-pytorch A PyTorch implementation of the TabTransformer architecture, an attention-based neural network for tabular data, also including the FT Tran… | 64 | 1099 | stable |
| NVIDIA/DreamDojo NVIDIA's official PyTorch codebase for DreamDojo, a generalist robot world model pretrained on 44k hours of human egocentric video and post… | 45 | 1099 | 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-… | 25 | 1099 | 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 | 3779 | maintenance |
| welltop-cn/ComfyUI-TeaCache A ComfyUI plugin integrating TeaCache, a training-free caching method that accelerates diffusion model inference by exploiting output diffe… | 32 | 1097 | 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.… | 39 | 1096 | active |
| YOLOs-CPP YOLOs-CPP is a production-ready, cross-platform C++ inference library for the YOLO model family (v5 through YOLO26), built on ONNX Runtime … | 82 | 1095 | active |
| Soul-AILab/SoulX-FlashHead SoulX-FlashHead is a 1.3B-parameter framework for high-fidelity, infinite-length, real-time streaming talking-head portrait video generatio… | 50 | 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 | 3756 | maintenance |
| NVlabs/Fast-dLLM NVIDIA's official implementation of Fast-dLLM, a family of training-free and fine-tuning-based acceleration techniques for diffusion-based … | 54 | 1090 | active |
| zju3dv/InfiniDepth InfiniDepth is a CVPR 2026 research library for monocular depth estimation that represents depth as neural implicit fields, allowing depth … | 59 | 1089 | active |
| mymusise/ChatGLM-Tuning A Python toolkit for fine-tuning the ChatGLM-6B large language model using LoRA (Low-Rank Adaptation) with the Alpaca dataset. It provides … | 10 | 3738 | maintenance |
| rhymes-ai/Aria Aria is an open multimodal native Mixture-of-Experts (MoE) model with 25.3B total parameters (3.9B activated per token) and a 64K multimoda… | 23 | 1088 | active |
| agi-brain/xuance XuanCe is an open-source Python library of deep reinforcement learning (DRL) and multi-agent reinforcement learning (MARL) algorithm implem… | 91 | 1087 | active |
| yerfor/Real3DPortrait Official PyTorch implementation of Real3D-Portrait, an ICLR 2024 Spotlight paper for one-shot realistic 3D talking portrait synthesis. It g… | 26 | 1087 | active |
| PointNeXt PointNeXt is the official PyTorch implementation of the NeurIPS'22 paper that improves PointNet++ via better training and model scaling str… | 87 | 1086 | stable |
| microsoft/Biodiversity Microsoft AI for Good Lab's biodiversity research hub providing open-source AI models and tools for wildlife monitoring and conservation, i… | 86 | 1086 | active |
| DSE-MSU/DeepRobust DeepRobust is a PyTorch library for adversarial robustness research, providing implementations of attack and defense methods for both image… | 41 | 1086 | active |
| GraphSAGE Reference implementation of the GraphSAGE algorithm for inductive representation learning on large graphs using stochastic graph convolutio… | 32 | 3728 | maintenance |
| mert-kurttutan/torchview torchview is a Python library that visualizes PyTorch models as visual graphs, showing tensors, modules, torch functions, and input/output … | 50 | 1084 | active |
| Alpha-VLLM/Lumina-mGPT-2.0 Lumina-mGPT 2.0 is a stand-alone decoder-only autoregressive model trained from scratch that unifies a broad range of image generation task… | 39 | 1084 | active |
| kimiyoung/transformer-xl Official implementation of Transformer-XL, an attention-based language model architecture that extends context beyond a fixed length via se… | 32 | 3717 | maintenance |
| bytedance/byteps BytePS is a high-performance parameter server framework for distributed deep neural network training, supporting TensorFlow, Keras, PyTorch… | 10 | 3717 | maintenance |
| vladmandic/face-api FaceAPI is a JavaScript library built on TensorFlow/JS that provides AI-powered face detection, rotation tracking, face description and rec… | 10 | 1082 | active |
| charlesq34/pointnet2 Official TensorFlow implementation of PointNet++, a deep neural network that learns hierarchical features on 3D point clouds using metric-s… | 32 | 3712 | maintenance |
| meta-pytorch/monarch Monarch is a distributed programming framework for PyTorch built on scalable actor messaging, with actors grouped into meshes, supervision-… | 83 | 1081 | active |
| Stonesjtu/pytorch_memlab A Python library providing line-level CUDA memory profiling and tensor inspection tools for PyTorch. It helps debug out-of-memory errors by… | 88 | 1080 | active |
| LeCAR-Lab/human2humanoid Official implementation of H2O and OmniH2O, reinforcement-learning-based systems for real-time whole-body teleoperation of full-sized human… | 23 | 1078 | active |
| FINN FINN is an open-source dataflow compiler from AMD/Xilinx that generates highly efficient FPGA accelerators for quantized neural network (QN… | 85 | 1077 | active |
| facebookresearch/hiera Hiera is the official PyTorch implementation of a hierarchical vision transformer from Meta AI (ICML 2023 Oral). It achieves state-of-the-a… | 20 | 1077 | active |
| arcee-ai/DistillKit DistillKit is an open-source Python toolkit for knowledge distillation of large language models, supporting both online and offline distill… | 57 | 1076 | active |
| facebookresearch/CutLER CutLER is a research codebase from Meta FAIR for training object detection and instance segmentation models without human annotations, usin… | 63 | 1075 | active |
| EchoMimic EchoMimic is a series of open-source models (V1-V3) from Ant Group for audio-driven human animation, generating lifelike talking-head, port… | 47 | 1075 | active |
| AILab-CVC/UniRepLKNet UniRepLKNet is a large-kernel ConvNet architecture (CVPR 2024, TPAMI 2025) that provides universal perception across image, audio, video, p… | 40 | 1074 | stable |
| THUDM/GLM GLM is a general language model pretrained with an autoregressive blank-filling objective, released with pretrained checkpoints and fine-tu… | 32 | 3680 | maintenance |
| cleardusk/3DDFA A PyTorch implementation of the TPAMI 2017 paper 'Face Alignment in Full Pose Range: A 3D Total Solution' (3DDFA). It fits a 3D Morphable M… | 23 | 3678 | maintenance |
| NVIDIA-Merlin/HugeCTR HugeCTR is a GPU-accelerated deep learning framework from NVIDIA designed for training and inference of large recommender models, especiall… | 71 | 1073 | active |
| open-gigaai/giga-train GigaTrain is an efficient and scalable Python training framework for large AI models, supporting distributed multi-GPU/multi-node execution… | 59 | 1072 | active |
| memoavatar/memo MEMO is an open-weight diffusion model for generating expressive, identity-consistent talking videos from a single reference image and an a… | 37 | 1071 | active |
| NVlabs/DiffusionNFT DiffusionNFT is a research library implementing an online reinforcement learning paradigm for diffusion models that optimizes policy direct… | 44 | 1070 | active |
| NVlabs/SegFormer Official PyTorch implementation of SegFormer, a transformer-based semantic segmentation framework with a hierarchical encoder and lightweig… | 32 | 3661 | maintenance |
| ZhengYinan-AIR/Diffusion-Planner Official PyTorch implementation of Diffusion Planner, an ICLR 2025 Oral paper using a DiT-based diffusion model for closed-loop autonomous … | 50 | 1069 | active |
| gangweix/pixel-perfect-depth Pixel-Perfect Depth is a monocular depth estimation model based on pixel-space diffusion transformers that produces flying-pixel-free depth… | 46 | 1069 | active |
| juntang-zhuang/Adabelief-Optimizer AdaBelief is a deep learning optimizer that adapts step sizes based on the 'belief' in observed gradients, combining Adam's fast convergenc… | 32 | 1069 | stable |
| yeates/PromptFix PromptFix is a PyTorch implementation of a diffusion-model-based image restoration model that follows natural language instructions to fix … | 25 | 1069 | active |
| lucidrains/mlp-mixer-pytorch A PyTorch implementation of Google AI's MLP-Mixer, an all-MLP architecture for image classification that uses neither convolutions nor atte… | 44 | 1068 | active |
| thuiar/MMSA MMSA is a unified Python framework for multimodal sentiment analysis, supporting 15 MSA models and datasets like MOSI, MOSEI, and CH-SIMS. … | 23 | 1067 | active |
| princeton-nlp/SimCSE SimCSE is a Python library and research codebase implementing simple contrastive learning for sentence embeddings, with pre-trained unsuper… | 23 | 3652 | maintenance |
| hujie-frank/SENet Official Caffe/CUDA implementation of Squeeze-and-Excitation Networks (SENet), channel-attention building blocks for convolutional neural n… | 32 | 3651 | maintenance |
| drprojects/superpoint_transformer Official PyTorch implementation of Superpoint Transformer (ICCV'23), SuperCluster (3DV'24), and EZ-SP (ICRA'26) for efficient semantic and … | 74 | 1063 | active |
| vijishmadhavan/ArtLine ArtLine is a deep learning project that converts portrait photos into line art portraits, with a ControlNet-based variant that adjusts styl… | 32 | 3634 | maintenance |
| henry123-boy/SpaTracker SpatialTracker is the official PyTorch implementation of a CVPR 2024 Highlight paper that tracks any 2D pixels in 3D space from RGB or RGBD… | 38 | 1062 | active |
| danijar/dreamerv2 A TensorFlow 2 implementation of the DreamerV2 model-based reinforcement learning agent that learns world models from high-dimensional imag… | 32 | 1062 | stable |
| zhuzilin/ring-flash-attention A Python library implementing RingAttention on top of FlashAttention for distributed long-context transformer training. It provides varlen … | 38 | 1061 | active |
| abertsch72/unlimiformer Unlimiformer is a method and official implementation for augmenting pretrained encoder-decoder transformers with retrieval-based attention,… | 30 | 1061 | stable |
| open-gigaai/giga-models GigaModels is an open-source Python framework providing pipelines for training, inference, deployment, and compression of multi-modal, gene… | 59 | 1060 | active |
| NousResearch/DisTrO DisTrO is a family of low-latency distributed optimizers that reduce inter-GPU communication requirements by three to four orders of magnit… | 41 | 1059 | active |
| mgsalem/Tensorflow-Project-Template A Python project template that provides a recommended folder structure and object-oriented skeleton (base model, base trainer, data loader,… | 32 | 3616 | maintenance |
| YunYang1994/tensorflow-yolov3 A TensorFlow 1.x implementation of the YOLOv3 real-time object detector, reproducing the 'YOLOv3: An Incremental Improvement' paper. It sup… | 23 | 3614 | maintenance |
| yoyo-nb/Thin-Plate-Spline-Motion-Model The official PyTorch implementation of the CVPR 2022 paper 'Thin-Plate Spline Motion Model for Image Animation'. It animates a source image… | 32 | 3607 | maintenance |
| apache/singa Apache SINGA is a distributed deep learning platform for training neural networks across multiple devices and machines. It provides a C++ c… | 64 | 3606 | maintenance |
| showlab/MotionDirector MotionDirector is a research library for customizing text-to-video diffusion models to generate videos with desired motions from a small se… | 28 | 1054 | active |
| zai-org/SCAIL SCAIL is the official inference implementation of a 14B diffusion transformer model that generates studio-grade character animation videos … | 49 | 1053 | active |
| zai-org/GLM-Image GLM-Image is an open-source image generation model combining a 9B autoregressive generator with a 7B diffusion decoder, excelling at text r… | 45 | 1053 | active |
| zju3dv/EfficientLoFTR Efficient LoFTR is a PyTorch implementation of a semi-dense local feature matching model that matches keypoints between image pairs with sp… | 68 | 1050 | active |
| qqlu/Entity EntitySeg is an open-source PyTorch toolbox for open-world, high-quality image segmentation, built on Detectron2. It aggregates multiple re… | 32 | 1049 | active |
| williamyang1991/VToonify Official PyTorch implementation of VToonify, a SIGGRAPH Asia 2022 framework for controllable high-resolution portrait video style transfer … | 32 | 3579 | maintenance |
| tonbistudio/turboquant-pytorch A from-scratch PyTorch implementation of Google's TurboQuant (ICLR 2026) algorithm for compressing LLM key-value caches, including an impro… | 47 | 1047 | active |
| 3DTopia/3DTopia-XL 3DTopia-XL is a 3D diffusion transformer model that generates high-quality 3D assets with PBR materials from a single image or text prompt … | 34 | 1047 | active |
| jiachenzhu/DyT Official PyTorch implementation of DynamicTanh (DyT), a learnable element-wise tanh operation that replaces normalization layers in Transfo… | 23 | 1047 | active |
| facebookresearch/pytorchvideo PyTorchVideo is a deep learning library from Facebook Research focused on video understanding research, built on PyTorch. It provides reusa… | 55 | 3570 | maintenance |
| TransNet TransNet V2 is a deep neural network for shot boundary detection in videos, achieving state-of-the-art results on benchmarks like ClipShots… | 32 | 1046 | stable |
| fudan-zvg/4d-gaussian-splatting Official PyTorch/CUDA implementation of 4D Gaussian Splatting (ICLR 2024), which represents and renders dynamic scenes in real time using 4… | 54 | 1045 | active |
| Tencent-Hunyuan/InstantCharacter InstantCharacter is a tuning-free framework built on diffusion transformers that generates character-consistent images from a single refere… | 26 | 1044 | active |
| Shawn1993/cnn-text-classification-pytorch A PyTorch implementation of Kim's CNN architecture for sentence classification, reproducing results from the paper 'Convolutional Neural Ne… | 62 | 1043 | active |
| PetarV-/GAT Reference implementation of Graph Attention Networks (GAT), the ICLR 2018 model by Veličković et al., written in TensorFlow 1.x with a mini… | 32 | 3553 | maintenance |
| foolwood/SiamMask Official PyTorch implementation of SiamMask, a deep learning framework for fast online visual object tracking and video object segmentation… | 32 | 3553 | maintenance |
| pytorch/tensordict TensorDict is a PyTorch library providing a batched, nested dict-like tensor container where operations like slicing, stacking, and arithme… | 100 | 1042 | active |
| OpenMined/TenSEAL TenSEAL is a Python library (with a C++ core) for performing homomorphic encryption operations on tensors, built on top of Microsoft SEAL. … | 96 | 1039 | active |
| Xiaoqi-Zhao-DLUT/MSNet-M2SNet Official PyTorch implementations of MSNet and M2SNet, multi-scale subtraction networks for medical image segmentation such as polyp, lung i… | 70 | 1039 | active |
| InSpatio-World InSpatio-World is a Python library and model for world-model video generation, providing a multi-step inference pipeline that combines vide… | 54 | 1039 | active |
| TorchEnsemble-Community/Ensemble-Pytorch A unified ensemble learning framework for PyTorch that implements strategies like voting, bagging, and gradient boosting to improve deep le… | 23 | 1038 | active |
| kuleshov-group/bd3lms BD3-LMs is a research implementation of Block Discrete Denoising Diffusion Language Models that interpolate between autoregressive and diff… | 31 | 1037 | active |
| tensorflow/hub TensorFlow Hub is a Python library for reusing parts of trained TensorFlow models (SavedModels) for transfer learning, wrapping them as Ker… | 23 | 3530 | maintenance |
| google/ml_collections ML Collections is a Python library providing dict-like data structures (ConfigDict and FrozenConfigDict) for managing ML experiment and mod… | 63 | 1036 | stable |
| graspnet/graspnet-baseline The official baseline deep learning model for the GraspNet-1Billion benchmark, detecting dense 6-DoF grasp poses from point clouds of clutt… | 32 | 1036 | stable |
| 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 … | 58 | 1035 | active |
| podgorskiy/ALAE Official PyTorch implementation of Adversarial Latent Autoencoders (ALAE/StyleALAE), a CVPR 2020 paper combining autoencoders with GAN trai… | 32 | 3509 | maintenance |
| Jumpat/SegmentAnythingin3D SA3D is a research framework that lifts 2D Segment Anything (SAM) masks into 3D segmentation of objects within a NeRF or 3D Gaussian Splatt… | 37 | 1031 | active |
| szagoruyko/pytorchviz A small Python package (torchviz) that generates visualizations of PyTorch execution graphs and autograd traces using Graphviz. It renders … | 32 | 3505 | maintenance |