AMD is seeking an influential engineer to optimize AI workloads running on AMD GPUs. This position offers a unique opportunity to be a founding member of a skilled team focused on shaping the future of AI computing solutions. The ideal candidate will thrive in a fast-paced, innovation-driven environment and will be responsible for driving performance at scale. The role emphasizes collaboration and a strong bias for action to solve complex problems.

• Experience with state-of-the-art LLM inference and training across both framework and kernel layers is highly desirable.
• Deep experience in AI infrastructure and open-source ecosystems such as vLLM, SGLang, JAX, XLA, PyTorch, and Triton.
• Strong kernel optimization expertise using DSLs and HIP or CUDA, including familiarity with PTX/SASS or equivalent.
• Solid understanding of modern GPU architectures.
• Proven track record of open-source contributions, such as on GitHub.
• Strong leadership presence with excellent communication and interpersonal skills.
• Bachelor's, Master's, or Ph.D. in Computer Engineering, Computer Science, Electrical Engineering, or a related technical field.

• Optimize AMD GPU performance for open-source LLM frameworks such as vLLM and SGLang.
• Analyze and resolve performance bottlenecks in AI workloads at the kernel level.
• Integrate AMD software stacks like ROCm and ATen into open-source frameworks including PyTorch, JAX, and Triton.
• Collaborate cross-functionally with software and hardware engineering teams to identify gaps and deliver performance improvements.
• Build strong technical relationships with internal teams and external partners.

• AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.