Senior AI Cluster Tools Developer

What you'll be doing:

• Build internal perf/power profiling and analysis tools and platform for AI workloads at cluster scale

• Build debugging tools for common encountered problems in GPU cluster

• Work with our users to build / calibrate perf/power models for next generation HW or system

• Partner with architects to propose new HW features or improve existing features with real world use cases

What we need to see:

• BS+ in Computer Science or related (or equivalent experience) and 5+ years of software development

• Strong software design and implementation ability with Python/Go/C++

• Good understanding of Deep Learning and AI frameworks like Pytorch, TensorFlow and etc

• Knowledge of AI cluster job scheduling, storage management and networking management

• Knowledge of Linux kernel

• Excellent problem solving skills and project management skills

• Flexibility for working in an evolving environment with changing requirements

Ways to stand out from the crowd:

• Proven experience in GPU cluster scale continuous profiling & analysis tools/platforms

• Solid experience in large AI job troubleshooting and failure detection/recovery

• Skillful in Deep Learning application performance analysis and optimization

• Knowledgeable in GPU / CPU architecture and application performance or power efficiency analysis

NVIDIA offers highly competitive salaries and a comprehensive benefits package. We have some of the most brilliant and talented people in the world working for us and, due to unprecedented growth, our world-class engineering teams are growing fast. If you're a creative and autonomous engineer with real passion for technology, we want to hear from you.

#LI-Hybrid

The base salary range is 148,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.