Principal MLIR Compiler Architect for AI Acceleration

Advanced Micro Devices

San Jose (CA)

On-site

USD 180,000 - 240,000

Full time

14 days+
Application generator

A complete application in a minute — tailored resume and cover letter, ready to send.

Get past ATS filters

Job summary

AMD is seeking a Principal Software Development Engineer to lead compiler technology and MLIR-based infrastructure for AI workloads on specialized accelerators. You will steer model-to-hardware optimization, enabling high-performance execution across NPUs and other compute engines.

The role sits at the intersection of compiler development, ML frameworks, and AI model execution, requiring broad cross-functional collaboration and technical leadership to shape accelerator software roadmaps.

Qualifications

  • Strong background in compiler development spanning front-end, middle-end, and/or back-end
  • Hands-on experience with MLIR/LLVM-based infrastructure and high-performance ML workloads
  • Proven ability to optimize neural network models and graphs for accelerator hardware

Responsibilities

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

Skills

Compiler development
MLIR/LLVM
C++ development
Neural network workloads
Graph-level optimizations
Leadership

Education

Bachelor’s or Master’s degree in CS/CE/EE

Tools

MLIR
LLVM

Job description

AMD is seeking a Principal Software Development Engineer to lead compiler technology and MLIR-based infrastructure for AI workloads on specialized accelerators. You will steer model-to-hardware optimization, enabling high-performance execution across NPUs and other compute engines.

The role sits at the intersection of compiler development, ML frameworks, and AI model execution, requiring broad cross-functional collaboration and technical leadership to shape accelerator software roadmaps.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Principal MLIR Compiler Architect for AI Acceleration
Principal MLIR Compiler Architect for AI Acceleration

AMD • San Jose (CA)

On-site
USD 240,000 - 360,000
Senior Compiler Lead for ML Acceleration
Senior Compiler Lead for ML Acceleration

Socket.dev • San Jose (CA)

Hybrid
USD 180,000 - 240,000
Principal Software Development Engineer - Compiler & ML Acceleration
Principal Software Development Engineer - Compiler & ML Acceleration

Advanced Micro Devices • San Jose (CA)

On-site
USD 180,000 - 240,000
Principal Software Development Engineer – Compiler & ML Acceleration
Principal Software Development Engineer – Compiler & ML Acceleration

Socket.dev • San Jose (CA)

Hybrid
USD 180,000 - 240,000
Principal Software Development Engineer – Compiler & ML Acceleration
Principal Software Development Engineer – Compiler & ML Acceleration

AMD • San Jose (CA)

On-site
USD 240,000 - 360,000
Senior Staff Compiler Engineer - MLIR/LLVM for GPUs
Senior Staff Compiler Engineer - MLIR/LLVM for GPUs

AMD • San Jose (CA)

Hybrid
USD 180,000 - 240,000
Senior AI/ML Compiler Engineer - GPU
Senior AI/ML Compiler Engineer - GPU

Advanced Micro Devices • San Jose (CA)

On-site
USD 190,000 - 230,000
Benefits at a glance
Equal opportunity employer
Lead AI/ML Compiler Engineer for High-Performance GPUs
Lead AI/ML Compiler Engineer for High-Performance GPUs

AMD • San Jose (CA)

On-site
USD 180,000 - 240,000
Principal Compiler Software Development Engineer - AI/ML
Principal Compiler Software Development Engineer - AI/ML

Advanced Micro Devices • San Jose (CA)

On-site
USD 180,000 - 260,000
ML Compiler Engineer for AI Accelerators
ML Compiler Engineer for AI Accelerators

Advanced Micro Devices, Inc. • San Jose (CA)

Hybrid
USD 120,000 - 180,000