NPU Compiler Engineer

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

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMDis shapingthefuture.

Whetheryou’redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger— technologythat moves the world forward.Join us and, together, we’ll advance your career.

AMD has a broad portfolio of several ML accelerator devices with NPUs (neural processing units) that are integrated in its Client and Embedded product portfolio. The Vitis unified platform Vitis Unified Software Platform targets several AMD Embedded accelerator cards. The Ryzen AI software platform Ryzen AI Software targets several client CPU devices. Both these devices have NPUs that are based of AMD's proprietary AI-engine architecture.

THE ROLE:

AMD is looking for a specialized software engineer who is passionate about improving the performance of key applications and benchmarks. You will be a member of a core team of incredibly talented industry specialists and will work with the very latest hardware and software technology.

THE PERSON:

The ideal candidate should be passionate about AI/ML engineering and possess a drive to solve sophisticated issues in a complex compiler tool stack. This role will be in software development, in the AI group in AMD. It will involve working on the next generation compiler tools to enhance the capabilities in Vitis-AI and Ryzen-AI software platforms, to compile Machine Learning networks in a broad scope, and cater to the cutting-edge AMD devices with AI-engines.

KEY RESPONSIBILITIES:

• Enhance, implement and design and implement compiler passes in MLIR framework for implementing and optimizing AI workloads on NPU.
• Analyze and debug functional and performance issues.
• Build infrastructure for analysis and debug of regression suites
• Apply AI and coding agents to compiler development, debugging, test generation, and performance analysis.
• Apply a data minded approach to target optimization efforts
• Debugging /fix existing issues and r esearch alternative, more efficient ways to accomplish the same work
• Develop technical relationships with peers and partners

PREFERRED EXPERIENCE:

• Strong object-oriented programming background, C/C++ preferred
• Deep hands-on experience with MLIR optimization and lowering passes
• Strong knowledge of compiler architecture, intermediate representations, transformation passes, and optimization pipelines.
• Strong understanding of NPU architecture, including parallel execution, memory hierarchy, caches, registers, synchronization, and occupancy.
• Strong C++ and python skill
• Experience with AI compilers, tensor or graph compilers, machine learning frameworks
• Experience with debugging using SystemC and RTL simulators

ACADEMIC CREDENTIALS:

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

#LI-JG1

#hybrid

This role is not eligible for visa sponsorship.

Benefits offered are described: AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.