Lead Forward Deployed Software Engineer

Advanced Micro Devices • Santa Clara, CA

84d • Hybrid

About The Position

As a Forward Deployed Software Engineer, you will work closely with our most strategic partners as a hands-on technical expert. You are responsible for turning AMD's cutting-edge AI technology into tangible business value. This role is a unique blend of customer relationship skills and elite software engineer; you will work side-by-side with clients to help them prove out and ultimately deploy AI solutions on AMD GPUs. You will be the trusted technical advisor and hands-on developer who makes it happen.

Requirements

• Strong programming skills in C/C++ and Python.
• Experience with GPU kernel programming using CUDA, HIP or OpenCL.
• Experience with collective communication and attentions.
• Strong performance analysis and optimization skills for both CPU and GPU.
• Expertise with modern AI/ML frameworks (e.g., PyTorch, TensorFlow, JAX).
• Experience with distributed training and inference frameworks.
• A track record of direct Client Engagement with experience working with customers to solve ambiguous technical problems.
• Experience with containerization and orchestration technologies like Singularity, Docker, and/or Kubernetes.
• Experience with open-source software development including collaboration with community maintainers and submitting contributions.
• Experience with software engineering methodologies such as Agile, Scrum, Kanban.
• Excellent analytical and problem-solving skills.
• Ability to work independently and as part of a team.
• BS/MS/PhD with related experience in Computer Science or related equivalent.

Nice To Haves

• Experience in compiler, ISA.
• Experience shipping software in an end-customer production environment.
• Experience in implementing and optimizing collective communication primitives on GPU accelerators (NCCL/RCCL).
• Experience in implementing and optimizing attention and KV-cache primitives on GPU accelerators.
• Experience in all phases of software development, from requirement gathering, analysis, design, development, testing to final release.
• Experience with test-driven development (TDD), CI/CD.
• Experience providing clear and timely communication related to status and other key aspects of the project to the executive leadership team.

Responsibilities

• Work closely with strategic customers to understand their requirements challenges and identify opportunities for AMD hardware and software to provide value.
• Close gaps in the AMD software stack needed to support customer solutions.
• Work hands-on as a technical expert and creative problem-solver, developing side-by-side with customers to drive projects from proof of concept to production.
• Act as the voice of the customer, translating their needs and your on-the-ground insights into actionable feedback that shapes the future of AMD's AI roadmap.
• Thrive in unfamiliar territory with a high degree of autonomy, finding novel ways to apply AMD technology to real-world problems.

What This Job Offers

• Career Level: Senior
• Industry: Computer and Electronic Product Manufacturing
• Education Level: Master's degree
• Number of Employees: 5,001-10,000 employees