GPU Machine Learning Solutions Engineer, AI Software Solutions

AMD • Austin, TX

439d

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 world-class 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 AI and computing.

Requirements

• Strong programming skills in C/C++ and Python.
• 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.
• Expertise in AI/ML frameworks and distributed training and inference frameworks.
• Experience with open-source software development and collaboration with community maintainers.
• Familiarity with software engineering methodologies such as Agile, Scrum, or Kanban.
• BS, MS, or PhD in Computer Science, Computer Engineering, or a related field with several years of relevant experience.

Nice To Haves

• Experience in Python development.
• Experience 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.
• Strong communication skills for project status updates to leadership.

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 limits of innovation in AI software solutions.
• Engage with clients to understand their needs and provide tailored software solutions.
• Work with the latest applications, libraries, frameworks, and SDKs to enhance software performance.

Benefits

• Base pay competitive with industry standards.
• Annual bonus or sales incentive eligibility.
• Employee Stock Purchase Plan with discounts on AMD stock.
• Comprehensive benefits package including health insurance and retirement plans.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Bachelor's degree