AMDGPU Kernel Development Engineer

工作描述

任职要求
You are a strong GPU performance engineer with a solid understanding of algorithms, model architectures, and kernel implementations. You can move fluidly from mathematical concepts to low-level optimization, and you excel in diagnosing real training bottlenecks. You are comfortable working directly with customers and collaborating across internal teams. KEY RESPONSIBILITIES: Support Customers: Ensure smooth training on AMD GPUs by identifying bottlenecks and delivering kernel-level performance improvements. Optimize Hot Operators: Design and optimize kernels using HIP, CUDA, and Triton …