1. 负责公司自研AI芯片上编译器及工具链软件设计及研发;
2. AI编译器上的模型性能调优;
3. 与团队协作,参与新特性的设计和实施,提高编译器的功能和易用性;
4. 解决与编译器相关的技术挑战,包括模型编译优化、自动并行化和硬件加速等方面问题;
5. 积极跟踪AI编译器工具链生态的发展,探索新技术应用到实际工作中的可行性。
任职要求:
1. 计算机科学或相关领域的硕士及以上学位
2. 具备Linux C/C++、Python等编程语言的熟练编程能力;
3. 熟悉深度学习编译器相关工作,具备MLIR/LLVM/Triton等一个或者多个开源项目的实际开发经验;
4、良好的沟通、协作能力。
5. 满足如下条件者优先:
a)芯片工具链、虚拟机、模拟器、系统级高性能计算库的开发;
b)深度学习执行引擎的开发;
c)熟悉通用/特定硬件的深度学习计算图优化方法;
d)了解计算机体系结构及相关量化分析方法;
e)熟悉深度学习编译框架(如XLA,IREE等);
f)了解常见深度学习模型和算法;
g)熟悉异构编程环境的相关研发(CUDA,OpenCL等);
h)熟悉LLVM/GCC/MLIR/TVM等编译优化工具研发