AI Frameworks - Tech Lead HW/SW Codesign

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 role, you will play an important part in helping align AMD's hardware designs with the needs of AI software today and in the future. Your responsibilities will encompass gaining a comprehensive understanding of Machine Learning networks, the important kernels that are used within them, and how best to map them to current and future hardware features. This hands-on experience will not only involve delving into the fundamentals of GPU hardware capabilities but also actively engaging with compiler teams, kernel teams and hardware design teams. As part of your role, you will have the chance to contribute to the development of various components within a sophisticated software stack, the workloads themselves, and ultimately to the hardware designs needed to make them better. Get ready to explore and contribute to the cutting edge of AI technology.

THE PERSON:

The ideal candidate is someone who embodies a genuine passion for software engineering while also being able to understand hardware operations and limitations. The candidate needs strong leadership skills and be 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:

• Understand, implement and optimize GPU kernels using cutting-edge compiler technology
• Holistically understand and 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, including prototyping in emulated environments.
• 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

PREFERRED EXPERIENCE:

• Strong object-oriented programming background, C/C++ preferred
• Proficient in crafting high performance GPU kernels
• Previous exposure to cutting edge machine learning models
• Familiarity with modern source control systems such as GitHub
• Excellent verbal communication and writing skills
• Proven expertise in optimizing high performance GPU kernels
• Prior hands-on experience using AI accelerators
• Past experience working with hardware teams and design feedback
• A collaborative team player with experience working in geographically dispersed teams

ACADEMIC CREDENTIALS:

• Advanced degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

#LI-JG1

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.