Senior AI GPU Compiler Engineer - LLVM

TN United Kingdom; Milton Keynes; Hybrid; GBP 60,000 - 100,000; Full time

Job summary

An established industry player is seeking a passionate software engineer to join their innovative team. This exciting role focuses on the development of open-source AI compilers and runtime stacks, specifically tailored for advanced GPU architectures. You will work on optimizing GPU kernels and benchmarking AI models, collaborating with top architecture specialists to push the boundaries of technology. If you are eager to contribute to cutting-edge projects and thrive in a collaborative environment, this opportunity is perfect for you.

Qualifications

• Strong background in object-oriented programming and GPU kernel optimization.
• Experience in compilers and dataflow graphs is essential.

Responsibilities

• Implement and optimize GPU kernels using advanced compiler technology.
• Conduct benchmarking on AI model workloads in various domains.

Skills

C/C++ Programming

GPU Kernel Optimization

Compiler Experience

Dataflow Graphs

GitHub

Machine Learning Exposure

Education

Advanced degree in Computer Science

Electrical Engineering or equivalent

Job description

Social network you want to login/join with:

col-narrow-left

Client:

Advanced Micro Devices, Inc

Location:

Milton Keynes, United Kingdom

Job Category:

-

EU work permit required:

Yes

col-narrow-right

Job Reference:

7ef9cbac5ddf

Job Views:

2

Posted:

02.05.2025

Expiry Date:

16.06.2025

col-wide

Job Description:

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.