AI编译器工程师/AI平台软件工程师(前端)

-岗位描述:

开发将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架构者佳;






简历投递通道

Email Adress:HR@storagex.ai


Data Centric Computing