算子工程师

职位描述

1、面向GPU/昇腾架构开展算子性能画像、热点定位与优化，持续提升吞吐与时延稳定性。

2、结合训练与推理场景，协同框架与编译链路进行内核调优、算子融合、内存访问与并行策略优化。

3、建设性能基线与度量方法，沉淀可复用的优化范式与最佳实践，推动持续优化闭环。

任职要求

1、理解计算图、并行化、向量化、内存层次与缓存友好性等关键要点；

2、针对目标硬件做深度调优：Occupancy、Memory Coalescing、Shared Memory等；

3、建立算子性能基线（Baseline）与回归体系，持续追踪吞吐、延迟、显存占用；

4、能独立完成性能分析、问题定位与优化落地，结果导向，热情好学，善于沟通协作与知识沉淀；

5、熟悉主流的性能分析工具，熟悉性能优化基本流程和步骤；

6、有较强的学习能力。

加分项：

1、在两种及以上平台完成过端到端算子优化经验；

2、有 Attention、GEMM、MoE 等核心算子的手搓或深度优化经验。