Senior AI Inference Compiler Engineer — Equity Eligible

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

Job summary

A leading tech company in Austin, Texas is seeking an AI & Deep Learning Compiler Engineer to join their Deep Learning & AI Compiler team. The role involves analyzing deep learning networks, developing optimization algorithms, and collaborating with software framework and GPU architecture teams. Candidates should have a degree in Computer Science or a related field with 3+ years of experience in compiler technologies. A strong background in C/C++ and Python is essential. The position offers a competitive salary range from $152,000 to $241,500, with equity and benefits.

Qualifications

• 3+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Ability to work independently and lead development efforts.
• Strong interpersonal skills in a dynamic team.

Responsibilities

• Analyze deep learning networks and develop compiler optimization algorithms.
• Collaborate with deep learning software framework teams and GPU architecture teams.
• Define public APIs, performance optimizations, and implement compiler techniques.

Skills

Compiler optimization algorithms

C/C++ programming

Python programming

Deep learning frameworks (PyTorch, JAX)

Performance analysis

Education

Bachelor’s, Master’s or Ph.D. in Computer Science or related field

Tools

MLIR

LLVM

XLA

Triton

Job description

A leading tech company in Austin, Texas is seeking an AI & Deep Learning Compiler Engineer to join their Deep Learning & AI Compiler team. The role involves analyzing deep learning networks, developing optimization algorithms, and collaborating with software framework and GPU architecture teams. Candidates should have a degree in Computer Science or a related field with 3+ years of experience in compiler technologies. A strong background in C/C++ and Python is essential. The position offers a competitive salary range from $152,000 to $241,500, with equity and benefits.