Senior AI GPU Compiler Engineer - LLVM

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. AMD together we advance_ THE ROLE: In this exciting opportunity, you will be immersed in the world of open-source AI compiler and runtime stack. Your responsibilities will encompass gaining a comprehensive understanding of compiling Machine Learning networks specifically tailored for AMD graphics cards. This hands-on experience will not only involve delving into the fundamentals of GPU hardware capabilities but also actively engaging with MLIR dialects, execution models, and code generation. As part of your role, you will have the chance to contribute to the development of various components within a sophisticated software tool chain, all crafted in C++. Get ready to explore and contribute to the cutting edge of AI compilation technology. THE PERSON: The ideal candidate is someone who embodies a genuine passion for software engineering and demonstrates strong leadership skills, capable of steering complex issues to successful resolutions. They should excel in effective communication and be adept at collaborating seamlessly with diverse teams across AMD. KEY RESPONSIBILITIES: Implement and optimize GPU kernels using cutting-edge compiler technology Conduct benchmarking on prevalent AI model workloads, including tasks in vision, speech recognition, language processing and generative models Collaborate with AMD’s architecture specialists to enhance future products Stay abreast of software and hardware trends and innovations, particularly in algorithms and architecture, ensuring the integration of the latest advancements into our projects Participating in new GPU bring ups for AI software Debug and resolve existing issues and research alternative, more efficient ways to accomplish the same work Cultivate technical relationships with peers and partners, fostering collaborative efforts and knowledge exchange KEY QUALIFICATIONS: Strong object-oriented programming background, C/C++ preferred Proficient in crafting high performance GPU kernels Prior experience with compilers, dataflow graphs Familiarity with modern source control systems such as GitHub Excellent verbal communication and writing skills WHAT MAKES YOU STAND OUT: Proven expertise in optimizing high performance GPU kernels Prior hands-on experience using AI accelerators Contributions in LLVM or MLIR projects Previous exposure to machine learning models A collaborative team player with experience working in geographically dispersed teams ACADEMIC CREDENTIALS: Advanced degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent LOCATION: UK, Ireland, Germany or Spain. Remote or hybrid. #LI-DB1 #LI-REMOTE 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.THE ROLE: In this exciting opportunity, you will be immersed in the world of open-source AI compiler and runtime stack. Your responsibilities will encompass gaining a comprehensive understanding of compiling Machine Learning networks specifically tailored for AMD graphics cards. This hands-on experience will not only involve delving into the fundamentals of GPU hardware capabilities but also actively engaging with MLIR dialects, execution models, and code generation. As part of your role, you will have the chance to contribute to the development of various components within a sophisticated software tool chain, all crafted in C++. Get ready to explore and contribute to the cutting edge of AI compilation technology. THE PERSON: The ideal candidate is someone who embodies a genuine passion for software engineering and demonstrates strong leadership skills, capable of steering complex issues to successful resolutions. They should excel in effective communication and be adept at collaborating seamlessly with diverse teams across AMD. KEY RESPONSIBILITIES: Implement and optimize GPU kernels using cutting-edge compiler technology Conduct benchmarking on prevalent AI model workloads, including tasks in vision, speech recognition, language processing and generative models Collaborate with AMD’s architecture specialists to enhance future products Stay abreast of software and hardware trends and innovations, particularly in algorithms and architecture, ensuring the integration of the latest advancements into our projects Participating in new GPU bring ups for AI software Debug and resolve existing issues and research alternative, more efficient ways to accomplish the same work Cultivate technical relationships with peers and partners, fostering collaborative efforts and knowledge exchange KEY QUALIFICATIONS: Strong object-oriented programming background, C/C++ preferred Proficient in crafting high performance GPU kernels Prior experience with compilers, dataflow graphs Familiarity with modern source control systems such as GitHub Excellent verbal communication and writing skills WHAT MAKES YOU STAND OUT: Proven expertise in optimizing high performance GPU kernels Prior hands-on experience using AI accelerators Contributions in LLVM or MLIR projects Previous exposure to machine learning models A collaborative team player with experience working in geographically dispersed teams ACADEMIC CREDENTIALS: Advanced degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent LOCATION: UK, Ireland, Germany or Spain. Remote or hybrid. #LI-DB1 #LI-REMOTEBenefits 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.