Principal Software Development Engineer, AI Open-Source Software

Advanced Micro Devices, Inc•Austin, TX

1d•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. 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.
This is a senior level role, you must have post college experience.

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.