Software Engineer, Performance Instrumentation

Thomas To; Santa Clara (CA); On-site; USD 168,000 - 270,000; Full time

Benefits offered by this job

Equity

Job summary

NVIDIA is looking for a Software Engineer, Performance Instrumentation to join our Santa Clara team. You will define data requirements, close visibility gaps, and implement low-overhead instrumentation across ML workloads and system software.

You’ll collaborate with partner teams to ensure data quality, measure overhead, and deliver robust telemetry for high-performance computing. The role requires a strong background in Python and C++, CUDA expertise, and experience with PyTorch or JAX.

Qualifications

• Bachelor’s degree in Computer Science, Computer Engineering, or a related field, or equivalent experience.
• 8+ years of relevant software engineering experience in instrumentation or tracing in production ML or systems software.
• Proficiency in CUDA, and working knowledge of PyTorch or JAX, plus NVTX annotation/tracing APIs.
• Experience maintaining data quality in a pipeline with validation, checks, alerting, and provenance.
• Ability to measure instrumentation overhead and trade-offs among overhead, fidelity, sampling, and aggregation.
• Track record implementing changes in code managed by other groups, with clear interfaces and delivery.
• Strong debugging across component boundaries with code reviews, tests, and automated tools.
• Excellent communication skills to convey evidence, uncertainty, and next steps to engineers and data users.
• Proficiency in Python and C++.

Responsibilities

• Define and maintain performance data requirements from profiling libraries, annotation APIs, frameworks, runtimes, and telemetry.
• Identify visibility gaps and collaborate with teams to address or build solutions.
• Inform build of low-overhead instrumentation and annotation APIs for consumers.
• Ensure data quality at ingestion with validation, sanitization, checks, and alerting.
• Measure overhead and observer effects on real workloads to keep within acceptable limits.
• Collaborate with multiple partner teams, managing commitments and progress.

Skills

Python

C++

CUDA

Annotation/Tracing APIs (NVTX)

Education

Bachelor’s degree in CS/EE or equivalent experience

Tools

NVTX

Profiling libraries

Annotation APIs

Job description

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in an inclusive, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

NVIDIA is looking for a Software Engineer, Performance Instrumentation to join our outstanding team in Santa Clara, CA. This is an outstanding opportunity to be part of a world-class organization, driving innovation at the intersection of AI and accelerated computing. As a part of our team, you will work on groundbreaking projects and have a direct impact on the future of technology.

What You’ll Be Doing

• Define and maintain the performance data requirements from various sources including profiling libraries, annotation APIs, frameworks, runtimes, communication libraries, schedulers, configurations, and infrastructure telemetry.
• Identify and prove visibility gaps with concrete evidence, demonstrating which diagnoses are not possible with current data. Collaborate with relevant teams to address these gaps or build solutions where necessary.
• Inform the build of low-overhead instrumentation and annotation APIs from the perspective of the consumer, ensuring the necessary information is conveyed efficiently and effectively.
• Ensure data quality at ingestion by implementing validation, sanitization, checks, and alerting mechanisms, so that any gaps or errors are detected before impacting users or automated analysis.
• Measure and manage the overhead and observer effects of instrumentation on real workloads, ensuring they remain within acceptable limits.
• Collaborate with multiple partner teams simultaneously, managing commitments and ensuring continuous progress.

What We Need To See

• Bachelor’s degree in Computer Science, Computer Engineering, or a related field, or equivalent experience.
• 8+ years of relevant software engineering experience, including hands-on experience with instrumentation or tracing in production machine learning or systems software.
• Proficiency in CUDA, and working knowledge of machine learning frameworks like PyTorch or JAX, as well as annotation or tracing APIs such as NVTX.
• Demonstrated experience in maintaining data quality in a pipeline, including validation, checks, alerting, and provenance.
• Proven ability to measure instrumentation overhead and make trade-offs among overhead, fidelity, sampling, and aggregation.
• Proven history of implementing modifications in code managed by other groups by identifying requirements, reaching consensus on interfaces, and ensuring completion through to delivery.
• Strong debugging skills across component boundaries and the ability to validate changes through code review, tests, and independent measurements, including using automated development tools.
• Excellent communication skills, especially in conveying evidence, uncertainty, and next steps with engineers and data users.
• Proficiency in Python and C++.

Ways To Stand Out From The Crowd

• Experience establishing telemetry conventions or schemas adopted by other teams.
• Background in storage, checkpointing, or data-loading performance.
• Expertise in crafting low-overhead event collection.
• Familiarity with network and collective-communication telemetry, including NCCL, NICs, and fabric counters.

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

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 2, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.