AMDAI Model Training Development Engineer

工作描述

任职要求
Responsibilities Develop and optimize core training operators on AMD GPUs (GEMM, GroupedGEMM, Attention, DeepEP, etc.), continuously pursuing state-of-the-art performance. Conduct in-depth analysis of performance bottlenecks in large-scale model training and drive targeted end-to-end performance optimizations. Collaborate closely with AMD’s software and hardware teams to enhance the performance and stability of the ROCm ecosystem. Participate in cutting-edge technology research, including but not limited to next-generation GPU hardware, compute-communication operator fusion, and AGI-driven automatic generation of high-performance operat…