返回岗位列表

AI高性能计算工程师

查看原岗位

地点：上海薪资：31-51K 日期：2026-03-03

岗位摘要

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

岗位职责

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

任职要求

计算机或电子信息类专业本科及以上学历
具备CUDA、OpenCL等高性能并行编程能力
熟悉AI模型的算子开发和优化
熟悉AI模型推理或训练场景的端到端优化
熟悉深度学习算法和主流AI模型，如ResNet50、BERT、GPT、LLaMA等
对计算机体系结构有深刻理解
精通Nsight Compute等性能分析工具
精通PyTorch、PaddlePaddle等至少一种深度学习框架
熟悉常用benchmark工具，如Sysbench、Iometer、Netperf
熟悉常用性能分析工具，如SystemTap、perf、VTune、gperftools、OProfile、火焰图