NPU AI Compiler Engineer — MLIR & Performance

Advanced Micro Devices, Inc.; San Jose (CA); Hybrid; USD 150,000 - 200,000; Full time

Benefits offered by this job

Hybrid work model

Job summary

Advanced Micro Devices, Inc. (AMD) seeks a specialized software engineer to advance compiler tools for AI workloads on NPUs within the Vitis-AI and Ryzen-AI software platforms.

You will join a core team focused on optimizing MLIR passes, enabling next-gen AI accelerators, and collaborating with hardware teams on AMD devices with AI-engines. The role requires strong C++ and Python, hands-on MLIR experience, and deep knowledge of compiler architecture.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent.

Responsibilities

• Enhance, implement and design compiler passes in MLIR for AI workloads on NPUs.
• Analyze and debug functional and performance issues.
• Build infrastructure for regression suite analysis and debugging.
• Apply AI and coding agents to compiler development, debugging, test generation, and performance analysis.
• Target optimization using data-driven approaches.
• Debug existing issues and research more efficient solutions.
• Develop technical relationships with peers and partners.

Skills

C/C++

MLIR

Compiler optimization

Python

NPU architecture

AI / ML engineering

SystemC debugging

Education

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

Tools

SystemC

RTL simulators

LLVM

Job description

Advanced Micro Devices, Inc. (AMD) seeks a specialized software engineer to advance compiler tools for AI workloads on NPUs within the Vitis-AI and Ryzen-AI software platforms.

You will join a core team focused on optimizing MLIR passes, enabling next-gen AI accelerators, and collaborating with hardware teams on AMD devices with AI-engines. The role requires strong C++ and Python, hands-on MLIR experience, and deep knowledge of compiler architecture.