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 |
|---|---|---|---|
| jadore801120/attention-is-all-you-need-pytorch A PyTorch implementation of the Transformer model from the paper 'Attention Is All You Need', including training and translation scripts fo… | 32 | 9816 | maintenance |
| Alibaba-Quark/LiveAvatar LiveAvatar is an open-source implementation of an ECCV 2026 paper for streaming, real-time, infinite-length audio-driven avatar video gener… | 58 | 2456 | active |
| facebookresearch/pifuhd PIFuHD is a PyTorch implementation of a CVPR 2020 research model that reconstructs high-resolution 3D human body meshes from a single 2D im… | 10 | 9724 | maintenance |
| TorchIO-project/torchio TorchIO is a Python library for loading, augmenting, and processing 3D medical images (MRI, CT) within PyTorch deep learning pipelines. It … | 91 | 2446 | active |
| AI-Hypercomputer/maxtext MaxText is a high-performance, scalable open-source LLM training library written in pure Python/JAX, targeting Google Cloud TPUs and GPUs. … | 98 | 2444 | active |
| shenweichen/DeepMatch DeepMatch is a Python library of deep matching models for recommendations and advertising, built on TensorFlow/Keras. It lets users train m… | 66 | 2435 | active |
| resemble-ai/resemble-enhance Resemble Enhance is an AI-powered Python tool that improves speech quality through denoising and enhancement, using a denoiser module and a… | 18 | 2434 | active |
| wolny/pytorch-3dunet A PyTorch implementation of 3D U-Net and its variants (residual, squeeze-and-excitation) for volumetric semantic segmentation, with 2D U-Ne… | 57 | 2423 | active |
| HuCaoFighting/Swin-Unet Official PyTorch implementation of Swin-Unet, a U-shaped pure Transformer model for medical image segmentation, published at ECCV 2022 Medi… | 38 | 2420 | stable |
| tflearn/tflearn TFLearn is a modular deep learning library providing a higher-level, Keras-like API on top of TensorFlow for building and training neural n… | 23 | 9570 | maintenance |
| X-PLUG/mPLUG-DocOwl mPLUG-DocOwl is a family of multimodal large language models from Alibaba for OCR-free document understanding, including DocOwl1.5 and DocO… | 37 | 2413 | active |
| OpenGVLab/InternVideo InternVideo is a series of open-source video foundation models for multimodal video understanding, spanning generative and discriminative l… | 69 | 2393 | active |
| zai-org/GLM-V GLM-V is the open-source repository for Zhipu AI's GLM-4.6V, GLM-4.5V, and GLM-4.1V-Thinking vision-language models, which perform versatil… | 63 | 2393 | active |
| google/neural-tangents Neural Tangents is a Python library built on JAX for defining, training, and evaluating neural networks of both finite and infinite width. … | 10 | 2385 | stable |
| TencentARC/Pixal3D Pixal3D is a research codebase for generating high-fidelity 3D assets from a single image using a pixel-aligned generation paradigm that ba… | 56 | 2383 | active |
| facebookresearch/perception_models Meta's Perception Models repository hosting state-of-the-art image, video, and audio encoders (Perception Encoder, PE) and a multimodal lan… | 51 | 2382 | active |
| apple/axlearn AXLearn is a Python deep learning library built on JAX and XLA for developing and training large-scale models, with an object-oriented conf… | 69 | 2375 | active |
| EasyRec EasyRec is a TensorFlow-based framework from Alibaba for building large-scale deep learning recommendation models covering matching, rankin… | 59 | 2361 | active |
| tencent-ailab/V-Express V-Express is a Python research project from Tencent AI Lab that generates talking head portrait videos from a reference image, audio, and V… | 25 | 2357 | active |
| Neural Circuit Policies (ncps) A Python package providing PyTorch and TensorFlow/Keras implementations of Neural Circuit Policies (NCPs), including liquid time-constant (… | 23 | 2351 | active |
| google-deepmind/optax Optax is a gradient processing and optimization library for JAX, offering composable building blocks like optimizers and loss functions. It… | 80 | 2350 | stable |
| ASLP-lab/DiffRhythm DiffRhythm is an open-source latent diffusion model for end-to-end full-length song generation, supporting lyrics-to-song, text style promp… | 42 | 2344 | active |
| SkyworkAI/Matrix-Game Matrix-Game is Skywork AI's open-source series of interactive world foundation models that generate real-time, streaming video in response … | 50 | 2342 | active |
| YvanYin/Metric3D Metric3D is the official PyTorch implementation of Metric3Dv1 and Metric3Dv2, monocular geometric foundation models that predict metric dep… | 31 | 2326 | active |
| Picotron Picotron is a minimalist, hackable distributed training framework for pre-training Llama-like large language models using 4D parallelism (d… | 37 | 2325 | active |
| PKU-YuanGroup/MoE-LLaVA MoE-LLaVA is an open-source Mixture-of-Experts based sparse large vision-language model, released with the MoE-Tuning training strategy fro… | 28 | 2324 | active |
| Cadene/pretrained-models.pytorch A Python library providing pretrained ConvNet models (ResNet, ResNeXt, InceptionV4, Xception, NASNet, SENet, DPN, etc.) for PyTorch behind … | 32 | 9101 | maintenance |
| nv-tlabs/lyra Project Lyra is NVIDIA's open series of generative 3D world models, including Lyra 1.0 for feed-forward 3D/4D scene generation from a singl… | 56 | 2317 | active |
| facebookresearch/ImageBind A PyTorch library from Meta AI implementing ImageBind, a model that learns a joint embedding space across six modalities: images, text, aud… | 51 | 9076 | maintenance |
| OpenHelix-Team/VLA-Adapter VLA-Adapter is the official implementation of a tiny-scale vision-language-action model that bridges vision-language representations to rob… | 47 | 2313 | active |
| frgfm/torch-cam TorchCAM is a Python library that extracts class activation maps (CAMs) from PyTorch CNN classifiers, supporting many CAM variants such as … | 88 | 2307 | active |
| traveller59/spconv SpConv is a spatially sparse convolution library for deep learning on 3D point clouds and sparse tensors, distributed as PyPI packages with… | 32 | 2299 | active |
| NVlabs/nvdiffrec nvdiffrec is NVIDIA's official implementation of a CVPR 2022 oral paper that jointly optimizes triangular 3D meshes, PBR materials, and lig… | 72 | 2298 | stable |
| codeplea/genann Genann is a minimal, well-tested C99 library for training and running feedforward artificial neural networks. It is contained in a single s… | 91 | 2293 | stable |
| aigc-apps/VideoX-Fun VideoX-Fun is a Python-based video generation pipeline built on Diffusion Transformer models (CogVideoX-Fun, Wan-Fun) that generates videos… | 68 | 2292 | active |
| facebookresearch/fairchem FAIR Chemistry's centralized Python library of machine learning models, datasets, and applications for materials science and quantum chemis… | 98 | 2289 | active |
| hkchengrex/MMAudio MMAudio is a PyTorch-based model for generating synchronized audio from video and/or text inputs, using multimodal joint training across au… | 40 | 2287 | active |
| opendatalab/DocLayout-YOLO DocLayout-YOLO is a real-time YOLO-v10-based model for detecting document layout elements (text blocks, tables, figures, etc.) in diverse d… | 26 | 2277 | active |
| RosettaCommons/RoseTTAFold RoseTTAFold is the official implementation of a deep learning system for predicting protein structures and interactions using a three-track… | 23 | 2269 | stable |
| ashawkey/stable-dreamfusion A PyTorch implementation of Dreamfusion that generates 3D models from text prompts or images using NeRF combined with Stable Diffusion guid… | 23 | 8865 | maintenance |
| stepfun-ai/Step1X-Edit Step1X-Edit is an open-source state-of-the-art instruction-based image editing model from StepFun, designed to rival closed-source editors … | 52 | 2266 | active |
| aigc-apps/EasyAnimate EasyAnimate is an end-to-end Python pipeline for high-resolution, long video and image generation based on transformer diffusion (DiT) mode… | 17 | 2266 | active |
| 666DZY666/micronet micronet is a Python library for deep neural network model compression and deployment built on PyTorch. It provides quantization (QAT, PTQ,… | 38 | 2265 | active |
| galilai-group/stable-worldmodel A Python library providing a unified platform for reproducible world model research, covering data collection, training, and evaluation via… | 78 | 2260 | active |
| fishaudio/Bert-VITS2 Bert-VITS2 is a text-to-speech model implementation combining the VITS2 architecture with multilingual BERT embeddings, written in Python. … | 64 | 8805 | maintenance |
| facebookresearch/fvcore fvcore is a lightweight Python core library providing common functionality shared across FAIR's computer vision frameworks such as Detectro… | 73 | 2254 | stable |
| CoinCheung/pytorch-loss A PyTorch library providing a collection of loss functions (focal loss, triplet loss, AMSoftmax, label-smooth CE, dice loss, lovasz-softmax… | 32 | 2251 | active |
| Alpha-VLLM/Lumina-T2X Lumina-T2X is a unified framework for text-to-any-modality generation built on flow-based large diffusion transformers. It supports generat… | 25 | 2250 | active |
| ByteDance-Seed/VeOmni VeOmni is a PyTorch-native framework for single- and multi-modal model pre-training and post-training, with a modular, trainer-free design … | 87 | 2246 | active |
| Raster Vision Raster Vision is an open source Python library and low-code framework for building computer vision models on satellite, aerial, and other l… | 57 | 2241 | active |
| NVlabs/MambaVision MambaVision is NVIDIA's official PyTorch implementation of a hybrid Mamba-Transformer vision backbone, published at CVPR 2025. It provides … | 46 | 2235 | active |
| ellisdg/3DUnetCNN A PyTorch library for building, training, and applying 3D U-Net convolutional neural networks for medical image segmentation. It provides c… | 42 | 2235 | active |
| microsoft/LLaVA-Med LLaVA-Med is a large language-and-vision assistant fine-tuned for the biomedicine domain, built on the LLaVA multimodal architecture. It su… | 38 | 2232 | active |
| PyGAD PyGAD is an open-source Python 3 library for implementing the genetic algorithm to optimize single- and multi-objective problems. It can al… | 91 | 2230 | active |
| NVIDIA/vid2vid A PyTorch implementation of NVIDIA's video-to-video synthesis method for generating high-resolution (e.g., 2048x1024) photorealistic videos… | 32 | 8692 | maintenance |
| facebookresearch/DiT Official PyTorch implementation of Diffusion Transformers (DiT) from the paper 'Scalable Diffusion Models with Transformers', including mod… | 10 | 8676 | maintenance |
| thuml/iTransformer Official PyTorch implementation of iTransformer, an ICLR 2024 Spotlight paper that inverts the Transformer architecture for multivariate ti… | 39 | 2225 | stable |
| apple/ml-ferret Apple's Ferret, an end-to-end multimodal large language model (MLLM) that accepts any-form referring and grounds anything in its responses,… | 70 | 8655 | maintenance |
| lifeiteng/vall-e An unofficial PyTorch implementation of VALL-E, a zero-shot text-to-speech model that treats TTS as a conditional language modeling task ov… | 37 | 2221 | active |
| google-deepmind/tapnet Google DeepMind's official repository for Tracking Any Point (TAP), containing the TAP-Vid and TAPVid-3D benchmarks, the TAPIR and TAPNext … | 75 | 2217 | active |
| RubixML/ML Rubix ML is a high-level open-source machine learning and deep learning library for PHP, offering 40+ supervised and unsupervised algorithm… | 98 | 2208 | active |
| NX-AI/xlstm Official PyTorch implementation of xLSTM, an extended Long Short-Term Memory recurrent architecture with exponential gating and matrix memo… | 76 | 2205 | active |
| lucidrains/lion-pytorch A PyTorch implementation of the Lion optimizer (Evolved Sign Momentum), discovered by Google Brain via genetic algorithms and claimed to ou… | 59 | 2205 | active |
| yyfz/Pi3 Pi3 (π³) is a feed-forward neural network for visual geometry reconstruction that eliminates the need for a fixed reference view, using a p… | 56 | 2188 | active |
| Harry24k/adversarial-attacks-pytorch Torchattacks is a PyTorch library providing implementations of adversarial attacks to generate adversarial examples against deep learning m… | 23 | 2185 | active |
| ViTAE-Transformer/ViTPose Official PyTorch implementation of ViTPose and ViTPose++, Vision Transformer models for human and generic body pose estimation from NeurIPS… | 55 | 2173 | stable |
| SpikingJelly SpikingJelly is an open-source deep learning framework for Spiking Neural Networks (SNNs) built on PyTorch. It provides a beginner-friendly… | 77 | 2159 | active |
| PKU-YuanGroup/Helios Helios is a 14B autoregressive diffusion model for real-time, minute-scale video generation supporting text-to-video, image-to-video, and v… | 56 | 2158 | active |
| facebookresearch/mae A PyTorch/GPU re-implementation of the Masked Autoencoders (MAE) paper for self-supervised vision learning. It includes pre-training code, … | 10 | 8362 | maintenance |
| jd-opensource/JoyAI-Image JoyAI-Image is a unified multimodal foundation model for image understanding, text-to-image generation, and instruction-guided image editin… | 55 | 2156 | active |
| google/trax Trax is an end-to-end deep learning library built on JAX and TensorFlow that focuses on clear code and speed, developed and maintained by t… | 10 | 8306 | maintenance |
| PKU-YuanGroup/LLaVA-CoT LLaVA-CoT is an 11B vision language model with training, inference, and dataset-generation code for spontaneous step-by-step multimodal rea… | 44 | 2131 | active |
| utkuozbulak/pytorch-cnn-visualizations A PyTorch library implementing a wide range of convolutional neural network visualization and interpretability techniques, including Grad-C… | 32 | 8236 | maintenance |
| lukemelas/EfficientNet-PyTorch A PyTorch implementation of the EfficientNet convolutional neural network family with pretrained ImageNet weights. It provides a simple pip… | 23 | 8219 | maintenance |
| autonomousvision/sdfstudio SDFStudio is a unified and modular framework for neural implicit surface reconstruction built on top of nerfstudio. It provides unified imp… | 32 | 2124 | active |
| 3DTopia/LGM LGM is the official PyTorch implementation of an ECCV 2024 Oral paper that generates high-resolution 3D models from text prompts or single-… | 26 | 2118 | active |
| eloialonso/diamond DIAMOND is a Python implementation of a reinforcement learning agent trained entirely inside a diffusion-based world model, presented as a … | 25 | 2118 | active |
| GAIR-NLP/daVinci-MagiHuman daVinci-MagiHuman is an open-source 15B-parameter single-stream transformer foundation model that jointly generates synchronized audio and … | 46 | 2114 | active |
| facebookresearch/ai4animationpy AI4AnimationPy is a Python framework for AI-driven character animation using neural networks, providing motion capture processing, training… | 56 | 2111 | active |
| Avalanche Avalanche is an end-to-end continual learning library built on PyTorch, developed by ContinualAI. It provides modules for benchmarks, train… | 24 | 2104 | active |
| kohya-ss/musubi-tuner Musubi Tuner is a set of Python scripts for training LoRA (Low-Rank Adaptation) adapters for video and image generation model architectures… | 88 | 2103 | active |
| River-Zhang/ICEdit ICEdit (In-Context Edit) is a research framework for instruction-based image editing built on large-scale Diffusion Transformers, using a L… | 42 | 2102 | active |
| SUDO-AI-3D/zero123plus Zero123++ is a diffusion base model that generates consistent multi-view images from a single input image, intended as a stepping stone for… | 28 | 2099 | active |
| HRM-Text HRM-Text is a 1B-parameter text generation model based on the hierarchical recurrent HRM architecture, released with a complete pretraining… | 58 | 2093 | active |
| alex-damian/pulse PULSE is a Python research implementation of a CVPR 2020 paper that upscales low-resolution face photos by searching the latent space of a … | 32 | 8026 | maintenance |
| jeshraghian/snntorch snnTorch is a Python library for gradient-based deep learning with spiking neural networks, built as an extension of PyTorch. It provides s… | 68 | 2075 | active |
| WenjieDu/PyPOTS PyPOTS is a Python toolbox for machine learning and data mining on partially-observed time series with missing values. It integrates 50+ st… | 89 | 2072 | active |
| bytedance/Protenix Protenix is an open-source PyTorch reproduction of AlphaFold 3 for high-accuracy biomolecular structure prediction, covering proteins, nucl… | 78 | 2070 | active |
| facebookresearch/ConvNeXt-V2 Official PyTorch implementation of ConvNeXt V2, a family of pure convolutional neural network models co-designed with a fully convolutional… | 10 | 2065 | stable |
| brightmart/text_classification A collection of deep learning baseline models for text classification in NLP, implemented in TensorFlow. It covers classic architectures li… | 32 | 7938 | maintenance |
| Plachtaa/VALL-E-X An open-source Python implementation of Microsoft's VALL-E X zero-shot text-to-speech model, with a community-trained pretrained checkpoint… | 10 | 7926 | maintenance |
| facebookresearch/theseus Theseus is a PyTorch-based library for building custom differentiable nonlinear optimization layers, supporting problems in robotics and vi… | 23 | 2060 | active |
| zju3dv/GVHMR GVHMR is a research codebase implementing the SIGGRAPH Asia 2024 paper 'World-Grounded Human Motion Recovery via Gravity-View Coordinates'.… | 66 | 2059 | active |
| shanglianlm0525/PyTorch-Networks A collection of PyTorch implementations of classic and modern CNN architectures, covering classification, detection, segmentation, face, an… | 49 | 2056 | active |
| DeePMD-kit DeePMD-kit is a deep learning package for building many-body potential energy representations and running molecular dynamics simulations. I… | 92 | 2055 | stable |
| jaywalnut310/vits VITS is the official PyTorch implementation of an end-to-end text-to-speech model based on a conditional variational autoencoder with adver… | 32 | 7894 | maintenance |
| pykeen/pykeen PyKEEN is a Python library for training and evaluating multimodal knowledge graph embedding models built on PyTorch. It provides a high-lev… | 69 | 2039 | active |
| SakanaAI/continuous-thought-machines The Continuous Thought Machine (CTM) is a neural network architecture from Sakana AI that uses neuron-level temporal dynamics and neural sy… | 43 | 2036 | active |
| tensorflow/recommenders TensorFlow Recommenders is a Python library for building recommender system models on top of TensorFlow and Keras. It covers the full workf… | 85 | 2034 | active |
| deep-floyd/IF DeepFloyd IF is an open-source text-to-image model library implementing a cascaded pixel diffusion architecture with a frozen T5 text encod… | 22 | 7793 | maintenance |