返回岗位列表

面壁智能

AI算子性能优化工程师

地点:北京 薪资:薪资未公开 日期:2026-03-03

岗位摘要

面向GPU/昇腾架构开展算子性能画像、热点定位与优化,持续提升吞吐与时延稳定性;结合训练与推理场景,协同框架与编译链路进行内核调优、算子融合、内存访问与并行策略优化;建设性能基线与度量方法,沉淀可复用的优化范式与最佳实践,推动持续优化闭环

岗位职责

  • 面向GPU/昇腾架构开展算子性能画像、热点定位与优化,持续提升吞吐与时延稳定性
  • 结合训练与推理场景,协同框架与编译链路进行内核调优、算子融合、内存访问与并行策略优化
  • 建设性能基线与度量方法,沉淀可复用的优化范式与最佳实践,推动持续优化闭环

任职要求

  • 理解计算图、并行化、向量化、内存层次与缓存友好性等关键要点
  • 熟悉主流AI算法与框架(如Transformers、PyTorch、TensorFlow等)
  • 具备GPU或NPU底层优化经验,熟悉CUDA或CANN编程模型
  • 掌握性能分析工具(如Nsight、Profiler等)的使用方法
  • 具有扎实的C++/Python编程能力和算法基础