异构计算性能优化工程师
Company Kuaishou
Focus Tech · Backend
Location 北京
Published May 19, 2026
岗位职责
1、硬件架构选型与设计:根据公司AI业务场景(大模型训练、AIGC、推荐系统等),主导GPU/AI加速器的架构评估与选型决策,制定符合业务需求的解决方案; 2、性能评估与优化:深入研究AI算法(深度学习、CV、NLP、多模态等)的计算特征,建立GPU/异构硬件的性能评测体系,输出优化建议并推动落地,提升训练与推理效率; 3、技术趋势洞察:跟踪多元GPU/AI芯片的技术演进,结合公司业务战略,规划硬件技术路线图; 4、软硬件协同优化:与算法团队、系统团队深度协作,开展算子优化、显存优化、通信优化等工作,打通从算法需求到硬件性能的全链路; 5、技术方案落地:撰写技术方案文档,推动硬件选型决策的标准化与规模化复制,支持AI业务快速迭代。
任职要求
1、计算机、电子工程、人工智能等相关专业本科及以上学历,硕士或博士优先; 2、扎实的AI理论基础:深入理解深度学习框架(TensorFlow/PyTorch)及主流模型架构; 3、硬件性能调优经验:熟悉GPU体系架构,具备CUDA/ROCm编程或性能profiling实战经验; 4、系统级视野:了解AI训练/推理的全栈性能瓶颈,能够从算法、框架、硬件多维度分析问题。