返回岗位列表

壁仞科技

AI模型性能优化工程师

地点:上海 薪资:薪资未公开 日期:2026-01-29

岗位摘要

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

岗位职责

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

任职要求

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