职位描述
工作职责
基于TVM框架,主导从Relay到自研AI芯片指令集的编译pass开发。
开发图优化Pass(图层优化、算子融合、内存分配策略、tiling切分等)
实现算子自动调度适配,提升算子算力和带宽利用率
根据自研芯片的特性进行算子性能瓶颈分析并实现相关优化改进任职要求
职位要求
熟悉常见深度学习算法,如CNN/Transformer
精通TVM架构及Relay/TIR底层机制
精通Python/C++ 双语言开发
理解AI芯片架构(如TPU/VPU/NPU的脉动阵列、片上存储 hierarchy)