职位详情
深度学习异构计算研发工程师 ( ai 推理引擎方向)
25000-50000元
应届毕业生 本科
职位描述
工作职责: 1、负责研发 ai 推理框架,适配异构计算芯片,支持抖音、剪映等产品中的 ai 算法落地,支撑字节ai方向重要业务的发展; 2、负责优化cv/audio/aigc等深度学习算法模型在异构加速芯片上的工程性能和精度,同时提升 ai 模型部署工程效率; 3、负责探索和落地 ai 模型工程部署前沿技术,包括但不限于稀疏计算、深度学习编译优化。 任职要求: 1、本科及以上学历,计算机/电子/信息/通信/自动化/软件等相关专业,有ai工程优化经验的优先; 2、精通c/c++,精通算法与数据结构,熟悉python; 3、深入理解计算机体系结构,熟悉gpu硬件架构,熟练掌握gpu并行计算优化软件栈cuda/cudnn,有华为昇腾卡等优化经验的优先; 4、熟练掌握 ai 编译优化技术,例如tvm/mlir/iree/triton等; 5、了解深度学习算法基本原理,熟悉神经网络基本架构和各算子计算方式,了解至少一种深度学习训练框架; 6、熟悉常用图像视觉计算库及深度学习推理计算库,例如tensorrt等,熟悉常用的深度学习算子优化方法; 7、了解主流 aigc 算法模型原理,有aigc模型加速优化经验者优先。
联系方式
2175人关注 收藏职位 投递简历