AI NPU软件开发专家--上海

本平台专注IC&AI&机器人&汽车行业职位招聘，擅长，数字设计，芯片架构，ESL，Validation，芯片验证，芯片后端PR， DFT，STA，版图，模拟设计（ADC，IO，serdes，PLL，Power，DC DC），算子，框架，推理架构，嵌入式开发，内核开发，编译器，工具链，产品市场，销售，FAE，AE，FAB 设备工艺等相关职位，欢迎关注，咨询。

微信号：zyl889988 电话：13120709386

AI NPU软件开发专家--上海

职位描述

1.负责NPU核心算子的汇编级或SIMD指令优化，针对大模型各类线性算子，进行分块参数调优与指令级优化。

2.负责计算图的轻量级优化，包括算子融合、内存复用策略设计等。

3.负责设计Host与NPU之间的异步流水线编排，编排数据搬运与矩阵计算的重叠，并和相关团队协作推动底层接口与编排策略的持续优化。

4.参与开发NPU核心的轻量级驱动模块，协助实现DMA描述符配置及寄存器地址映射，确保主机端与NPU的数据通路畅通。

5.基于定制的多NPU固定流水线，参与开发硬编码的调度执行器，管理各NPU的相关计算任务并处理跨NPU的数据流同步。

6.配合离线编译器，在固件中实现静态内存分配表，确保权重和中间层数据在Local Memory间无冲突流转。

职位要求

1.精通C/C++，具备扎实的底层性能调优经验，熟悉汇编级优化技巧。

2.熟悉大模型推理中各类算子的计算特点，能够根据算子形状和硬件资源选择最优策略。

3.熟悉常见性能分析工具，能定位访存带宽、计算吞吐、指令开销等层面的性能瓶颈。

4.熟悉寄存器级编程，能独立阅读NPU/DSP微架构手册，具备配置DMA传输的能力。

5.熟练使用AI辅助编程工具。

加分项：

1.有AI编译器（如TVM/MLIR）算子生成或内存复用优化经验。

2.具备裸机或RTOS环境下的嵌入式开发经验，熟悉ARM/RISC-V汇编。