About this role

Join NVIDIA as a Software Engineer, ML Performance to optimize the performance of cutting-edge AI workloads across the entire technology stack. You will analyze distributed training and inference, identify bottlenecks, and drive improvements to ensure our groundbreaking solutions lead the industry. This role offers a unique opportunity to work with the latest AI technologies at an unparalleled scale.

What we are looking for

6

Optimize performance of distributed ML training and inference workloads
Analyze performance across model, framework, compiler, CUDA, and network layers
Define and track key performance metrics for GPU fleets
Diagnose and resolve performance regressions across diverse configurations
Collaborate with build partners to implement performance improvements
Proficiency in Python and CUDA

Skills mentioned

1

Python