外企

AI 优化工程师

超威半导体(AMD)

上海人工智能社会招聘

岗位描述

岗位描述:负责开发并推动高效软件在复杂新技术及新产品导入项目中的落地执行;为关键客户项目提供快速且稳定的技术支持;开发优化功能并在交付客户前完成验证;参与高效能特性团队协作;与多个团队紧密合作,交付关键规划方案及其支撑技术;参与未来高可扩展、高可靠、创新系统架构的设计与实现;与开发团队及项目经理紧密协作推动结果达成;能够使用 AI Agent 工具辅助测试与分析工作,提升日常工作效率。该岗位聚焦关键应用与基准测试的性能优化,并使用前沿软硬件技术开展工作。 岗位要求:具备扎实的软件工程基础和良好的沟通协作能力,能够推动复杂问题解决;精通并具备 C、C++、Python 实战经验;熟悉 flashInfer、TensorRT-LLM、CUTLASS、Aiter(AMD)等 AI 算子库,具备相关算子及算子融合的使用和定制经验;熟悉 Triton 编译器后端,能够基于 llir/ASM 显式布局进行流水线优化;具备 CUDA/Triton/Gluon kernel 实现经验,并有能力在计算密集型 kernel 上达到至少 50% MFU;具备扎实的 roofline 性能分析能力,熟悉 Nsight、ROCprofiler 等性能分析工具;熟悉 AMD CNDA/RDNA SIMD 架构,有基于 AMD 平台进行 kernel 开发经验者优先;本科或硕士学历,计算机科学、计算机工程、电子工程或相关等效专业。