【27届留用实习生】-AI编译器工程师和算子优化-系统平台中心

1、针对特定硬件（GPU/DSA)进行推理和训练场景下算子开发和性能优化，包括DSL扩展，硬件无关优化，流水排布。 2、分析模型性能瓶颈，找出瓶颈算子并进行高性能计算加速和优化，包括计算通信融合，自动融合算子生成。 3、承担国产AI编译器开发，与国产厂商合作支持多后端(GPU/DSA), 并进行相关Pass开发。

1、本科及以上学历，熟悉CUDA/华为昇腾AscendC/triton编程语言。 2、熟悉C++/Python，编程语言基础扎实，熟悉计算机体系结构及操作系统原理。 3、了解深度学习算法，对大模型推理和训练框架vllm, sglang, megatron的应用或开发经验。 4、具备优秀的分析问题和解决问题的能力，至少实习4个月及以上。 加分项（具备以下任一经验者优先）： 1、计算机领域相关的编程大赛获奖； 2、熟悉AI领域常见框架、模型，有过实战调优经验； 3、熟悉大模型算法和优化方法、熟悉MLIR, TVM, XLA, LLVM； 4、在国产芯片厂商(寒武纪，璧仞，燧原，华为，沐曦，天数)性能优化经验.