Sr GPU Machine Learning Solutions Engineer, AI Software Solutions

AMD - Bellevue, WA

Full-time - Mid Level

Bellevue, WA

Computer and Electronic Product Manufacturing

About the position

The Software Engineer position at AMD focuses on enabling and optimizing software for next-generation GPU computational accelerators, particularly in the context of artificial intelligence applications. The role involves working with a team to develop custom AI software solutions for industry-leading clients, pushing the boundaries of efficiency and performance in both on-prem and cloud environments. Candidates will engage directly with clients and leverage the latest hardware capabilities to solve complex challenges in the AI landscape.

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.
• Utilize programming skills in C/C++ and potentially Python to develop software.
• Participate in all phases of software development, including requirement gathering, analysis, design, development, testing, and release.

Requirements

• Strong programming skills in C/C++.
• Experience with GPU kernel programming using CUDA, HIP, or OpenCL.
• Experience with performance analysis for both CPU and GPU.
• Experience with Singularity, Docker, and/or Kubernetes.
• Experience with open-source software development and collaboration with community maintainers.
• Familiarity with software engineering methodologies such as Agile, Scrum, or Kanban.
• BS in Computer Science or Computer Engineering with several years of related experience, or MS/PhD with relevant experience.

Nice-to-haves

• Experience in Python programming.
• Experience in an end customer product delivery environment.
• Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Experience providing clear and timely communication related to project status to leadership.

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 with discounts.