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 |
|---|---|---|---|
| thuml/Transfer-Learning-Library TLlib is a PyTorch-based open-source library for transfer learning, covering domain adaptation, task adaptation (finetuning), and domain ge… | 23 | 3940 | active |
| hustvl/Vim Vision Mamba (Vim) is a PyTorch implementation of a generic vision backbone built on bidirectional Mamba state space models, published at I… | 27 | 3913 | active |
| safetensors/safetensors Safetensors is a simple, secure file format and library for storing and distributing tensors, designed as a fast zero-copy alternative to p… | 87 | 3901 | stable |
| Plachtaa/seed-vc Seed-VC is a Python tool and model for zero-shot voice conversion, real-time voice conversion, and singing voice conversion, cloning a voic… | 10 | 3892 | active |
| vllm-project/llm-compressor LLM Compressor is a Python library for applying quantization and pruning algorithms to large language models, producing compressed-tensors … | 92 | 3886 | active |
| hkust-nlp/simpleRL-reason A research codebase from HKUST-NLP implementing a simple reinforcement learning recipe (rule-based rewards on GSM8K/Math data) to train LLM… | 44 | 3877 | active |
| Stability-AI/stable-audio-tools Stability AI's training and inference toolkit for conditional audio generation models, including Stable Audio Open. It supports training cu… | 73 | 3868 | active |
| open-mmlab/mmpretrain MMPretrain is OpenMMLab's PyTorch-based toolbox and benchmark for image classification model pre-training, covering supervised, self-superv… | 23 | 3855 | active |
| thu-ml/TurboDiffusion TurboDiffusion is a Python framework that accelerates end-to-end video diffusion model generation by 100-200x using SageAttention, Sparse-L… | 58 | 3851 | active |
| Avatarify Avatarify is an open-source application that drives photorealistic avatars in real time for video-conferencing apps like Zoom and Skype, ba… | 23 | 16520 | maintenance |
| mseitzer/pytorch-fid A PyTorch port of the official TensorFlow implementation of the Fréchet Inception Distance (FID), a metric for measuring similarity between… | 23 | 3847 | stable |
| google-research/scenic Scenic is a JAX-based library from Google Research focused on attention-based models for computer vision, providing shared lightweight libr… | 76 | 3842 | active |
| lightly-ai/lightly LightlySSL is a Python library built on PyTorch for self-supervised learning on images, offering modular implementations of methods like Si… | 91 | 3821 | active |
| google/deepvariant DeepVariant is a deep learning-based genomic variant caller that converts aligned DNA sequencing reads (BAM/CRAM) into pileup image tensors… | 78 | 3818 | stable |
| ddbourgin/numpy-ml numpy-ml is a collection of machine learning models and algorithms implemented exclusively in NumPy and the Python standard library, coveri… | 32 | 16323 | maintenance |
| SandAI-org/MAGI-1 MAGI-1 is an open-source autoregressive video generation model from Sand.ai, released with Apache-2.0 licensed code and weights. It generat… | 56 | 3791 | active |
| AI4Finance-Foundation/FinRL-Trading FinRL-X is an open-source, AI-native modular infrastructure for quantitative trading that unifies data processing, strategy composition, ba… | 76 | 3782 | active |
| HazyResearch/ThunderKittens ThunderKittens is a C++/CUDA framework of tile-based primitives for writing fast deep learning GPU kernels. It embeds natively into CUDA so… | 68 | 3754 | active |
| HeartMuLa HeartMuLa is a family of open-source music foundation models that generate music conditioned on lyrics and tags with multilingual support. … | 60 | 3753 | active |
| genmoai/mochi Mochi 1 is Genmo's open-source, state-of-the-art text-to-video generation model released under Apache 2.0, with a Python API, CLI, and Grad… | 67 | 3750 | active |
| fudan-generative-vision/hallo2 Hallo2 is a Python research library from Fudan University that animates a single portrait image using audio input, producing long-duration … | 23 | 3746 | active |
| Polyaxon Polyaxon is an open-source AI engineering control plane that teams self-host on their own Kubernetes clusters to schedule, track, and gover… | 75 | 3735 | active |
| PaddlePaddle/FastDeploy FastDeploy is a high-performance inference and deployment toolkit for large language models and vision-language models built on PaddlePaddl… | 82 | 3719 | active |
| microsoft/Bringing-Old-Photos-Back-to-Life The official PyTorch implementation of 'Bringing Old Photos Back to Life' (CVPR 2020 Oral), a deep learning model that restores old photos … | 23 | 15716 | maintenance |
| Stability-AI/StableLM StableLM is Stability AI's repository of open-weight decoder-only transformer language models, including the 3B-parameter StableLM-3B-4E1T … | 30 | 15671 | maintenance |
| NExT-GPT/NExT-GPT NExT-GPT is an end-to-end any-to-any multimodal large language model that accepts and generates arbitrary combinations of text, image, vide… | 34 | 3638 | active |
| cmusatyalab/openface OpenFace is a free and open source Python and Torch implementation of face recognition based on Google's FaceNet deep neural network. It ge… | 66 | 15443 | maintenance |
| facebookresearch/detr DETR is Facebook Research's PyTorch implementation of Detection Transformer, an end-to-end object detection model that replaces hand-crafte… | 10 | 15352 | maintenance |
| nv-tlabs/kimodo Kimodo is NVIDIA's official implementation of a kinematic motion diffusion model trained on 700 hours of motion capture data to generate hi… | 58 | 3615 | active |
| SAM 3D SAM 3D Body is a promptable model for single-image full-body 3D human mesh recovery (HMR), estimating body, feet, and hand pose using the M… | 46 | 3610 | active |
| NVlabs/Eagle Eagle is NVIDIA's family of frontier vision-language models (Eagle, Eagle 2, Eagle 2.5) built with data-centric training strategies, plus L… | 61 | 3599 | active |
| ZhaoJ9014/face.evoLVe A high-performance face recognition library built on PaddlePaddle and PyTorch, providing comprehensive tools for face-related analytics and… | 34 | 3594 | active |
| NVIDIA/TransformerEngine Transformer Engine is an NVIDIA library for accelerating Transformer model training and inference on NVIDIA GPUs using low-precision format… | 99 | 3576 | active |
| pathwaycom/bdh BDH (Dragon Hatchling) is a biologically inspired large language model architecture that bridges deep learning and neuroscience, implemente… | 51 | 3571 | active |
| SkalskiP/make-sense makesense.ai is a free, browser-based tool for labeling photos to prepare datasets for computer vision projects. It runs entirely client-si… | 23 | 3571 | active |
| Vision Transformer Google Research's official Jax/Flax codebase for training large-scale vision models such as Vision Transformer, SigLIP, MLP-Mixer, and LiT … | 39 | 3551 | active |
| guandeh17/Self-Forcing Official implementation of Self Forcing, a training method for autoregressive video diffusion models that simulates inference during traini… | 34 | 3551 | active |
| First Order Motion Model Official PyTorch/Jupyter implementation of the First Order Motion Model for image animation (NeurIPS 2019). It animates a static source ima… | 32 | 15005 | maintenance |
| microsoft/hummingbird Hummingbird is a Microsoft library that compiles trained traditional ML models (decision trees, random forests, XGBoost, LightGBM) into ten… | 65 | 3545 | active |
| cszn/KAIR A PyTorch image restoration toolbox providing training and testing code for many restoration models including DnCNN, FFDNet, SRMD, USRNet, … | 23 | 3525 | active |
| google-deepmind/alphafold Open-source implementation of the AlphaFold 2 inference pipeline for predicting protein structures from amino acid sequences, including Alp… | 54 | 14887 | maintenance |
| EverMind-AI/MSA MSA (Memory Sparse Attention) is a Python framework for end-to-end trainable sparse latent-memory attention that scales LLM context to 100M… | 50 | 3520 | active |
| visionml/pytracking PyTracking is a PyTorch-based framework for visual object tracking and video object segmentation, providing official implementations of tra… | 23 | 3516 | active |
| google-ai-edge/LiteRT LiteRT is Google's successor to TensorFlow Lite, an on-device runtime for high-performance ML and GenAI inference on edge platforms. It pro… | 86 | 3510 | active |
| huggingface/optimum Optimum is a Hugging Face library that extends Transformers, Diffusers, timm, and Sentence Transformers with hardware-specific optimization… | 91 | 3503 | active |
| imgaug imgaug is a Python library for augmenting images in machine learning experiments, converting a small set of input images into a much larger… | 23 | 14744 | maintenance |
| Tencent-Hunyuan/Hunyuan3D-1 Tencent Hunyuan3D-1.0 is an open-source two-stage diffusion-based model for generating 3D assets from text prompts or images. It provides i… | 43 | 3489 | active |
| facebookresearch/ijepa Official PyTorch implementation of I-JEPA, a self-supervised learning method that predicts latent representations of image regions from oth… | 10 | 3488 | active |
| POSTECH-CVLab/PyTorch-StudioGAN PyTorch-StudioGAN is a PyTorch library providing unified implementations of representative GAN architectures (BigGAN, StyleGAN2/3, etc.) fo… | 23 | 3483 | stable |
| PennyLaneAI/pennylane PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. It lets users build an… | 88 | 3476 | stable |
| MiniMax AI Platform Official repository for MiniMax-Text-01 and MiniMax-VL-01, open-weight large language and vision-language models built on a linear attentio… | 31 | 3466 | active |
| shenweichen/DeepCTR-Torch DeepCTR-Torch is a PyTorch library providing easy-to-use, modular, and extendable implementations of deep-learning-based CTR (click-through… | 71 | 3460 | active |
| PaddlePaddle/PARL PARL is a flexible, high-performance reinforcement learning framework built on PaddlePaddle, providing Model/Algorithm/Agent abstractions a… | 38 | 3457 | active |
| XinJingHao/DRL-Pytorch A unified PyTorch implementation collection of popular deep reinforcement learning algorithms including DQN variants, PPO, DDPG, TD3, SAC, … | 40 | 3447 | active |
| aqlaboratory/openfold OpenFold is a faithful, trainable PyTorch reproduction of DeepMind's AlphaFold 2 for protein structure prediction. It is memory-efficient a… | 45 | 3435 | active |
| NVlabs/stylegan The official TensorFlow implementation of StyleGAN, NVIDIA's style-based generator architecture for generative adversarial networks from th… | 32 | 14414 | maintenance |
| microsoft/nni NNI (Neural Network Intelligence) is an open-source AutoML toolkit from Microsoft that automates hyperparameter tuning, neural architecture… | 10 | 14360 | maintenance |
| davidsandberg/facenet A TensorFlow implementation of the FaceNet face recognizer that generates 128-dimensional face embeddings, including face detection via MTC… | 32 | 14357 | maintenance |
| NovaSky-AI/SkyThought SkyThought is the open-source repository behind Sky-T1, a family of reasoning language models trained for under $450, including training sc… | 22 | 3401 | active |
| opengeos/geoai GeoAI is a Python package that integrates artificial intelligence with geospatial data analysis, built on PyTorch, Transformers, and segmen… | 89 | 3390 | active |
| magenta/ddsp DDSP is a Python library of differentiable digital signal processing components (synthesizers, filters, waveshapers) that can be embedded i… | 68 | 3380 | active |
| CompVis/latent-diffusion The official research code and pretrained model zoo for Latent Diffusion Models (LDM), the paper behind Stable Diffusion, enabling high-res… | 32 | 14168 | maintenance |
| WongKinYiu/yolov7 Official PyTorch implementation of the YOLOv7 paper, a state-of-the-art real-time object detector with trainable bag-of-freebies techniques… | 23 | 14153 | maintenance |
| PhysicsNeMo NVIDIA PhysicsNeMo is an open-source Python deep-learning framework for building, training, fine-tuning, and inferring physics AI models us… | 93 | 3373 | active |
| libAudioFlux/audioFlux audioFlux is a C-based library with Python bindings for audio and music analysis and feature extraction. It supports dozens of time-frequen… | 50 | 3372 | active |
| OpenTalker/SadTalker SadTalker is a CVPR 2023 deep learning tool that generates realistic talking head videos from a single portrait image and an audio clip by … | 23 | 14100 | maintenance |
| VainF/Torch-Pruning Torch-Pruning is a PyTorch framework for structural neural network pruning based on the DepGraph algorithm from CVPR 2023. It automatically… | 44 | 3359 | active |
| Peterande/D-FINE D-FINE is the official PyTorch implementation of an ICLR 2025 Spotlight paper that redefines the regression task in DETR-style detectors as… | 64 | 3334 | active |
| microsoft/LoRA loralib is the official PyTorch implementation of LoRA (Low-Rank Adaptation), which fine-tunes large language models by injecting trainable… | 23 | 13805 | maintenance |
| facebookresearch/tribev2 TRIBE v2 is a multimodal deep learning model from Meta AI that predicts fMRI brain responses to naturalistic video, audio, and text stimuli… | 52 | 3293 | active |
| Tencent-Hunyuan/HunyuanImage-3.0 HunyuanImage-3.0 is Tencent's open-source native multimodal model for text-to-image and image-to-image generation, with inference code and … | 54 | 3279 | active |
| jixiaozhong/Sonic Sonic is the official PyTorch implementation of the CVPR 2025 paper 'Sonic: Shifting Focus to Global Audio Perception in Portrait Animation… | 46 | 3273 | active |
| LeelaChessZero/lc0 Lc0 is an open-source, UCI-compliant chess engine that plays chess using neural networks trained via AlphaZero-style self-play reinforcemen… | 74 | 3248 | active |
| Beckschen/TransUNet Official PyTorch implementation of TransUNet, a U-Net-style architecture that uses a Vision Transformer encoder for medical image segmentat… | 60 | 3245 | stable |
| determined-ai/determined Determined is an open-source deep learning platform that combines distributed training, hyperparameter tuning, experiment tracking, and GPU… | 36 | 3245 | active |
| Nerogar/OneTrainer OneTrainer is a GUI and CLI application for fine-tuning diffusion image models, supporting full fine-tuning, LoRA, and embeddings across ma… | 75 | 3243 | active |
| MisoLabsAI/MisoTTS Miso TTS 8B is an open-source text-to-speech model based on an RVQ Transformer architecture with a Llama 3.2-style 8B backbone, designed fo… | 49 | 3240 | active |
| onnx/onnx-tensorrt A C++ parser library and backend that converts ONNX models into TensorRT engines for high-performance GPU inference. It is maintained by NV… | 91 | 3238 | active |
| Jittor/jittor Jittor is a high-performance deep learning framework from Tsinghua University based on just-in-time (JIT) compilation and meta-operators, w… | 68 | 3237 | active |
| Pointcept Pointcept is a PyTorch-based research codebase for point cloud perception, providing implementations of state-of-the-art 3D scene understan… | 74 | 3233 | active |
| prs-eth/Marigold Marigold is a family of diffusion-based models and a fine-tuning protocol that adapts pretrained latent diffusion models like Stable Diffus… | 69 | 3233 | active |
| mit-han-lab/bevfusion BEVFusion is a PyTorch-based multi-task multi-sensor fusion framework that unifies camera and LiDAR features in a shared bird's-eye view re… | 10 | 3231 | stable |
| VMamba VMamba is a PyTorch implementation of a visual state space model (SSM) vision backbone based on Mamba, featuring 2D Selective Scan (SS2D) f… | 18 | 3230 | active |
| facebookresearch/dinov2 PyTorch implementation and pretrained models for DINOv2, a self-supervised vision transformer method from Meta AI that learns robust visual… | 75 | 13422 | maintenance |
| jy0205/Pyramid-Flow Pyramid Flow is the official PyTorch implementation of a training-efficient autoregressive video generation model based on pyramidal flow m… | 23 | 3213 | active |
| webdataset/webdataset A Python library providing a high-performance sequential I/O system based on tar-shard files for large-scale deep learning training, with s… | 49 | 3204 | stable |
| Arm NN Arm's Compute Library is a C++ collection of over 100 low-level machine learning and computer vision functions optimized for Arm Cortex-A/N… | 98 | 3197 | active |
| imbue-bit/AlphaGPT AlphaGPT is an open-source automated factor factory based on deep reinforcement learning for quantitative finance. It mines and generates a… | 52 | 3187 | active |
| MiniMax AI Platform MiniMax-M1 is an open-weight, large-scale hybrid-attention reasoning language model released by MiniMax under Apache-2.0. The repository pr… | 29 | 3185 | active |
| stepfun-ai/Step-Video-T2V Step-Video-T2V is an open-source text-to-video generation model from StepFun, released with inference code and pretrained weights (includin… | 22 | 3185 | active |
| Rudrabha/Wav2Lip Wav2Lip is the official research code for the ACM Multimedia 2020 paper 'A Lip Sync Expert Is All You Need for Speech to Lip Generation In … | 41 | 13221 | maintenance |
| tekaratzas/RustGPT A transformer-based large language model implemented entirely in pure Rust with no external ML frameworks, using only ndarray for matrix op… | 35 | 3175 | active |
| megvii-research/NAFNet NAFNet is the official PyTorch implementation of a state-of-the-art image restoration network that removes nonlinear activation functions. … | 32 | 3170 | stable |
| ali-vilab/VGen VGen is the official repository for a holistic video generation ecosystem built on diffusion models, including the I2VGen-XL cascaded image… | 28 | 3154 | active |
| cleardusk/3DDFA_V2 3DDFA_V2 is the official PyTorch implementation of the ECCV 2020 paper 'Towards Fast, Accurate and Stable 3D Dense Face Alignment'. It regr… | 23 | 3150 | stable |
| modelscope/3D-Speaker 3D-Speaker is an open-source Python toolkit for single- and multi-modal speaker verification, speaker recognition, and speaker diarization,… | 53 | 3148 | active |
| dexmal/dexbotic Dexbotic is an open-source PyTorch-based toolbox for developing Vision-Language-Action (VLA) models for embodied intelligence. It unifies p… | 70 | 3144 | active |
| docarray/docarray DocArray is a Python library for representing, transmitting, storing, and retrieving multimodal data such as text, images, audio, and video… | 10 | 3125 | active |
| thuml/Time-Series-Library TSLib is an open-source Python library providing a unified codebase of advanced deep learning models for general time series analysis. It s… | 63 | 12923 | maintenance |
| naver/mast3r MASt3R is the official PyTorch implementation of 'Grounding Image Matching in 3D with MASt3R' (ECCV 2024), a model that performs dense 3D r… | 34 | 3114 | active |