Deep Learning Performance Architect, CUTLASS DSL
TECHNICAL STACK · 2 TAGS
OVERVIEW
Are you passionate about programming languages, compiler technology, and GPU performance? Do you want to help shape the future of high-performance kernel development for AI? We are looking for outstanding engineers to buildCUTLASSDSL, a Python-native language for GPU kernel development, along with the MLIR dialects and lowering passes behind it. In this role, you willalsohelp accelerate kernel compilation while delivering performance comparable to CUTLASS C++, enabling efficient hardware-software co-design for NVIDIA's next generation of AI platforms.
WHATYOU'LLBE DOING:
Design, develop, andoptimizeCUTLASSDSL, a Python-native language for high-performance GPU kernel development
Build and advance the MLIR dialects, lowering passes, and code generation flows that power theCUTLASSDSL stack
Drive innovations that improve kernel compilation speed whilemaintainingperformance on par with CUTLASS C++
Collaborate closely with architecture, research, software product teams, and the open-source community to bringcutting-edgeoptimizations into real products
WHAT WE NEED TO SEE:
MS, PhD, or equivalent experience in Computer Science, Software Engineering, or a related field
2+ years ofrelevant work experience
Excellent programming skills in Python and strongproficiencyin C++
Hands-on experience with DSLs, compilers, or code generation systems
Strong command of the MLIR/LLVM stack, including IR design and pass optimization
Strong communicationskills and the ability to thrive in a highly collaborative environment
WAYS TO STAND OUT FROM THE CROWD:
Deep understanding of the CUDA GPU programming model, GPU microarchitecture, and performance analysis and optimization techniques
Familiarity with key high-performance computing abstractions such as Layout, Tile, MMA, and TMA in theCuTeecosystem
QUESTIONS AND ANSWERS
- Where is this Deep Learning Performance Architect, CUTLASS DSL role based?
- The role is based in China, Shanghai.
- What experience does NVIDIA expect for this role?
- The posting is tagged as a lead-level role, typically 7+ years of experience. Check the requirements section for specifics.
- How was this posting sourced?
- This role was pulled directly from NVIDIA's Workday careers site. Apply links open in the employer's own ATS — no reposts or aggregator middleware.
Apply links open in the employer's official ATS. Always verify recruitment messages on the company's careers page before sharing personal information.