Senior AI/ML Software Engineer

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences-building blocks for data centers, AI, PCs, gaming, and embedded systems. 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.

SENIOR AI/ML SOFTWARE ENGINEER

THE ROLE:

We are recruiting a highly qualified senior engineer to join a team at the forefront of the rapidly developing AI field, with a critical impact on AMD's AI/ML products. The team's responsibilities include providing scalable and performant solutions for efficient mapping and implementation of ML workloads on existing and next-generation AMD products, aligned with AMD's software and hardware roadmaps. We seek an outstanding engineer with a strong background in ML use cases, hardware architectures for AI/ML, and analytical modeling of ML use cases mapped onto target hardware architectures, to provide performance projections for exploring and enabling scalable, optimized solutions on current and future architectures.

KEY RESPONSIBILITIES:

• Analytical modeling of compute and data movement for performance-optimized mapping of ML use cases onto parallel processing architectures such as NPU and GPU
• Applying strong software engineering skills to implement and improve performance optimization techniques integrated into performance projection tools
• Using performance projection tools to deliver results for target ML workloads and hardware architectures
• Engaging in technical interactions with internal engineering teams and AMD customers
• Staying informed of trends in ML models/algorithms, software stacks, and hardware architectures

PREFERRED EXPERIENCE:

• Strong Python and C/C++ programming skills
• Deep understanding of AI/ML algorithms, tools/frameworks, and model representations such as ONNX and PyTorch
• Experience in analytical modeling of ML operators on target architectures, focusing on compute and data movement
• Background in using optimization libraries and solvers (e.g., PuLP, CBC, Gurobi) is a plus
• Effective communication and problem-solving skills

ACADEMIC CREDENTIALS:

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

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 employers and consider all applicants without regard to age, ancestry, color, marital status, medical condition, disability, national origin, race, religion, political 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 throughout the recruitment process.