← Back to opportunities
About the Role
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 ...
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 ...
Ready to Join Through a Referral?
Apply now and get connected directly with the hiring team
Apply for this Position