岗位职责

1、针对特定硬件（GPU/DSA)进行推理和训练场景下算子开发和性能优化，包括DSL扩展，硬件无关优化，流水排布

2、分析模型性能瓶颈，找出瓶颈算子并进行高性能计算加速和优化，包括计算通信融合，自动融合算子生成

3、承担国产AI编译器开发，与国产厂商合作支持多后端(GPU/DSA), 并进行相关Pass开发"

岗位要求

岗位要求：

1、熟悉cuda/华为昇腾AscendC/triton编程语言；

2、熟悉C++/Python，编程语言基础扎实；

3、熟悉计算机体系结构及操作系统原理；

4、了解深度学习算法，对大模型推理和训练框架vllm, sglang, megatron的应用或开发经验;

5、优秀的分析问题和解决问题的能力。

具有以下条件者优先：

1、计算机领域相关的编程大赛获奖者优先；

2、熟悉AI领域常见框架、模型，有过实战调优经验者优先；

3、熟悉大模型算法和优化方法者优先；

4、熟悉MLIR, TVM, XLA, LLVM优先。

5、在国产芯片厂商(寒武纪，璧仞，燧原，华为，沐曦，天数)性能优化经验优先

理工科专业优先