Senior Software Engineer, CUTLASS Performance

Nvidia Corporation; Santa Clara (CA); On-site; USD 152,000 - 287,500; Full time

Job summary

Nvidia Corporation is seeking an experienced professional to benchmark deep learning model performance in Santa Clara. The role demands a Masters or PhD in Computer Science and strong programming skills in Python and C++. You will work on performance optimization of GPU kernels and engage with teams across the organization.

The expected salary for this position ranges from 152,000 USD to 241,500 USD for Level 3, and 184,000 USD to 287,500 USD for Level 4, alongside equity and benefits.

Qualifications

• 3+ years of relevant industry experience.
• Experience with performance benchmarking of DL frameworks like PyTorch or JAX.
• Deep understanding of state-of-the art DL model architectures.

Responsibilities

• Benchmark the performance of deep learning models' inference and training.
• Identify gaps between theoretical and realized performance.
• Develop tooling to automate benchmarking and optimization processes.
• Engage with various teams at NVIDIA as a performance resource.

Skills

Programming in Python

Programming in C++

Software performance analysis

Deep understanding of computer architecture

Education

Masters or PhD degree in Computer Science or related field

Job description

NVIDIA's high-performance computing platforms are powering the AI revolution across many applications and industries. Within our software stack, CUTLASS stands out as a popular open-source ecosystem dedicated to high-performance linear algebra and Tensor Core primitives. Since 2017, it has provided the community with C++ and Python abstractions to implement custom matrix multiply (GEMM) and related math and deep learning computations on NVIDIA GPUs.

What you'll be doing:

• Benchmark the performance of state-of-the-art deep learning models' inference and training passes to identify key GPU kernel and fusion opportunities.
• Identify gaps between theoretical and realized performance, and suggest software improvements or model adjustments to resolve them.
• Develop tooling to automate the benchmarking, analysis, and performance optimization loop to push the limit of CUTLASS kernel performance within DL networks.
• Be the authoritative resource on kernel performance in the team and engage with teams across NVIDIA including GPU architecture, DL frameworks, and QA as the performance representative for the CUTLASS team.

What we need to see:

• Masters or PhD degree in Computer Science, Computer Engineering, or related field (or equivalent experience).
• 3+ years of relevant industry experience.
• Strong programming skills in Python and C++.
• Experience in software performance analysis and optimization.
• Deep understanding of computer architecture and familiarity with GPUs or similar parallel processing architectures.

Ways to stand out from the crowd:

• Deep understanding of state-of-the art DL model architectures.
• Hands-on experience with performance benchmarking of DL frameworks like PyTorch, JAX, SGLang, vLLM, TRT-LLM, or others.
• Experience in developing performance models and performance regression systems.

NVIDIA provides a salary range of 152,000 USD – 241,500 USD for Level 3, and 184,000 USD – 287,500 USD for Level 4. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until June 5, 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.