Senior CUDA Kernel Engineer - Tensor Core GEMM

NVIDIA; Redmond (WA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity and benefits

Job summary

NVIDIA is seeking a talented individual for the CUTLASS team, focusing on high-performance computing and deep learning. You will write and optimize Tensor Core-based deep learning kernels, collaborating with various teams to deliver efficient and effective software solutions.

The ideal candidate holds a Masters or PhD in Computer Science or a related field, with strong skills in C++ and multiple parallel programming tools. NVIDIA offers a competitive salary range and a commitment to diversity and inclusion.

Qualifications

• 3+ years of relevant industry experience.
• Strong proficiency in C++ programming and software design.
• Experience with parallel programming languages targeting accelerators.

Responsibilities

• Write Tensor Core-based deep learning kernels using CUTLASS.
• Optimize kernels for peak throughput.
• Collaborate with various teams across NVIDIA.

Skills

C++ programming

Software design

CUDA

Deep understanding of computer architecture

Education

Masters or PhD in Computer Science or related field

Tools

CUDA

OpenCL

HIP

SYCL

Mojo

Pallas

Triton

Mosaic

Halide

Job description

NVIDIA is seeking a talented individual for the CUTLASS team, focusing on high-performance computing and deep learning. You will write and optimize Tensor Core-based deep learning kernels, collaborating with various teams to deliver efficient and effective software solutions.

The ideal candidate holds a Masters or PhD in Computer Science or a related field, with strong skills in C++ and multiple parallel programming tools. NVIDIA offers a competitive salary range and a commitment to diversity and inclusion.