软件库性能优化工程师/架构师
岗位摘要
负责 AI 模型的算子开发与优化工作；进行推理或者训练模型的端到端性能调优；与软硬件架构师合作探索软件流程及架构；驱动并解决跨团队的性能瓶颈问题；计算机或电子信息类专业本科及以上学历；具备 CUDA、OpenCL 等高性能并行编程能力
岗位职责
- 负责 AI 模型的算子开发与优化工作
- 进行推理或者训练模型的端到端性能调优
- 与软硬件架构师合作探索软件流程及架构
- 驱动并解决跨团队的性能瓶颈问题
任职要求
- 计算机或电子信息类专业本科及以上学历
- 具备 CUDA、OpenCL 等高性能并行编程能力
- 熟悉 AI 模型的算子开发和优化流程
- 熟悉 AI 模型推理或训练场景端到端优化
- 熟悉深度学习算法和主流 AI 模型，如 RN50、BERT、GPT 等
- 对计算机体系结构有深刻理解，精通性能分析工具
- 精通 Pytorch 或 PaddlePaddle 等至少一种深度学习框架
工作地址
杭州滨江区海威天地T3-28