AI Compiler Engineer for GPU & MLIR Optimizations

NVIDIA Corporation; Austin (TX); On-site; USD 184,000 - 288,000; Full time

Job summary

NVIDIA seeks an AI Compiler Engineer to advance compiler technologies and developer tools through applied machine learning and AI. You will work with top engineers to shape compiler capabilities, design end-to-end optimization workflows, and deploy learning-based decision systems integrated with compiler passes.

The role emphasizes collaboration, reliability, and performance across workloads, with base salary ranges and equity benefits.

Qualifications

• BS/MS/PhD in Computer Science or a related field (or equivalent experience).
• 8+ years of software engineering and AI/ML experience, preferably in tools or systems development.
• Strong knowledge of compilers, code generation, and GPU architecture.
• Proficiency in Python and C/C++.
• Hands-on experience with LLVM/MLIR-based compiler development, including optimization passes, code generation, or frontend integration.
• Experience building or integrating LLM-based systems, agents, or developer tooling in production environments.

Responsibilities

• Join a team driving innovative solutions in compilers and developer tools through applied machine learning and AI.
• Collaborate with world-class engineers to shape next-generation compiler capabilities that power large-scale, high-impact products.
• Design and implement end-to-end compiler optimization workflows, from feature engineering and model development to compiler integration and production rollout.
• Develop and integrate learning-based decision systems into compiler passes, code generation flows, and optimization pipelines.
• Build robust interfaces between LLM systems and compiler infrastructure for optimization recommendations, decision support, and workflow automation.
• Partner with compiler, architecture, and performance teams to validate impact on representative workloads and benchmark suites.
• Lead experimentation, evaluation, and deployment with strong standards for reliability, scalability, and performance.

Skills

Python

C/C++

LLVM/MLIR

ML/AI

GPU architecture

Education

BS/MS/PhD in Computer Science/related

Job description

NVIDIA seeks an AI Compiler Engineer to advance compiler technologies and developer tools through applied machine learning and AI. You will work with top engineers to shape compiler capabilities, design end-to-end optimization workflows, and deploy learning-based decision systems integrated with compiler passes.

The role emphasizes collaboration, reliability, and performance across workloads, with base salary ranges and equity benefits.