AI编译器工程师

职位描述

1. 负责开发面向自研芯片的AI模型编译工具链。

2. 设计并实现模型优化、图编译、算子融合、内存优化等关键技术。

3. 与芯片团队紧密合作，针对硬件特性（如指令集、内存架构）进行深度优化。

4. 负责工具链的文档建设和开发者体验优化。

任职要求

- 精通Python和C++，具备高性能代码开发和调试能力。

- 熟悉AI模型部署全流程，包括模型量化、编译优化等。

- 深入理解CNN/Transformer等网络结构及其计算特性。

- 熟悉至少一种主流深度学习框架（PyTorch/TensorFlow）。

- 有MLIR开发经验。

加分项：

- 有AI算子开发经验，熟悉芯片特定指令集优化。

- 参与过完整AI工具链开发项目，产出过成熟部署工具。

- 了解自研芯片架构（如NPU），能针对硬件特性优化工具链。