返回岗位列表

壁仞科技

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

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

岗位摘要

负责AI模型算子开发及GPU CUDA/Triton级性能优化;完成推理或训练模型的端到端系统级调优;与软硬件架构师共同探索软件流程与架构方案;主导并解决跨团队性能瓶颈问题;计算机/电子信息类本科及以上学历

岗位职责

  • 负责AI模型算子开发及GPU CUDA/Triton级性能优化
  • 完成推理或训练模型的端到端系统级调优
  • 与软硬件架构师共同探索软件流程与架构方案
  • 主导并解决跨团队性能瓶颈问题

任职要求

  • 计算机/电子信息类本科及以上学历
  • 熟练掌握CUDA、OpenCL等高性能并行编程
  • 熟悉AI算子开发与优化实践
  • 具备AI模型推理或训练场景端到端优化经验
  • 熟悉RN50、BERT、GPT、LLaMA等主流深度学习模型
  • 深入理解计算机体系结构
  • 精通Nsight Compute等性能分析工具
  • 熟练使用PyTorch/PaddlePaddle等深度学习框架
  • 熟悉Sysbench、iometer、netperf等benchmark工具
  • 掌握SystemTap、perf、vtune、gprof、oprofile及火焰图等性能分析手段

福利待遇

  • 年薪15薪,30-50K月薪,极具竞争力
  • 参与国产高端GPU核心软件栈建设
  • 与顶尖芯片与算法团队深度合作
  • 上海闵行核心科技园区办公环境

工作地址

上海闵行区壁仞科技16号楼