Principal Software Engineer, E2E Performance...

NVIDIA Ltd.
Santa Clara, United States of America
13 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Compensation
$ 272K

Job location

Santa Clara, United States of America

Tech stack

Profiling
Nvidia CUDA
Computer Engineering
Memory Management
Firmware
Python
Open Source Technology
Pattern Recognition
Strategies of Testing
Large Language Models
Pandas
Information Technology
Machine Learning Operations
TensorRT

Job description

We're looking for a Principal Engineer to join our CSP Engagements team as the technical focal point for end-to-end performance, working directly with engineering teams of key CSP/hyperscale customers to ensure they achieve various performance targets on NVIDIA platforms. In this role, you will augment NVIDIA's performance and benchmark teams with a dedicated CSP-facing focus. You will drive work streams with CSP engineering teams to build shared understanding of platform performance characteristics, gather and incorporate their workload-specific feedback into NVIDIA's optimization priorities, and validate that performance targets are met in customer-representative configurations. Your cross-CSP visibility enables you to identify patterns and drive systemic improvements in documentation, configuration guidance, and tooling.

What you'll be doing:

  • Drive performance characterization work streams with engineering teams of key CSP/hyperscale customers - ensuring they understand platform performance expectations, profiling methodology, and tuning options for their specific workloads

  • Gather and synthesize CSP performance feedback - identify gaps between expected and actual throughput, and champion optimization priorities back into NVIDIA's CUDA, NCCL, driver, and firmware teams

  • Ensure key open-source performance and stress tools (e.g., STREAM, GPU Burn, GPU BLAST) are updated and validated for the latest NVIDIA rack-scale systems, GPU architectures, and CPU platforms - so customers and internal teams have reliable baseline measurements from day one

  • Work closely with CSPs to ensure their own performance and validation tooling reflects the latest GPU capabilities, memory hierarchy changes, and platform-specific tuning parameters

  • Conduct cross-CSP performance comparison and pattern analysis - identify configuration, software, or workload differences that explain performance gaps between deployments

  • Collaborate with CSPs to ensure performance-related integration work (profiling infrastructure, benchmark harnesses, config validation) is ready ahead of deployment milestones

  • Define test strategies and tooling requirements for performance validation - both for NVIDIA internal certification and customer acceptance

Requirements

  • 15+ years of experience in systems performance engineering, ideally in GPU/HPC/ML infrastructure. BS or MS in Computer Science, Computer Engineering, or related field (or equivalent experience)

  • Proficiency in GPU workload profiling: nsight systems, nsight compute, DCGM metrics, or equivalent instrumentation

  • Understanding of distributed training performance dynamics: computation/communication overlap, pipeline bubbles, memory bandwidth utilization, collective efficiency

  • Statistical methods for performance analysis: regression detection, confidence intervals, A/B comparison at scale

  • Understanding of how the full software stack impacts performance: driver overhead, collective algorithm selection, memory allocation, scheduling, firmware power management

  • Strong data analysis and visualization skills (Python, pandas, dashboards). Customer obsession - genuine passion for understanding why customers aren't achieving expected performance and driving solutions

  • Ability to communicate performance findings to both deep technical audiences and executive leadership

  • Demonstrated success influencing multiple engineering teams to prioritize performance improvements

Ways to stand out from the crowd:

  • Experience profiling and optimizing distributed training at 1000+ GPU scale (Megatron-LM, DeepSpeed, FSDP)

  • Background in ML infrastructure performance at a CSP/hyperscaler

  • Familiarity with NVIDIA platforms (DGX, HGX, NVLink topology) and profiling tools

  • Experience building automated performance regression detection systems for production environments

  • Understanding of inference workload performance dynamics (vLLM, TensorRT-LLM, SGLang, continuous batching)

Benefits & conditions

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

About the company

NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services. We have some of the most forward-thinking and hardworking people on the planet working for us. If you're creative, hardworking and self-motivated, we want to hear from you!

Apply for this position