返回岗位列表

壁仞科技

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

地点:上海 薪资:50-55K 日期:2026-03-03

岗位摘要

主导AI模型算子级CUDA/OpenCL并行开发与性能优化;完成推理/训练端到端流程调优,提升系统吞吐与延迟指标;与软硬件架构师联合探索软件流程及架构创新方案;驱动跨团队定位并解决复杂性能瓶颈问题

岗位职责

  • 主导AI模型算子级CUDA/OpenCL并行开发与性能优化
  • 完成推理/训练端到端流程调优,提升系统吞吐与延迟指标
  • 与软硬件架构师联合探索软件流程及架构创新方案
  • 驱动跨团队定位并解决复杂性能瓶颈问题

任职要求

  • 计算机/电子信息类硕士及以上学历
  • 精通CUDA、OpenCL等高性能并行编程
  • 熟悉AI模型算子开发及主流模型结构(RN50/BERT/GPT/LLaMA)
  • 掌握Nsight Compute、perf、vtune等性能分析工具
  • 熟练使用PyTorch/PaddlePaddle等深度学习框架

福利待遇

  • 年薪50-55K,具备行业竞争力
  • 参与国产GPU前沿技术攻关,技术成长空间大
  • 上海浦江科技广场核心办公,交通便利
  • 与顶尖芯片及AI专家共事,跨团队学习机会丰富

工作地址

上海闵行区浦江科技广场8号楼上海市闵行区陈行公路2388号16幢13层1302室