Senior Software Engineer, AI Performance Analysis

at Nvidia

📍 Santa Clara, United States

USD 184,000-356,500 per year

SENIOR

✅ On-site

Tech Stack
Tag name is followed by "@" symbol and proficiency level value. About proficiency levels:

1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;

3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;

7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;

10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.

AI @ 4 GPU Machine Learning @ 6 Performance Analysis @ 4 Profiling @ 4 Python @ 7 Software Development @ 7

Details

NVIDIA's GPU Architecture Group is seeking a senior software engineer to automate and optimize performance analysis workflows for AI training and inference workloads. The role focuses on building scalable tools and workflows that improve how performance analysis is performed across a broad audience of engineers.

Responsibilities

Design and build performance analysis tools and workflows for AI training and inference workloads.
Understand how AI performance engineers work and translate their needs into scalable, intuitive tooling.
Develop integrations between profiling infrastructure and AI frameworks and workflows.
Collaborate with performance engineers, hardware architects, and software teams to ensure profiling capabilities align with real-world AI workloads.
Identify performance bottlenecks in AI workloads and develop automated approaches to detect and diagnose them.

Requirements

M.S. or PhD in Computer Science, Computer Engineering, or a related field, or equivalent experience.
6+ years of relevant work experience.
Deep knowledge of AI workloads, frameworks, and performance characteristics.
Experience building tools, workflows, or infrastructure used by other engineers.
Strong software development skills; Python and C++ preferred.
Ability to translate user requirements into scalable tooling solutions.
Up-to-date knowledge of AI-enabled tooling for software development and performance analysis.
Strong interpersonal skills for understanding engineer difficulties and working across multifunctional teams.

Preferred Qualifications

Experience profiling or optimizing AI training or inference pipelines at scale.
Background building developer tools or platforms for machine learning engineers.
Contributions to open-source AI tooling or frameworks.

Compensation and Benefits

The base salary range is USD 184,000–287,500 for Level 4 and USD 224,000–356,500 for Level 5. Base salary is determined based on location, experience, and the pay of employees in similar positions. The role also includes eligibility for equity and benefits.

Applications will be accepted at least until August 16, 2026. NVIDIA uses AI tools in its recruiting processes and is an equal opportunity employer.