AI Inference Compiler Engineer — Shape Next-Gen GPU AI

NVIDIA; Santa Clara (CA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity benefits

Generous benefits package

Job summary

NVIDIA is seeking an AI Compiler Engineer in Santa Clara, CA, to drive innovation in compiler technology and improve AI performance. This role focuses on enhancements for next-generation GPUs, solving AI workload compilation challenges, and collaborating on hardware/software co-design.

The ideal candidate should have a BS or MS in Computer Science, with 3+ years of compiler experience, as well as proficiency in C/C++ and Python. This position offers competitive salaries ranging from $152,000 to $241,500 and equity benefits, supporting a diverse and inclusive work environment.

Qualifications

• 3+ years of experience in compiler optimizations and synthesis.
• Hands-on experience with MLIR.
• Strong communication and interpersonal skills.

Responsibilities

• Develop kernel generation and computational graph optimizations.
• Solve complex compilation problems for AI workloads.
• Collaborate on hardware/software co-design.

Skills

C/C++ programming

Python programming

Compiler optimizations

Performance analysis

MLIR

Education

BS or MS in Computer Science or related field

PhD in relevant area

Tools

AI workload frameworks

Job description

NVIDIA is seeking an AI Compiler Engineer in Santa Clara, CA, to drive innovation in compiler technology and improve AI performance. This role focuses on enhancements for next-generation GPUs, solving AI workload compilation challenges, and collaborating on hardware/software co-design.

The ideal candidate should have a BS or MS in Computer Science, with 3+ years of compiler experience, as well as proficiency in C/C++ and Python. This position offers competitive salaries ranging from $152,000 to $241,500 and equity benefits, supporting a diverse and inclusive work environment.