职位描述
1、基于AI芯片的基础编译器开发和优化Triton编程模型
2、通过编译器优化方法提升指定模型的性能
3、追踪各类AI编译器技术并实现原型对接
4、其他新型编译器/编程模型的技术生态对接任职要求
1、3年以上工作经验,良好的C/C++编程功底,扎实的数据结构与算法基础
2、熟悉计算机体系结构、芯片架构、指令集、编译器性能优化基础技术
3、熟悉NPU/GPU/DSP等高性能计算或异构加速芯片硬件
4、有LLVM/MLIR/GCC等基础编译器框架开发经验。
5、良好的团队沟通能力与问题解决能力
6、满足如下条件者优先:
a)、有Triton或类CUDA-C/TVM/CUTLASS/XLA等AI编译技术开发经验
b)、有丰富的NPU/GPU/DSP性能优化经验,对某优化策略有深入研究
c)、有编译器后端各类Pass设计与开发者经验优先考虑