返回岗位列表

高性能计算工程师

查看原岗位

地点：北京薪资：30-60K 日期：2026-03-03

岗位摘要

负责人工智能模型的算子开发与优化工作；进行推理或者训练模型的端到端调优；与软硬件架构师合作进行软件流程及架构探索；驱动并解决跨团队的性能问题；计算机或电子信息类专业本科及以上学历；具备 CUDA、OpenCL 等高性能并行编程能力

岗位职责

负责人工智能模型的算子开发与优化工作
进行推理或者训练模型的端到端调优
与软硬件架构师合作进行软件流程及架构探索
驱动并解决跨团队的性能问题

任职要求

计算机或电子信息类专业本科及以上学历
具备 CUDA、OpenCL 等高性能并行编程能力
熟悉人工智能模型的算子开发和优化流程
熟悉人工智能模型推理或训练场景端到端优化
熟悉深度学习算法和主流模型如 BERT、GPT 等
对计算机体系结构有深刻理解
精通 Nsight Compute 等性能分析工具
精通 Pytorch 或 PaddlePaddle 等至少一种深度学习框架

福利待遇

薪资可进一步面议
合适的应届生也可以投递

工作地址

北京朝阳区浦项中心B座10层