Senior AI GPU Compiler Engineer
CV Library, Austin, TX, United States
Overview:
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- 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_
Responsibilities:
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 . 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, 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++
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
Qualifications:
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 .
AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or -based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to , ancestry, , marital status, medical condition, mental or physical , , , , political and/or third-party affiliation, , , , , 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.