Senior AI GPU Compiler Engineer

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

• 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

PREFERRED EXPERIENCE: 

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

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

#LI-JG1

At AMD, your base pay is one part of your total rewards package.  Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD’s Employee Stock Purchase Plan. You’ll also be eligible for competitive benefits described in more detail here.

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.