高性能计算研发工程师-抖音(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工程化落地,端到端地优化与部署经验的优先。
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请 立即举报
字节跳动
文娱内容
D轮及以上
北京(总部)
查看其他 337 个职位