Advance your career at AMD as a Senior AI Software Engineer focusing on Triton kernel optimization. Your work will directly enhance the performance of AI models and systems.
In this role, you will be responsible for designing and optimizing Triton and Gluon kernels aimed at large-scale AI workloads. Collaborating with diverse teams, you will contribute to AMD's AI roadmap by ensuring the competitiveness and efficiency of its leading hardware solutions.
Key Responsibilities:
• Research and optimize transformer kernels with Triton
• Productionize critical AI kernels for vLLM and SGL
• Collaborate with compiler engineers on backend maintenance
• Optimize kernel performance in memory and execution layers
• Profile and debug performance issues end-to-end
Requirements:
• 3+ years in GPU programming or performance optimization
• Strong Triton expertise, including custom kernel writing
• Insight into modern GPU architectures and scheduling
• Contributions to...