NVIDIA’s vice president of Applied Deep Learning Research, with contributions spanning cuDNN, neural graphics and distributed language-model training.[1][3]
Start with Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelism →
Education
Ph.D., Electrical Engineering and Computer Sciences
Completed 2011
[2]Links
NVIDIA technical writingVice President, Applied Deep Learning Research · NVIDIA
Background
Catanzaro earned his Berkeley EECS doctorate in 2011, advised by Kurt Keutzer. Berkeley also records his work with Adam Coates and Andrew Ng at Baidu on deep-learning training systems and end-to-end speech recognition. The connection between these projects is practical computation: how to make learning systems train and operate at useful scale.[2]
Career
Catanzaro leads NVIDIA research applying deep learning to language understanding, graphics and chip design. NVIDIA credits his research with leading to cuDNN and credits him with helping lead the team behind DLSS 2.0. These are different forms of contribution: a library foundation for neural-network computation and team leadership on a graphics system.[1]
Contributions
Splitting a model across GPUs
The 2019 Megatron-LM paper, which Catanzaro co-authored, tackles models that are too large to fit comfortably on one graphics processor. Its method splits computation within transformer layers across multiple GPUs. The authors implemented the approach in PyTorch with a small number of communication operations, making the distribution strategy available without a new compiler. The contribution is a concrete training method, rather than a claim that larger models always solve a task better.[3]
Combining forms of parallelism
A 2021 follow-up studies how to combine tensor, pipeline and data parallelism. These divide work inside layers, across successive model stages and across training data. Combining them carelessly can leave processors waiting or communicating excessively. The paper analyzes those trade-offs and introduces an interleaved pipeline schedule, giving researchers a way to reason about efficient training across large GPU clusters. Catanzaro is one of its co-authors.[4]
Tools, writing & talks
Writing, research & talks
Start here
Start here for a concrete contribution: dividing transformer-layer computation across GPUs to train models that exceed individual device memory. Some familiarity with neural-network training helps.
Co-author[3]The follow-up explains how different ways of splitting training work interact, including communication costs and idle processors. Read after the original Megatron paper.
Co-author[4]- PyCUDA and PyOpenCLarxiv.org
An earlier paper on using scripting to generate GPU code at runtime. It connects Catanzaro’s later model-training research with his work on parallel-programming tools.
Co-author[5]