职位描述
1. 负责开发面向自研芯片的AI模型编译工具链。
2. 设计并实现模型优化、图编译、算子融合、内存优化等关键技术。
3. 与芯片团队紧密合作,针对硬件特性(如指令集、内存架构)进行深度优化。
4. 负责工具链的文档建设和开发者体验优化。任职要求
- 精通Python和C++,具备高性能代码开发和调试能力。
- 熟悉AI模型部署全流程,包括模型量化、编译优化等。
- 深入理解CNN/Transformer等网络结构及其计算特性。
- 熟悉至少一种主流深度学习框架(PyTorch/TensorFlow)。
- 有MLIR开发经验。
加分项:
- 有AI算子开发经验,熟悉芯片特定指令集优化。
- 参与过完整AI工具链开发项目,产出过成熟部署工具。
- 了解自研芯片架构(如NPU),能针对硬件特性优化工具链。