Software Engineer - ML Compilation
About The Position
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.
Requirements
- Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, related field, or equivalent
Nice To Haves
- 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
Responsibilities
- 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
Benefits
- AMD benefits at a glance
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
