高性能计算研发工程师-抖音(AI推理引擎方向)
薪资面议
人工智能 上海 不限 1-3年
岗位关键词
岗位职责
职位描述
1、负责开发和优化字节公司级的AI推理框架,通过CPU/GPU/DSP/NPU的并行计算、架构设计、稀疏优化等多种高性能优化技术打造业界领先的高性能异构AI推理引擎;
2、负责将深度学习算法在各平台做端到端地优化与部署,落地到抖音、剪映等产品,支撑字节AI方向核心业务的发展;
3、负责端上训练、联邦学习等核心技术开发与优化;
4、负责编译优化技术方向的探索、开发与落地。
岗位要求
职位要求
1、熟练掌握深度学习编译原理以及相关编译优化技术,例如TVM/MLIR/IREE等;
2、熟练掌握CPU(ARM/x86),GPU(Adreno/Mali/PowerVR/Intel/Nvidia/Apple)、DSP(HVX)中的1个或多个平台的高性能计算优化技术,具备较好的计算机体系结构开发背景,熟悉ARM NEON指令及汇编优化,或者熟悉移动端GPU优化;
3、了解深度学习基本原理,熟悉神经网络基本架构和各layer计算方式,熟悉至少一种深度学习训练框架及其模型文件的解析,如tensorflow、pytorch;
4、熟悉常用图像视觉计算库及深度学习推理计算库,包括arm compute library、tensorflow lite、nnpack、ncnn、mnn、tnn等;
5、熟悉常用的卷积计算加速方法,深度学习量化优化实现。
加分项:
1、深入理解计算机体系结构优先;
2、有深度学习框架或主流推理引擎开发优化经验者优先;
3、有AI工程化落地,端到端地优化与部署经验的优先。
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请
立即举报