Python ML Framework 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: As a specialist in high-performance AI accelerators, your primary responsibility is to innovate and optimize software frameworks that enhance the speed, efficiency, and scalability of artificial intelligence algorithms. This role typically sits at the intersection of research and production, requiring you to collaborate closely with both research teams pushing the boundaries of AI algorithms and production teams striving to deploy these algorithms in real-world applications. THE PERSON: The ideal candidate will have an interest in following the research space while adapting state-of-the-art ML infrastructure improvements into a robust framework. You will be familiar with modern model architectures beyond the high level structure to lower level compute requirements and techniques for improving inference at scale KEY RESPONSIBILITIES: Performance Evaluation - conduct performance evaluation of models with high level analysis of bottlenecks ML Modelling - able to reproduce model computation and evaluate correctness for novel ML architectures. Software Design - simplify model architecture into a usable framework for external vendors with an emphasis on comprehensibility Leadership - be able to delegate subcomponent design and evaluation in a fast moving field PREFERRED EXPERIENCE: Strong programming ability in Python Able to design high level APIs for serving ML models Familiarity with reading academic papers for production implementation Experience in Machine Learning model architecture ACADEMIC CREDENTIALS: Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent #LI-EM1 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.