Senior CUDA Kernel Engineer - Tensor Core GEMM

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 skilled developer to join their CUTLASS team in Austin, Texas, focusing on Tensor Core-based deep learning kernels. The ideal candidate will possess advanced C++ programming skills and experience in CUDA, with a strong background in computer architecture. This role offers competitive salary packages and opportunities for equity and benefits.

Join NVIDIA's innovative environment alongside passionate and creative professionals dedicated to high-performance computing solutions that drive the AI revolution.

Qualifications

• 3+ years of relevant industry experience.
• Strong proficiency in C++ programming and software design.
• Experience with CUDA, OpenCL, HIP, and other programming languages.
• Deep understanding of computer architecture.

Responsibilities

• Write Tensor Core-based deep learning kernels.
• Optimize kernels for peak throughput.
• Collaborate with teams across NVIDIA for kernel delivery.

Skills

C++ programming

CUDA

Software design

Debugging

Performance evaluation

Education

Masters or PhD in Computer Science, Computer Engineering

Tools

CUDA

OpenCL

HIP

Mojo

Pallas

Triton

Mosaic

Halide

Job description

NVIDIA is seeking a skilled developer to join their CUTLASS team in Austin, Texas, focusing on Tensor Core-based deep learning kernels. The ideal candidate will possess advanced C++ programming skills and experience in CUDA, with a strong background in computer architecture. This role offers competitive salary packages and opportunities for equity and benefits.

Join NVIDIA's innovative environment alongside passionate and creative professionals dedicated to high-performance computing solutions that drive the AI revolution.