Senior Deep Learning Profiling Tools Engineer

NVIDIA

2.7

(9)

Multiple Locations

#JR1988113

Position summary

ineers to devise innovative approaches to hardware and software profiling and incorporate them into our internal tools. Your work will accelerate progress toward our broader mission, which spans both hardware and software, to consistently deliver the world's fastest accelerated computing systems in domains ranging from autonomous vehicles to supercomputers. Join our technically diverse team of GPU architects, software engineers, and infrastructure experts to advance the frontiers of computing performance!

What you'll be doing:

  • Design and develop software tools to profile and analyze AI workloads on NVIDIA systems

  • Work with multi-disciplinary teams to design, implement, and verify new features for profiling and monitoring, often incorporating new hardware capabilities

  • Define software/hardware metrics for performance analysis of AI workloads and verify them for upcoming architectures

  • Constantly learn about the latest techniques and frameworks for training, deploying, and optimizing the performance of AI to improve the efficiency and effectiveness of our tools

What we need to see:

  • Bachelors, Masters, or PhD in relevant field (e.g. CS, EE, CE) or equivalent experience

  • 8+ years of relevant experience (including graduate work if applicable)

  • Proficiency in C++ and Python.

  • Experience with deep learning Frameworks (e.g. PyTorch, JAX, TRT, ONNX, Triton) and strong understanding of deep learning fundamentals

  • Strong computer science fundamentals - algorithms, data structures, optimization, debugging, operating systems, and parallel computing

Ways To Stand Out From The Crowd:

  • Experience with performance analysis of AI training/inference applications

  • Knowledge of device drivers and/or compiler implementation

  • Knowledge of GPU and/or CPU architecture and general computer architecture principles

The base salary range is 184,000 USD - 287,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.