返回岗位列表

软件库性能优化工程师

查看原岗位

地点：上海薪资：40-70K 日期：2026-03-03

岗位摘要

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

岗位职责

负责AI模型算子开发与性能优化
完成推理或训练模型的端到端性能调优
与软硬件架构师共同探索软件流程及系统架构
驱动并解决跨团队性能问题

任职要求

计算机或电子信息类专业本科及以上学历
具备CUDA、OpenCL等高性能并行编程能力
熟悉AI模型算子开发与优化
熟悉AI模型推理或训练场景端到端优化
熟悉深度学习算法及主流AI模型如RN50、BERT、GPT、LLaMA
深刻理解计算机体系结构
精通Nsight Compute等性能分析工具
精通PyTorch/PaddlePaddle等深度学习框架
熟悉Sysbench、iometer、netperf等benchmark工具
熟悉SystemTap、perf、vtune、gprof、oprofile、火焰图等性能分析工具

福利待遇

15薪高薪激励
参与前沿大模型与GPU集群优化项目
与顶尖架构师协同成长
上海核心科技园区办公

工作地址

上海闵行区壁仞科技12层