cuDNN for Deep Learning
Convolutions and tensor ops at scale.
The Engine Behind AI Frameworks
cuDNN is NVIDIA's deep learning library. PyTorch and TensorFlow lean on it for fast convolutions and other neural net building blocks. 🧠
Why Convolutions Need Help
A convolution slides a small filter over an image, doing huge amounts of math. cuDNN implements it far faster than a naive kernel.
All lessons in this course
- cuBLAS GEMM Done Right
- Thrust Vectors and Transforms
- Thrust Reduce, Scan, and Sort
- cuDNN for Deep Learning