Principal MLIR Compiler Architect for AI Acceleration

AMD; San Jose (CA); On-site; USD 240,000 - 360,000; Full time

Job summary

AMD is seeking a Principal Software Development Engineer to lead compiler technology, MLIR-based infrastructure, and model-to-hardware optimization for AI workloads. You will shape scalable compiler pipelines and collaborate with hardware and software teams.

The role sits at the intersection of compiler development, ML frameworks, and accelerated ML execution on NPUs and specialized compute engines. We value strong C++ skills, production experience, and the ability to influence architecture

Qualifications

• Experience in compiler development across front-end/middle-end/back-end and code optimization.
• Hands-on experience with MLIR and/or LLVM-based compiler infrastructure.
• Proven experience working on neural network workloads.
• Deep understanding of graph-level and compiler-level optimizations for ML models.

Responsibilities

• Lead architecture design and development of compiler components and optimization pipelines for machine learning.
• Design and implement MLIR-based compiler flows to lower high-level ML representations into highly optimized hardware-specific code.
• Drive model compilation and data movement optimization for ML inference workloads.
• Define and implement compiler strategies for operator fusion, memory planning, scheduling, and performance optimization.
• Work cross-functionally with hardware, runtime, frontend, and systems teams to align compiler capabilities with evolving accelerator architectures.
• Provide technical leadership and mentorship, influencing compiler direction and best practices across the organization.

Skills

Compiler development

MLIR/LLVM

C++

Neural networks

NPUs/AI accelerators

ML model optimization

Production ML infra

Graph-level optimization

Education

Bachelor’s or Master’s degree in Computer Science/Computer Engineering/Electrical Engineering or equivalent

Tools

MLIR/LLVM-based compiler infrastructure

Job description

AMD is seeking a Principal Software Development Engineer to lead compiler technology, MLIR-based infrastructure, and model-to-hardware optimization for AI workloads. You will shape scalable compiler pipelines and collaborate with hardware and software teams.

The role sits at the intersection of compiler development, ML frameworks, and accelerated ML execution on NPUs and specialized compute engines. We value strong C++ skills, production experience, and the ability to influence architecture