cuDNN: Efficient Primitives for Deep Learning
Not on the map. Why?
Abstract (arXiv)
We present a library of efficient implementations of deep learning primitives. Deep learning workloads are computationally intensive, and optimizing their kernels is difficult and time-consuming. As parallel architectures evolve, kernels must be reoptimized, which makes maintaining codebases difficult over time. Similar issues have long been addressed in the HPC community by libraries such as the Basic Linear Algebra Subroutines (BLAS). However, there is no analogous library for deep learning. Without such a library, researchers implementing deep learning workloads on parallel processors must create and optimize their own implementations of the main computational kernels, and this work must be repeated as new parallel processors emerge. To address this problem, we have created a library similar in intent to BLAS, with optimized routines for deep learning workloads. Our implementation contains routines for GPUs, although similarly to the BLAS library, these routines could be implemented for other platforms. The library is easy to integrate into existing frameworks, and provides optimized performance and memory usage. For example, integrating cuDNN into Caffe, a popular framework for convolutional networks, improves performance by 36% on a standard model while also reducing memory consumption.
Key facts. Cited by 240 papers in the map (data through 7 Oct 2026). First posted on arXiv 3 Oct 2014.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 64; 2022: 47; 2023: 39; 2024: 32; 2025: 26; 2026: 32.
240 papers.
Sort:
By area:
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 520
- 2021Scaling Local Self-Attention for Parameter Efficient Visual Backbonescited by 220
- 2024Vidur: A Large-Scale Simulation Framework For LLM Inferencecited by 115
- 2021Proof-of-Learning: Definitions and Practicecited by 81
- 2021Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reportscited by 80
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2021Stochastic Training is Not Necessary for Generalizationcited by 63
- 2021Randomness In Neural Network Training: Characterizing The Impact of Toolingcited by 57
- 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimizationcited by 52
- 2021Dorylus: Affordable, Scalable, and Accurate GNN Training with Distributed CPU Servers and Serverless Threadscited by 51
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNscited by 49
- 2024Efficient Deformable ConvNets: Rethinking Dynamic and Sparse Operator for Vision Applicationscited by 49
38 more
- 2021Nondeterminism and Instability in Neural Network Optimizationcited by 47
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 46
- 2021LiBRe: A Practical Bayesian Approach to Adversarial Detectioncited by 40
- 2022Structural Pruning via Latency-Saliency Knapsackcited by 40
- 2024Mirage: A Multi-Level Superoptimizer for Tensor Programscited by 39
- 2022TensorIR: An Abstraction for Automatic Tensorized Program Optimizationcited by 38
- 2024Lightweight Deep Learning for Resource-Constrained Environments: A Surveycited by 38
- 2021Continuum: Simple Management of Complex Continual Learning Scenarioscited by 33
- 2022Enable Deep Learning on Mobile Devices: Methods, Systems, and Applicationscited by 32
- 2022Tensor Program Optimization with Probabilistic Programscited by 30
- 2021A Runtime-Based Computational Performance Predictor for Deep Neural Network Trainingcited by 29
- 2022Hidet: Task-Mapping Programming Paradigm for Deep Learning Tensor Programscited by 28
- 2022BiViT: Extremely Compressed Binary Vision Transformercited by 28
- 2021Dual-side Sparse Tensor Corecited by 27
- 2022Coarsening the Granularity: Towards Structurally Sparse Lottery Ticketscited by 27
- 2022Flashlight: Enabling Innovation in Tools for Machine Learningcited by 25
- 2022Towards Training Reproducible Deep Learning Modelscited by 23
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUscited by 20
- 2021"Adversarial Examples" for Proof-of-Learningcited by 19
- 2021RED++ : Data-Free Pruning of Deep Neural Networks via Input Splitting and Output Mergingcited by 19
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Searchcited by 19
- 2021Anytime Dense Prediction with Confidence Adaptivitycited by 18
- 2021Bolt: Bridging the Gap between Auto-tuners and Hardware-native Performancecited by 18
- 2021TSM: Temporal Shift Module for Efficient and Scalable Video Understanding on Edge Devicecited by 17
- 2021The CoRa Tensor Compiler: Compilation for Ragged Tensors with Minimal Paddingcited by 17
- 2021Post-Training Sparsity-Aware Quantizationcited by 16
- 2021HELP: Hardware-Adaptive Efficient Latency Prediction for NAS via Meta-Learningcited by 16
- 2021Clairvoyant Prefetching for Distributed Machine Learning I/Ocited by 14
- 2021SparseDNN: Fast Sparse Deep Learning Inference on CPUscited by 13
- 2021Enabling Design Methodologies and Future Trends for Edge AI: Specialization and Co-designcited by 13
- 2021A compact butterfly-style silicon photonic-electronic neural chip for hardware-efficient deep learningcited by 13
- 2022Teal: Learning-Accelerated Optimization of WAN Traffic Engineeringcited by 13
- 2023Efficient On-device Training via Gradient Filteringcited by 13
- 2021Dynamic Precision Analog Computing for Neural Networkscited by 12
- 2021Real-Time Glaucoma Detection from Digital Fundus Images using Self-ONNscited by 12
- 2022Temporal Label Smoothing for Early Event Predictioncited by 12
- 2022Soft Masking for Cost-Constrained Channel Pruningcited by 12
- 2023Relax: Composable Abstractions for End-to-End Dynamic Machine Learningcited by 12
50 most cited of 240 All 240 (JSON)
- 2026Long-WAM: Scaling the Context of World-Action Models7 Oct 2026
- 2026Beyond LLM Serving: Characterizing Vision-Language-Action Workloads for Embodied AI System Design4 Oct 2026
- 2026RLX: A Unified Multi-Backend Tensor Compiler and Distributed Runtime in Rust29 Sep 2026
- 2026Benchmarking Attention for Tabular Foundation Models25 Sep 2026
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization24 Sep 2026
- 2026KerColle: Unlocking Fine-Grained GPU Concurrency in Vision-Language-Action Models16 Sep 2026
- 2026Tensor Seeks Layout: Formalizing Layout Selection for ML Compilers21 Aug 2026
- 2026HIERA: Workload-Aware Planning Across Implementation Spaces for GPU Kernel Optimization21 Aug 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026Flash EQ-Linear: Accelerating Equivariant Linear Layers via Group-wise Discrete Fourier Transform23 Jul 2026
- 2026Lazy Arithmetic using Systolic Arrays for Closing the Verification Gap on Embedded Systems16 Jul 2026
- 2026Nova: An End-to-End MLIR Compiler for Deep Learning15 Jul 2026
38 more
- 2026GPU Parallelization Strategies for Forward and Backward Propagation in Shallow Neural Networks: A CUDA-Based Comparative Study29 Jun 2026
- 2026Learning 1-Bit LiDAR-based Localization with Auxiliary Objective26 Jun 2026
- 2026Toward Compiler World Models: Learning Latent Dynamics for Efficient Tensor Program Search8 Jun 2026
- 2026Evaluating CUDA Tile for AI Workloads on Hopper and Blackwell GPUs25 Apr 2026
- 2026Prism: Symbolic Superoptimization of Tensor Programs16 Apr 2026
- 2026TCL: Enabling Fast and Efficient Cross-Hardware Tensor Program Optimization via Continual Learning14 Apr 2026
- 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe1 Apr 2026
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Search25 Mar 2026
- 2026KernelFoundry: Hardware-aware evolutionary GPU kernel optimization12 Mar 2026
- 2026Beyond Convolution: A Taxonomy of Structured Operators for Learning-Based Image Processing12 Mar 2026
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization10 Mar 2026
- 2026Making LLMs Optimize Multi-Scenario CUDA Kernels Like Experts7 Mar 2026
- 2026PolyBlocks: A Compiler Infrastructure for AI Chips and Programming Frameworks6 Mar 2026
- 2026CUCo: An Agentic Framework for Compute and Communication Co-design2 Mar 2026
- 2026GPU-Fuzz: Finding Memory Errors in Deep Learning Frameworks11 Feb 2026
- 2026Learning to Optimize by Differentiable Programming23 Jan 2026
- 2026Cross-Domain Object Detection Using Unsupervised Image Translation16 Jan 2026
- 2026Monte-Carlo Tree Search with Neural Network Guidance for Lane-Free Autonomous Driving14 Jan 2026
- 2026Hardware-Aware Reformulation of Convolutions for Efficient Execution on Specialized AI Hardware: A Case Study on NVIDIA Tensor Cores9 Jan 2026
- 2026Few-Shot Video Object Segmentation in X-Ray Angiography Using Local Matching and Spatio-Temporal Consistency Loss2 Jan 2026
- 2025GPU Kernel Optimization Beyond Full Builds: An LLM Framework with Minimal Executable Programs15 Dec 2025
- 2025Accelerated Rotation-Invariant Convolution for UAV Image Segmentation9 Dec 2025
- 2025SSCATeR: Sparse Scatter-Based Convolution Algorithm with Temporal Data Recycling for Real-Time 3D Object Detection in LiDAR Point Clouds9 Dec 2025
- 2025PointCNN++: Performant Convolution on Native Points28 Nov 2025
- 2025Low-Rank GEMM: Efficient Matrix Multiplication via Low-Rank Approximation with FP8 Acceleration24 Nov 2025
- 2025Optimizing PyTorch Inference with LLM-Based Multi-Agent Systems21 Nov 2025
- 2025GraphNet: A Large-Scale Computational Graph Dataset for Tensor Compiler Research28 Oct 2025
- 2025Hyperparameter Optimization and Reproducibility in Deep Learning Model Training16 Oct 2025
- 2025Hybrid Dual-Batch and Cyclic Progressive Learning for Efficient Distributed Training30 Sep 2025
- 2025Towards Robust Agentic CUDA Kernel Benchmarking, Verification, and Optimization16 Sep 2025
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization9 Sep 2025
- 2025Efficient identification of critical regions via Flow Matching-based Monte Carlo initialization21 Aug 2025
- 2025Improving Neural Network Training using Dynamic Learning Rate Schedule for PINNs and Image Classification29 Jul 2025
- 2025DMQ: Dissecting Outliers of Diffusion Models for Post-Training Quantization17 Jul 2025
- 2025AXLearn: Modular, Hardware-Agnostic Large Model Training7 Jul 2025
- 2025Ark: An Open-source Python-based Framework for Robot Learning24 Jun 2025
- 2025QiMeng-Attention: SOTA Attention Operator is generated by SOTA Attention Algorithm14 Jun 2025
- 2025Semi-Tensor-Product Based Convolutional Neural Networks12 Jun 2025
50 newest of 240 All 240 (JSON)
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimizationcited by 46
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Searchcited by 19
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimizationcited by 9
- 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTecited by 6
- 2025Optimizing PyTorch Inference with LLM-Based Multi-Agent Systemscited by 4
- 2026CUCo: An Agentic Framework for Compute and Communication Co-designcited by 1
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimizationcited by 0
- 2026KernelOPT: Dispatch-Aware Agentic Search for GPU Kernel Optimization24 Sep 2026
- 2026CuTeGen: An LLM-Based Agentic Framework for Generation and Optimization of High-Performance GPU Kernels using CuTe1 Apr 2026
- 2026AVO: Agentic Variation Operators for Autonomous Evolutionary Search25 Mar 2026
- 2026KernelSkill: A Multi-Agent Framework for GPU Kernel Optimization10 Mar 2026
- 2026CUCo: An Agentic Framework for Compute and Communication Co-design2 Mar 2026
- 2025Optimizing PyTorch Inference with LLM-Based Multi-Agent Systems21 Nov 2025
- 2025Astra: A Multi-Agent System for GPU Kernel Performance Optimization9 Sep 2025
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networkscited by 9
- 2025DMQ: Dissecting Outliers of Diffusion Models for Post-Training Quantizationcited by 6
- 2022ParticleGrid: Enabling Deep Learning using 3D Representation of Materialscited by 2
- 2025Efficient identification of critical regions via Flow Matching-based Monte Carlo initializationcited by 2
- 2025Efficient identification of critical regions via Flow Matching-based Monte Carlo initialization21 Aug 2025
- 2025DMQ: Dissecting Outliers of Diffusion Models for Post-Training Quantization17 Jul 2025
- 2023DeepPCR: Parallelizing Sequential Operations in Neural Networks28 Sep 2023
- 2022ParticleGrid: Enabling Deep Learning using 3D Representation of Materials15 Nov 2022
- 2022Tuning of Mixture-of-Experts Mixed-Precision Neural Networks29 Sep 2022
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 520
- 2021Scaling Local Self-Attention for Parameter Efficient Visual Backbonescited by 220
- 2024Vidur: A Large-Scale Simulation Framework For LLM Inferencecited by 115
- 2021Proof-of-Learning: Definitions and Practicecited by 81
- 2021Generalized Radiograph Representation Learning via Cross-supervision between Images and Free-text Radiology Reportscited by 80
- 2023Full Stack Optimization of Transformer Inference: a Surveycited by 79
- 2021Stochastic Training is Not Necessary for Generalizationcited by 63
- 2021Randomness In Neural Network Training: Characterizing The Impact of Toolingcited by 57
- 2021Dorylus: Affordable, Scalable, and Accurate GNN Training with Distributed CPU Servers and Serverless Threadscited by 51
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNscited by 49
- 2024Efficient Deformable ConvNets: Rethinking Dynamic and Sparse Operator for Vision Applicationscited by 49
- 2021Nondeterminism and Instability in Neural Network Optimizationcited by 47
12 most cited of 219 All 219 (JSON)
- 2026RLX: A Unified Multi-Backend Tensor Compiler and Distributed Runtime in Rust29 Sep 2026
- 2026Benchmarking Attention for Tabular Foundation Models25 Sep 2026
- 2026Tensor Seeks Layout: Formalizing Layout Selection for ML Compilers21 Aug 2026
- 2026HIERA: Workload-Aware Planning Across Implementation Spaces for GPU Kernel Optimization21 Aug 2026
- 2026PTXBench: Benchmarking and Adapting LLMs for GPU Kernel Optimization with Architecture-specific PTX18 Aug 2026
- 2026Flash EQ-Linear: Accelerating Equivariant Linear Layers via Group-wise Discrete Fourier Transform23 Jul 2026
- 2026Lazy Arithmetic using Systolic Arrays for Closing the Verification Gap on Embedded Systems16 Jul 2026
- 2026Nova: An End-to-End MLIR Compiler for Deep Learning15 Jul 2026
- 2026GPU Parallelization Strategies for Forward and Backward Propagation in Shallow Neural Networks: A CUDA-Based Comparative Study29 Jun 2026
- 2026Learning 1-Bit LiDAR-based Localization with Auxiliary Objective26 Jun 2026
- 2026Evaluating CUDA Tile for AI Workloads on Hopper and Blackwell GPUs25 Apr 2026
- 2026Prism: Symbolic Superoptimization of Tensor Programs16 Apr 2026
12 newest of 219 All 219 (JSON)