Sr GPU Machine Learning Solutions Engineer, AI Solutions

AMD - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

The Sr GPU Machine Learning Solutions Engineer at AMD is a senior-level role focused on enabling and optimizing software for advanced GPU computational accelerators in both on-premises and cloud environments. This position involves working with a team of software engineers to develop custom AI software solutions, fine-tune large language models, and enhance the performance of AI applications. The role requires strong programming skills, particularly in C/C++, and experience in GPU kernel programming, performance analysis, and client engagement.

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 push the boundaries of efficiency and performance.
• Engage directly 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 relevant 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 to leadership regarding project status.

Benefits

• Base pay dependent on skills and experience.
• Eligibility for annual bonuses or sales incentives.
• Opportunity to own shares of AMD stock through the Employee Stock Purchase Plan.
• Competitive benefits package.