Lead Forward Deployed Software Engineer (2)

Advanced Micro Devices, Inc.•Santa Clara, CA

28d•Hybrid

About The Position

At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the world's most important challenges-striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career. THE GROUP: AI is defining the next era of computing, and this is just the beginning. We see the benefits of AI every day-enabling medical research, curbing credit card fraud, reducing congestion in cities, or simply making life easier. In the ever-evolving landscape of artificial intelligence, we are a powerhouse - a cutting-edge 'AI Forward Deployment Software Development Team'. Specialized in end-to-end AI optimization, fine-tuning large language models to unlock unprecedented Generative AI efficiency. Our expertise extends beyond the hardware realm, encompassing 3P enablement, where we develop custom AI Software Solutions for Industry leading AI customers. Are you excited to work with one of Top 1% of the AI companies in the world? THE ROLE: As a Forward Deployment 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, and you will spot existing AI software optimization opportunities and push the boundary with other software development teams. You will be the trusted technical advisor and hands-on developer who makes it happen. Do you love working with customers and bringing the most cutting-edge AI software from AMD to customers? Do you love using a microscope to dig into the performance of AI models end-to-end? Do you feel excited when you squeeze another 1% performance out of the hardware with software? All sound like you? We would love to hear from you!

Requirements

Strong programming skills in C/C++ and Python
Experience with GPU kernel programming using CUDA, HIP or OpenCL.
Proefficient on common ML performance analysis tools.
A track record of direct Client Engagement with experience working with customers to solve ambiguous technical problems.
Strong performance analysis and optimization skills for both CPU and GPU.
Experience with containerization and orchestration technologies like Singularity, Docker, and/or Kubernetes.
Expertise with modern AI/ML frameworks (e.g., PyTorch, TensorFlow, JAX).
Experience with distributed training and inference frameworks.
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 communication primitives on GPU accelerators (NCCL/RCCL, OpenMP, MPI)
Experience in all phases of software development, from requirement gathering, analysis, design, development, testing to final release.
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.