Senior AI Training Performance Architect
NVIDIA, US, CA, Santa Clara
About the job
We are now looking for a Senior AI Training Performance Architect
NVIDIA is seeking a senior engineer who is obsessed with performance analysis and optimization to help us squeeze every last clock cycle out of AI training, the workload driving the design and construction of the largest and most powerful compute systems in the world. If you are willing to work across all layers of the hardware/software stack - from GPU architecture to the application code - to achieve peak performance, we want to hear from you. This role offers the opportunity to directly impact the hardware and software roadmap in a fast-growing technology company that leads the AI revolution. Join us and help design and build the world's most powerful compute systems!
What you will be doing
- Understand, analyze, profile, and optimize AI training workloads on state-of-the-art hardware and software platforms.
- Identifying performance bottlenecks of AI training on GPUs, prioritizing and then solving problems across the key AI training workloads.
- Implement production-quality software across multiple layers of NVIDIA's deep learning platform stack, from drivers to DL frameworks.