算子开发工程师

职位描述

1. 面向国内外多种 GPU 或其他 AI 算力进行高性能算子开发与优化。

2. 面向实际模型与实际硬件平台，进行算子与推理引擎的综合优化。

任职要求

1. 计算机科学或相关领域硕士研究生及以上学历。

2. 具有良好的学习能力和团队合作精神。

3. 掌握 Python、C++、C 等通用编程语言，以及 CUDA 或 AscendC 等算子编程语言。

4. 具备 GPU 等加速卡的算子优化或算子自动编译优化经验。

5. 具备量化矩阵乘法、Flash Attention、计算通信融合算子等复杂算子优化经验者优先。

6. 熟悉 PyTorch 框架、熟悉大模型完整推理流程者、熟悉算子与推理引擎联合优化者优先。

7. 熟悉国内外最新 AI 算力硬件体系结构及其优化方法者优先。