Sr GPU Machine Learning Solutions Engineer, AI Solutions

$182,800 - $274,200/Yr

AMD - Santa Clara, CA

Full-time - Mid Level

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

The Sr GPU Machine Learning Solutions Engineer at AMD is responsible for enabling and optimizing software for next-generation GPU computational accelerators. This role involves working with a world-class team to develop custom AI software solutions for industry-leading AI customers, focusing on performance analysis and optimization in both on-prem and cloud environments. The engineer will leverage their expertise in programming and software development methodologies to push the boundaries of efficiency and performance in AI applications.

Responsibilities

• Enable models, libraries, and applications for Instinct GPUs in on-prem and Cloud environments.
• Analyze and optimize the performance of software for GPU computational accelerators.
• Collaborate with a team of Software Engineers to develop custom AI Software Solutions.
• Engage with clients to understand their needs and provide tailored solutions.
• Contribute to open-source software development and collaborate with community maintainers.
• Utilize software engineering methodologies such as Agile, Scrum, and Kanban.

Requirements

• Strong programming skills in C/C++.
• Experience with GPU kernel programming using CUDA, HIP, or OpenCL.
• Performance analysis skills for both CPU and GPU.
• Experience with Singularity, Docker, and/or Kubernetes.
• Experience with open-source software development.
• Excellent analytical and problem-solving skills.
• Ability to work independently and as part of a team.
• BS, MS, or PhD in Computer Science, Computer Engineering, or a related field with several years of related experience.

Nice-to-haves

• Experience in Python.
• Experience developing software in an end customer product delivery environment.
• Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Experience in all phases of software development from requirement gathering to final release.
• Experience providing clear and timely communication related to project status to leadership.

Benefits

• Base pay competitive with market rates.
• Annual bonus or sales incentive eligibility.
• Opportunity to own shares of AMD stock.
• Discount on AMD stock through Employee Stock Purchase Plan.