返回岗位列表

壁仞科技

软件库性能优化工程师

地点:上海 薪资:30-60K 日期: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薪高绩效回报
  • 参与前沿AI大模型与芯片架构创新
  • 与顶尖软硬件团队深度合作
  • 上海浦江科技广场优质办公环境

工作地址

上海闵行区浦江科技广场16号楼13楼