返回岗位列表

壁仞科技

软件库性能优化工程师/架构师

地点:杭州 薪资:30-50K 日期:2026-03-03

岗位摘要

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

岗位职责

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

任职要求

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

工作地址

杭州滨江区海威天地T3-28