招聘信息

岗位职责：

1. 面向国内外多种 GPU 进行高性能算子开发与优化

2. 面向在不同 GPU 间移植高性能算子的需求，参与编译技术开发

任职要求：

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

2. 熟悉 Python、C++、C 等编程语言

3. 具备编译优化经验或 GPU 等加速卡的算子优化经验

4. 熟悉深度学习、熟悉 PyTorch 等深度学习框架者优先

5. 具备矩阵乘、flash attention 等计算密集型算子优化经验者优先

6. 熟悉 Triton、TVM、MLIR 等深度学习专用编译器或编译器组件等的优先

7. 有良好的沟通和协调能力，具备团队合作精神，能够承受工作压力，对技术有强烈的热情和求知欲

联系电话：400-010-8918