AI解决方案工程师/架构师
职位描述
职位介绍
职位描述:
开发将 Tensorflow / Pytorch / MxNet 等 AI编程框架模型布署至 AI加速计算平台的模型编译器和运行库,一些工作职责包括:
1. 开发转化、融合、拆解算子、生成指令、以及分配资源的AI模型编译器。
2. 利用通用运算平台 (CPU/GPU) 开发算子/模型的运行库,模拟其在AI加速平台上的精度和性能。
3. 制定测试计划和测试框架,以验证软件工具链的正确性和有效性。
任职要求:
1. 计算机科学或相关领域的本科/硕士/博士学位或同等学历。
2. 在AI模型开发框架(如Tensorflow / PyTorch / MxNet)方面有1年以上的经验。
3. 了解常见AI模型算子的概念。具备类似 TensorRT / TVM / TF-Lite / MNN / NCNN / ARM-NN 等 AI 推理框架的经验者为佳。
4. 具备 TVM / XLA / Glow / NGraph之类的 AI 编译器方面的经验者为佳。
5. 较强的沟通能力,领导能力和职业道德。