The TWIML AI Podcast (formerly This Week in Machine Learning & Artificial Intelligence)

Parallelism and Acceleration for Large Language Models with Bryan Catanzaro - #507

08.05.2021 - By Sam CharringtonPlay

Download our free app to listen on your phone

Download on the App StoreGet it on Google Play

Today we’re joined by Bryan Catanzaro, vice president of applied deep learning research at NVIDIA.

Most folks know Bryan as one of the founders/creators of cuDNN, the accelerated library for deep neural networks. In our conversation, we explore his interest in high-performance computing and its recent overlap with AI, his current work on Megatron, a framework for training giant language models, and the basic approach for distributing a large language model on DGX infrastructure. 

We also discuss the three different kinds of parallelism, tensor parallelism, pipeline parallelism, and data parallelism, that Megatron provides when training models, as well as his work on the Deep Learning Super Sampling project and the role it's playing in the present and future of game development via ray tracing. 

The complete show notes for this episode can be found at twimlai.com/go/507.

More episodes from The TWIML AI Podcast (formerly This Week in Machine Learning & Artificial Intelligence)