职位描述 1、面向国内外多种 gpu 进行高性能算子开发与优化. 2、面向在不同 gpu 间移植高性能算子的需求,参与编译技术开发

职位要求 1、计算机科学或相关领域硕士研究生及以上学历 2、熟悉 python、c++、c 等编程语言 3、具备编译优化经验或 gpu 等加速卡的算子优化经验 4、熟悉深度学习、熟悉 pytorch 等深度学习框架者优先 5、具备矩阵乘、flash attention 等计算密集型算子优化经验者优先 6、熟悉 triton、tvm、mlir 等深度学习专用编译器或编译器组件等的优先 7、有良好的沟通和协调能力,具备团队合作精神,能够承受⼯作压力,对技术有强烈的热情和求知欲