-岗位描述:
开发将Tensorflow / Pytorch / MxNet 等 AI编程框架模型布署至 AI加速计算平台的模型编译器和运行库,具体工作职责包括:
1-开发转化/融合/拆解算子/生成指令以及分配资源的AI模型编译器;
2-利用通用运算平台(CPU/GPU)开发算子/模型的运行库,模拟其在AI加速平台上的精度和性能;
3-制定测试计划和测试框架,以验证软件工具链的正确性和有效性;
-任职条件:
1-计算机科学或相关领域的本科/硕士/博士学位;
2-8至10年以上C / C ++ 编程经验,在AI模型开发框架(如Tensorflow / PyTorch / MxNet)方面有1年以上的经验;
3-了解常见AI模型算子的概念,具备类似 TensorRT / TVM / TF-Lite / MNN / NCNN / ARM-NN 等 AI 推理框架的经验者佳;
4-了解 MLIR / TVM/XLA/Glow之类的 AI 编译器;
-以下一项或者多项知识是-以下一项或者多项知识是加分项:
1-具有优化数值计算库(例如CUDNN / MKL)经验者为佳;
2-熟悉视频编解码器/DSP/PCIe协定/嵌入式开发/或CPU/GPU 及 AI ASIC架构者佳;
简历投递:HR@storagex.ai