AI编译器开发工程师(A171052)

该职位来源于猎聘 岗位描述： 1.

基于自研GPGPU架构进行AI编译器开发，自动生成算子； 2.

基于Triton对各类Pass进行功能开发和优化设计； 3.

开发Triton DSL算子并进行Lower，满足CNN/LLM 网络对于性能的要求； 4.

基于性能分析工具，对CNN/LLM进行性能分析，并基于自研芯片架构进行优化，以匹配硬件特性并充分发挥硬件性能； 职位要求： 1．

具有3年以上工作经验，丰富的C++/Python编程经验和扎实的编程功底，擅长调试分析和解决问题； 2．

具有2年以上MLIR开发经验； 3．

熟悉GPU/NPU/DSP硬件架构，具有相关的性能优化经验； 4．

熟练使用 git, Makefile等开发工具； 5．

良好的自驱力、沟通能力以及团队合作能力； 加分项： a.

具有Triton开发经验； b.

熟悉基于Transformer的LLM，例如LLAMA，Qwen等； c.

具有深度学习低精度量化、剪枝等开发经验； d.

具有LLVM的开发经验