Senior Software Engineer, AI Performance Analysis
$184k - $357k • US, CA, Santa Clara
Posted 1d ago
About the job
NVIDIA GPU Architecture Group is seeking a senior software engineer to automate and optimize performance analysis workflows for AI training and inference workloads. You will not only perform analysis but also reshape how it's done, building tools and workflows that scale across a broad audience of engineers. NVIDIA is a world-class leader in AI workload optimization, and your work will push these boundaries even further!
Responsibilities
- Design and build performance analysis tools and workflows for AI training and inference workloads.
- Translate AI performance engineer 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 align profiling capabilities with AI workloads.
- Identify performance bottlenecks in AI workloads and develop automated detection and diagnosis approaches.
Requirements
- M.S. or PhD in Computer Science, Computer Engineering, or 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 for other engineers.
- Strong software development skills, with Python and C++ preferred.
- Ability to translate user requirements into scalable tooling solutions.
- Familiarity with AI-enabled tooling for software development and performance analysis.
- Strong interpersonal skills for understanding engineer difficulties and working across multi-functional teams.
Benefits
- Equity
- Benefits