Software Engineer - ML Compilation

AMD; San Jose (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

AMD is seeking an AI/ML software/compiler engineer to work on the next generation compiler tools for Vitis-AI and Ryzen-AI software platforms. You will contribute to compiling machine learning networks across AMD devices with AI-engines.

The role involves collaborating with senior developers and domain experts in a dynamic, fast-paced environment, focusing on building a robust compiler from the ground up while supporting cross-team integration.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or related field.
• Strong programming skills in C++ and experience with compiler/tooling.
• Experience with MLIR and dialects is a plus.
• Proficiency with GitHub or modern version control workflows.
• Knowledge of machine learning model structures and AI development tasks.

Responsibilities

• Apply data-driven and structured analysis to optimize compiler performance.
• Contribute to AMD’s AI/ML compiler innovations and architectural changes.
• Collaborate with multiple engineering teams across geographies.
• Debug, analyze, and resolve issues; explore more efficient alternatives.

Skills

C++

Compiler stack

MLIR

GitHub

ML models

AI agents

Education

BSc/MSc in CS/CE/EE or related

Tools

GitHub

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.
At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And we’re looking for talent who feel the same: people who want to leave the planet better than they found it, those who don’t shy away from humanity’s challenges but are determined to help solve them.

ADVANCE YOUR CAREER. ADVANCE THE WORLD.
At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And we’re looking for talent who feel the same: people who want to leave the planet better than they found it, those who don’t shy away from humanity’s challenges but are determined to help solve them.
AMD is powering the next generation of supercomputing, high-performance computing, cloud, and AI. Whether you’re designing next-gen processors, enabling AI breakthroughs, or creating go-to-market plans, every role at AMD contributes to something bigger — technology that moves the world forward.

The Role

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 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

Specifically, the candidate will work on a new compiler along with other senior developers and domain experts. It will involve working in a dynamic and fast paced development environment, and require collaboration with multiple engineering teams that are geographically spread. This is an exciting opportunity involved in a roadmap to build the next generation compiler from the bottom-up. Some general expectation on the role are:

• Apply a data-driven and structured approach for analysis and optimization efforts
• Contribute to AMD's innovation, algorithms, and architectural design changes
• Develop relationships with technical leads and peers
• Debug, analyze, resolve existing issues, and explore more efficient alternatives

Preferred Experience

• Strong programming skills in C++
• Experience with compilation stack to any custom hardware accelerator
• Knowledge of MLIR and dialects is an added plus
• Prior work using Modern source control system such as GitHub
• Knowledge of machine learning model structures
• Experience with AI agents for development or qualification tasks

Academic Credentials

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

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.