GPU Kernel Developer - AI Models

As a GPU Kernel Developer specializing in AI Models at AMD, you will play a crucial role in developing high-performance GPU kernels for cutting-edge and future GPU hardware. Your expertise in GPU kernel development and optimization for AI/HPC applications will be essential in accelerating next-generation computing experiences. Working alongside a team of industry specialists, you will have the opportunity to work with the latest hardware and software technology. **Key Responsibilities:** - Develop high-performance GPU kernels for key AI operators on AMD GPUs - Optimize GPU code using a structured and disciplined methodology, including profiling, roofline-analysis, identification of optimizations, and prototype development - Support mission-critical workloads in NLP/LLM, Recommendation, Vision, and Audio - Collaborate with system-level performance architects, GPU hardware specialists, and various other teams to analyze and optimize training and inference for AI - Collaborate with open-source framework maintainers to integrate code changes upstream - Debug, maintain, and optimize GPU kernels, focusing on AI operator performance improvement - Apply software engineering best practices to your work **Qualifications Required:** - Knowledge of GPU computing technologies such as HIP, CUDA, OpenCL, Triton - Experience in optimizing GPU kernels - Proficiency in using profiling and debugging tools - Strong understanding of GPU hardware - Excellent programming skills in C/C++/Python, including debugging, performance analysis, and test design In addition to your technical skills, a Master's or PhD in Computer Science, Computer Engineering, or a related field is preferred. Join AMD in advancing the boundaries of innovation and making a significant impact on the world with your GPU kernel development expertise.,