NVIDIA
Posted 2mo ago

Deep Learning Performance Architect, CUTLASS DSL Testing

NVIDIA
Shanghai or Beijing
OnsiteFull Time
Responsibilities
  • Develop testing
  • Define strategies
  • Ensure quality
Requirements
  • MS/PhD or equivalent
  • 3+ years in software testing
  • Strong Python and scripting skills
  • Experience with test tools and automated GPU testing
  • Code coverage and regression detection
Technical tools mentioned
PythonScriptingTest ToolsGPU TestingMLIR

Job description

Are you excited about building world-class quality systems for advanced GPU software? Do you enjoy combining automation, product validation, and code analysis to support fast-moving compiler and kernel innovation? We are seeking a strong test engineer to develop the NVIDIA CUTLASS DSL testing framework, shape product test strategy, and ensure end-to-end code quality across the MLIR-based compilation pipeline. In this role, you will drive automated testing, and regression detection to make sure every code change is validated for correctness, and the product is ready for shipping at any time. 

What you'll be doing: 

  • Develop and evolve the NVIDIA CUTLASS DSL testing framework for next-generation GPU software

  • Define, refine, and execute robust product test strategies for shipping to the open-source community 

  • Ensure end-to-end code quality across the MLIR-based compilation pipeline and related functional coverage infrastructure 

  • Build automated testing, code coverage measurement, and regression detection workflows at scale 

  • Partner with multiple teams to make sure every operator change meets a high bar for correctness, quality, and performance 

What we need to see: 

  • MS, PhD, or equivalent experience in Computer Science, Software Engineering, or a related field 

  • 3+ years of relevant work experience 

  • Excellent Python and scripting skills 

  • Strong experience developing and using test tools, with a solid understanding of software testing best practices 

  • Hands-on experience with automated testing in GPU environments, including correctness testing, code coverage improvements, and regression detection 

  • Strong communication skills and proven ability to collaborate effectively across teams 

Ways to stand out from the crowd: 

  • Familiarity with common AI agent technologies and applications 

  • Experience in quality assurance of open-source products 

About NVIDIA

Designs GPU-accelerated computing and artificial intelligence hardware.

Similar jobs

Deep Learning Performance Architect roles near Shanghai, Shanghai
3w
Save
Mark Applied
Hide
Deep Learning Performance Architect
Shanghai, Shanghai, China
OnsiteFull Time
NVIDIA
NVIDIANASDAQ: NVDA: Designs graphics processing units and artificial intelligence hardware.
3+ YOEBSc/MS/PhD in CS/EE/Math; experience with GPU-based deep learning platforms and system architecture; performance analysis and optimization experience.
GPU