SR. ML Compilation Engineer

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 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., 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, andexplore more efficient alternatives, AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

• 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, * Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, related field, or equivalent

$172,000.00/Yr.-$258,000.00/Yr.

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.