Senior GPU Kernel Engineer, CUTLASS Platform

NVIDIA; Austin (TX); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a passionate software engineer to join the CUTLASS team in Austin, Texas. This role involves developing core components of CUDA C++ and CUTLASS Python DSL, contributing to compiler advancements, and showcasing GPU hardware features through example kernels.

Ideal candidates will have a Master's or PhD in relevant fields, coupled with 3+ years of industry experience in C++ programming and high-performance code generation. NVIDIA offers an inclusive workplace and diverse benefits, making it a desirable employer in the tech industry.

Qualifications

• 3+ years of relevant industry experience.
• Experience with high-performance code generation.
• Deep understanding of computer architecture and parallel computing.

Responsibilities

• Develop core components of the CUTLASS platform.
• Contribute to the MLIR-based backend compiler stack.
• Author example kernels utilizing CUTLASS abstractions.

Skills

C++ programming

Software design

Debugging

Performance evaluation

Testing

Education

Masters or PhD in Computer Science or related field

Job description

NVIDIA is seeking a passionate software engineer to join the CUTLASS team in Austin, Texas. This role involves developing core components of CUDA C++ and CUTLASS Python DSL, contributing to compiler advancements, and showcasing GPU hardware features through example kernels.

Ideal candidates will have a Master's or PhD in relevant fields, coupled with 3+ years of industry experience in C++ programming and high-performance code generation. NVIDIA offers an inclusive workplace and diverse benefits, making it a desirable employer in the tech industry.