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
junyanz/CycleGAN A Torch (Lua) implementation of CycleGAN and pix2pix for unpaired image-to-image translation using cycle-consistent adversarial networks. I…	32	12874	maintenance
AutoArk/GPA GPA (General Purpose Audio) is a unified autoregressive audio-language model that performs text-to-speech, automatic speech recognition, an…	51	3109	active
jina-ai/clip-as-service CLIP-as-service is a low-latency, high-scalability server for embedding images and text into fixed-length vectors using OpenAI's CLIP model…	23	12831	maintenance
TensorFlow Lite for Microcontrollers TensorFlow Lite for Microcontrollers (TFLM) is a C++ port of TensorFlow Lite for running ML models on microcontrollers, DSPs, and other mem…	76	3098	active
ridgerchu/matmulfreellm A Python implementation of MatMul-Free LM, a language model architecture that eliminates matrix multiplication operations using ternary wei…	76	3093	active
guillaume-be/rust-bert A Rust-native library providing ready-to-use NLP pipelines and transformer-based models (BERT, DistilBERT, GPT-2, RoBERTa, BART, etc.), por…	57	3084	active
sonos/tract Tract is Sonos' tiny, self-contained neural-network inference engine written in Rust. It loads ONNX, TensorFlow/TFLite, and NNEF models, op…	97	3075	active
Neural Amp Modeler Neural Amp Modeler (NAM) is a Python project that trains neural network models to emulate guitar amplifiers and exports them as .nam files.…	80	3060	active
RFDiffusion RFdiffusion is an open-source method for de novo protein structure generation using diffusion models, with or without conditional informati…	59	3058	active
hegelai/prompttools PromptTools is an open-source Python library for testing, experimenting with, and evaluating prompts, LLMs (OpenAI, Anthropic, LLaMA, etc.)…	46	3056	active
tensorflow/agents TF-Agents is a TensorFlow-based Python library for building, training, and deploying reinforcement learning and contextual bandit algorithm…	57	3028	active
luminal-ai/luminal Luminal is a high-performance general-purpose ML inference compiler written in Rust that lowers models to a minimal 15-op dataflow IR and c…	77	3026	active
deepseek-ai/DualPipe DualPipe is a Python library implementing a bidirectional pipeline parallelism algorithm that overlaps forward and backward computation wit…	45	3017	active
ZQPei/deep_sort_pytorch A PyTorch implementation of the Deep SORT multi-object tracking algorithm, pairing YOLOv3/YOLOv5 (or Mask R-CNN) detectors with a CNN re-id…	32	3013	active
osmr/imgclsmob A research sandbox providing (re)implementations of numerous deep learning computer vision models for classification, segmentation, detecti…	23	3012	active
pytorch/TensorRT Torch-TensorRT is a compiler library that accelerates PyTorch model inference on NVIDIA GPUs using TensorRT. It supports just-in-time compi…	93	3009	active
thu-pacman/chitu Chitu (赤兔) is a high-performance inference framework for large language models, positioned as a production-grade LLM serving engine. It sup…	82	3009	active
pytorch/ao TorchAO is a PyTorch-native library for model optimization through quantization and sparsity. It supports quantizing weights, gradients, op…	87	3001	active
MeiGen-AI/MultiTalk MultiTalk is an audio-driven framework for generating multi-person conversational videos from multi-stream audio, a reference image, and a …	53	3000	active
benedekrozemberczki/pytorch_geometric_temporal PyTorch Geometric Temporal is a temporal (dynamic) extension library for PyTorch Geometric providing spatiotemporal signal processing with …	64	2992	active
Biohub/esm A Python library and model release providing ESMC protein language models, ESMFold2 structure prediction, and the ESM Atlas of 6.8 billion …	88	2987	active
sovrasov/flops-counter.pytorch A Python library (ptflops) that computes the theoretical number of multiply-add operations (FLOPs) and parameter counts for neural network …	36	2983	active
bethgelab/foolbox Foolbox is a Python library for generating adversarial examples that fool deep neural networks, with state-of-the-art gradient-based and de…	44	2979	active
patrick-kidger/equinox Equinox is a Python library providing neural networks and scientific computing utilities for JAX, using PyTorch-like class-based syntax whe…	88	2976	stable
sunsmarterjie/yolov12 YOLOv12 is a PyTorch implementation of attention-centric real-time object detectors, published at NeurIPS 2025. It provides detection model…	66	2961	active
karpathy/char-rnn char-rnn is a Torch/Lua implementation of multi-layer recurrent neural networks (RNN, LSTM, GRU) for character-level language modeling. It …	32	12120	maintenance
iscyy/ultralyticsPro A PyTorch-based collection of improved YOLO-family object detection models (YOLOv5 through YOLOv13, RT-DETR) with pluggable modules for bac…	45	2954	active
TylerYep/torchinfo torchinfo is a Python library that provides Keras-style model summaries for PyTorch, showing layer types, input/output shapes, parameter co…	66	2953	active
bghira/SimpleTuner SimpleTuner is a Python fine-tuning toolkit for image, video, and audio diffusion models built on Hugging Face Diffusers. It provides a web…	93	2940	active
state-spaces/s4 Official implementations of Structured State Space (S4) sequence models and related variants like S4D, HiPPO, and SaShiMi, built in PyTorch…	32	2927	active
keras-team/keras-tuner KerasTuner is a hyperparameter optimization framework for Keras models with a define-by-run syntax for configuring search spaces. It ships …	47	2923	active
NVlabs/FoundationStereo FoundationStereo is NVIDIA's official PyTorch implementation of a foundation model for zero-shot stereo depth estimation, published as a CV…	44	2923	active
decoderesearch/circuit-tracer A Python library for circuit tracing and attribution graphs in language model interpretability, based on cross-layer MLP transcoders. It co…	78	2918	active
elixir-nx/nx Nx is an Elixir library providing multi-dimensional arrays (tensors) and numerical definitions with pluggable backends. The monorepo also i…	98	2915	active
ModelTC/LightX2V LightX2V is a lightweight, high-performance inference framework for image and video generation, supporting tasks like text-to-video, image-…	84	2914	active
AllenNLP AllenNLP is an open-source NLP research library built on PyTorch for developing state-of-the-art deep learning models on linguistic tasks. …	10	11886	maintenance
Saiyan-World/goku Goku is a family of flow-based (rectified flow Transformer) foundation models for joint image and video generation, released by HKU and Byt…	21	2906	active
learnables/learn2learn learn2learn is a PyTorch library for meta-learning research, providing utilities for few-shot task creation, high-level wrappers for algori…	45	2894	active
explosion/thinc Thinc is a lightweight deep learning library from the makers of spaCy offering a type-checked, functional-programming API for composing neu…	74	2891	active
ideogram-oss/ideogram4 Ideogram 4 is an open-weight text-to-image foundation model trained from scratch, with inference code and weights released in Python. It fe…	51	2888	active
KellerJordan/Muon Muon is a PyTorch optimizer for the hidden layers of neural networks, based on orthogonalized momentum updates via Newton-Schulz iteration.…	56	2887	active
werner-duvaud/muzero-general A commented and documented PyTorch implementation of DeepMind's MuZero reinforcement learning algorithm, designed to be easily adapted to a…	32	2870	active
OpenGVLab/InternImage InternImage is a large-scale CNN-based vision foundation model that uses deformable convolutions as its core operator, released with pretra…	25	2854	stable
huggingface/nanotron Nanotron is a minimalistic Python library from Hugging Face for pretraining large language models with 3D parallelism (data, tensor, and pi…	63	2843	active
TMElyralab/MuseV MuseV is a diffusion-based framework for generating high-fidelity virtual human videos of infinite length using a Visual Conditioned Parall…	26	2843	active
huggingface/setfit SetFit is a Python library for efficient, prompt-free few-shot fine-tuning of Sentence Transformers for text classification. It achieves hi…	88	2842	active
AdapterHub Adapters is a Python add-on library for HuggingFace Transformers that integrates 10+ parameter-efficient fine-tuning methods (bottleneck ad…	74	2831	active
pytorch/xla PyTorch/XLA is a Python package that connects the PyTorch deep learning framework to XLA devices such as Google Cloud TPUs via the XLA deep…	63	2801	active
Mega4alik/ollm oLLM is a lightweight Python library for large-context LLM inference built on Hugging Face Transformers and PyTorch. It offloads weights an…	57	2798	active
lucidrains/DALLE2-pytorch A PyTorch implementation of OpenAI's DALL-E 2 text-to-image synthesis model, focusing on the diffusion prior network that predicts image em…	23	11302	maintenance
FasterDecoding/Medusa Medusa is a framework that accelerates LLM text generation by adding multiple decoding heads to an existing model, avoiding the need for a …	19	2776	active
NVlabs/stylegan2 The official TensorFlow implementation of StyleGAN2, NVIDIA's improved style-based generative adversarial network for high-quality uncondit…	32	11180	maintenance
NVIDIA/FastPhotoStyle FastPhotoStyle is NVIDIA's official PyTorch implementation of the ECCV 2018 paper 'A Closed-form Solution to Photorealistic Image Stylizati…	23	11171	maintenance
apple/turicreate Turi Create is a Python library from Apple that simplifies building custom machine learning models for tasks like image classification, obj…	10	11155	maintenance
voxelmorph/voxelmorph VoxelMorph is a Python library for learning-based image registration and alignment, using unsupervised deep learning to model deformations …	75	2754	active
xdit-project/xDiT xDiT is a scalable inference engine for Diffusion Transformers (DiTs) that enables parallel deployment across multiple GPUs and machines. I…	91	2732	active
artidoro/qlora QLoRA is the official implementation of the QLoRA paper, an efficient finetuning approach that backpropagates through a frozen 4-bit quanti…	30	11035	maintenance
qualcomm/aimet AIMET (AI Model Efficiency Toolkit) is a Python library from Qualcomm providing advanced quantization and compression techniques for traine…	99	2723	active
intel/neural-compressor Intel Neural Compressor is an open-source Python library providing state-of-the-art low-bit quantization (INT8/FP8/MXFP8/INT4/MXFP4/NVFP4),…	95	2722	active
black-forest-labs/flux2 Official inference repository for Black Forest Labs' FLUX.2 family of open-weight image generation and editing models. It provides minimal …	45	2720	active
prophesier/diff-svc Diff-SVC is a deep learning project that performs singing voice conversion using diffusion models, transforming input singing audio into a …	58	2719	active
yuqinie98/PatchTST Official PyTorch implementation of PatchTST, an ICLR 2023 Transformer model for long-term time series forecasting based on patching and cha…	32	2719	stable
lengstrom/fast-style-transfer A TensorFlow implementation of fast neural style transfer that applies the style of famous paintings to photos and videos in real time. It …	32	10957	maintenance
bmild/nerf The official TensorFlow implementation of NeRF (Neural Radiance Fields), the ECCV 2020 paper representing scenes as neural radiance fields …	36	10944	maintenance
yuweihao/MambaOut MambaOut is a PyTorch implementation of Gated CNN models from the CVPR 2025 paper 'MambaOut: Do We Really Need Mamba for Vision?', which qu…	16	2710	stable
TMElyralab/MusePose MusePose is a diffusion-based, pose-guided image-to-video generation framework for creating virtual human videos, where a character in a re…	25	2706	active
dreamzero0/dreamzero DreamZero is NVIDIA's World Action Model (WAM) that jointly predicts future video and actions from a pretrained video diffusion backbone, e…	48	2700	active
ZHZisZZ/dllm dLLM is a Python library that unifies training, inference, and evaluation of diffusion language models such as LLaDA and Dream. It builds o…	56	2699	active
HiLab-git/SSL4MIS A benchmark and code collection of semi-supervised learning methods for medical image segmentation, re-implementing approaches like Mean Te…	40	2699	active
huggingface/text-generation-inference Text Generation Inference (TGI) is a Rust, Python and gRPC toolkit for deploying and serving large language models with high performance, p…	10	10877	maintenance
roboflow/maestro maestro is a Python library from Roboflow that streamlines fine-tuning of multimodal vision-language models such as Florence-2, PaliGemma 2…	61	2696	active
baaivision/EVA EVA is a family of large-scale vision foundation models from BAAI, including masked image models (EVA-01/02) and scaled CLIP models (EVA-CL…	23	2694	active
SkyReels SkyReels V1 is an open-source human-centric video foundation model with Text-to-Video and Image-to-Video variants, fine-tuned from HunyuanV…	22	2694	active
openai/DALL-E The official PyTorch package for the discrete VAE (dVAE) component of OpenAI's DALL·E model. It does not include the transformer that gener…	10	10820	maintenance
MrGiovanni/UNetPlusPlus Official implementation of UNet++, a nested U-Net architecture for medical image segmentation, in both Keras and PyTorch. It redesigns skip…	73	2685	stable
InfiniteYou InfiniteYou (InfU) is a research framework from ByteDance for identity-preserved text-to-image generation built on Diffusion Transformers l…	34	2685	active
Tencent-Hunyuan/HY-World-2.0 HY-World 2.0 is Tencent Hunyuan's open-source multi-modal world model framework that reconstructs, generates, and simulates 3D worlds from …	55	2680	active
google-deepmind/mctx Mctx is a JAX-native Python library implementing Monte Carlo tree search algorithms such as AlphaZero, MuZero, and Gumbel MuZero. It suppor…	81	2678	active
stochasticai/xTuring xTuring is a Python library for fine-tuning, evaluating, and running open-source large language models such as LLaMA, GPT-J, GPT-2, Qwen, a…	65	2675	active
StableSR StableSR is a Python research library that leverages pre-trained Stable Diffusion priors for real-world blind image super-resolution. It pr…	21	2672	stable
Tencent/MimicMotion MimicMotion is a diffusion-based framework from Tencent for generating high-quality human motion videos guided by pose sequences, featuring…	44	2668	active
open-gigaai/giga-brain-0 GigaBrain-0/0.7 is an open-source vision-language-action (VLA) model family for generalist embodied agents, powered by world models and a t…	60	2659	active
phillipi/pix2pix The original Torch (Lua) implementation of pix2pix, a conditional GAN for image-to-image translation tasks such as synthesizing photos from…	32	10659	maintenance
FeatherTalk An ultralight talking-head (digital human) model that animates a person's face from audio input and runs in real time on mobile devices. It…	61	2648	active
swz30/Restormer Restormer is an efficient Transformer architecture for high-resolution image restoration, published as a CVPR 2022 Oral paper. It provides …	41	2648	stable
ultralytics/yolov3 Ultralytics' PyTorch implementation of YOLOv3, YOLOv3-SPP, and YOLOv3-tiny for real-time object detection. It provides training, validation…	66	10613	maintenance
NVlabs/LongLive LongLive is an NVIDIA research framework providing parallel training and inference infrastructure for real-time long video generation, usin…	60	2639	active
lucidrains/audiolm-pytorch A PyTorch implementation of AudioLM, Google Research's language modeling approach to audio generation, including a MIT-licensed SoundStream…	34	2625	active
meta-pytorch/torchrec TorchRec is a PyTorch domain library for building recommendation systems at scale. It provides distributed sharding of large embedding tabl…	95	2615	active
crowsonkb/k-diffusion A PyTorch library implementing Karras et al. (2022) diffusion models with enhancements like improved sampling algorithms and transformer-ba…	49	2599	active
atong01/conditional-flow-matching TorchCFM is a PyTorch library implementing Conditional Flow Matching (CFM), a simulation-free training objective for continuous normalizing…	72	2596	active
asteroid-team/asteroid Asteroid is a PyTorch-based audio source separation toolkit for researchers, providing modular building blocks (filterbanks, encoders, mask…	66	2593	active
median-research-group/LibMTL LibMTL is an open-source PyTorch library for Multi-Task Learning (MTL). It provides implementations of many MTL architectures and gradient-…	38	2593	active
facebookresearch/demucs Demucs is a state-of-the-art music source separation model from Meta AI that splits songs into stems like drums, bass, and vocals using a h…	10	10373	maintenance
kijai/ComfyUI-HunyuanVideoWrapper A set of custom ComfyUI nodes that wrap Tencent's HunyuanVideo text-to-video and image-to-video diffusion model for use inside ComfyUI work…	35	2591	active
Memento-Teams/Memento Memento is a Python framework for building LLM agents that continually improve from experience via memory-based case-based reasoning, witho…	36	2581	active
vita-epfl/Stable-Video-Infinity Stable Video Infinity (SVI) is a research codebase for infinite-length video generation using video diffusion transformers with an error-re…	60	2579	active
KomputeProject/kompute Kompute is a general-purpose GPU compute framework built on Vulkan that works across vendor GPUs (AMD, NVIDIA, Qualcomm, etc.) with both C+…	63	2577	active
Tencent-Hunyuan/HY-Motion-1.0 HY-Motion 1.0 is a series of text-to-3D human motion generation models from Tencent Hunyuan, based on Diffusion Transformer and Flow Matchi…	55	2574	active
advimman/lama LaMa is a PyTorch-based image inpainting model that fills large missing regions in images using fast Fourier convolutions, generalizing wel…	32	10291	maintenance