AMD is seeking an influential engineer to optimize AI workloads on AMO GPUs. This role involves being a founding member of a specialized team focused on defining the future of AI computing solutions. Candidates should have a strong motivation and a 'Just-Do-It' mindset, with hands-on experience in LLM inference and training. The position emphasizes collaboration and innovation in enhancing GPU performance.

• Bachelor's, Master's, or Ph.D. in Computer Engineering, Computer Science, Electrical Engineering, or a related technical field.
• Hands-on experience with state-of-the-art LLM inference and training at both the framework and kernel levels.
• Deep AI infrastructure experience with open-source frameworks such as SGLang, vLLM, Jax, XLA, Pytorch, and Triton.
• Strong kernel optimization skills using DSLs and HIP (or CUDA), plus PTX/SASS equivalents.
• Demonstrated open-source contributions on GitHub.

• Improve AMO GPU performance on open-source repositories for LLM workloads such as vLLM and SGLang.
• Co-optimize AI workloads on current AMO GPUs by analyzing the bottlenecks and mitigating them at the kernel level.
• Integrate AMO software stacks (RoCm, ATen) into open-source frameworks such as Pytorch, JAX, and Triton.
• Build strong technical relationships with peers and partners, and report learnings and gaps to GPU software and hardware engineers.

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