软件工程师

职位描述

1.负责自研NPU配套的AI工具链设计和实现，针对NPU硬件特性实现高性能编译策略；

2.负责与NPU系统架构工程师及数字设计工程师深度协作，推动软硬件协同优化；

3.协同NPU Runtime库的架构设计与开发，实现软件对NPU的高效率调度；

4.参与AI模型的部署和调优，包括视觉模型及大语言模型等。

任职要求

1.具有AI芯片编译器的设计和开发项目经验，熟悉编译器开发的整体流程和关键细节。从零开始参与设计和开发自研NPU编译器的系列工作；

2.深入理解至少一种主流AI编译器框架，如TVM、MLIR、XLA等，具有NPU编译器项目开发及优化经验，具备LLVM后端开发、调试经验；

3.熟练掌握使用C/C++、Python进行编译器核心开发的能力，熟悉CUDA、OpenCL等异构计算编程模型；

4.熟悉NPU硬件架构和指令集架构，能结合SIMD、Tensor Core等硬件加速指令设计适配优化策略，能从编译角度提出硬件优化建议；

5.熟悉PyTorch、ONNX、TensorFlow等主流AI框架及其内部机制，熟悉模型从训练到NPU部署的流程。

6.工作态度积极主动，善于沟通协作。能够协同软硬件同事共同完成AI芯片编译器的开发与硬件优化。