AI加速软件工程师
职位描述
职位介绍
岗位职责:
开发将 Tensorflow / Pytorch / MxNet 等 AI编程框架模型布署至 AI加速计算平台的模型编译器和运行库,一些工作职责包括:
1.开发转化、融合、拆解算子、生成指令、以及分配资源的AI模型编译器;
2.了解计算平台的指令集和体系结构,开发平台上的算子、并设计内存分配、执行顺序和分区的优化策略;
3.利用通用运算平台 (CPU/GPU) 开发算子/模型的运行库,模拟其在AI加速平台上的精度和性能;
4.制定测试计划和测试框架,以验证软件工具链的正确性和有效性。
任职要求:
1.计算机科学或相关领域的本科/硕士/博士学位或同等学历;
2.熟练使用C / C ++ 编程。在AI模型开发框架(如Tensorflow / PyTorch / MxNet)方面有经验;
3.了解常见AI模型算子的概念。具备类似 TensorRT / TVM / TF-Lite / MNN / NCNN / ARM-NN 等 AI 推理框架的经验者为佳;
4.具备 TVM / XLA / Glow / NGraph之类的 AI 编译器方面的经验者为佳;
5.具有优化数值计算库(例如CUDNN / MKL)经验者为佳;
6.熟悉视频编解码器、DSP、PCIe协定、嵌入式开发、或CPU,GPU 及 AI ASIC架构者为佳;
7.较强的沟通能力,领导能力和职业道德。