职位描述

工作职责:
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模型加速优化经验者优先。
工作地点
地址:上海杨浦区上海-杨浦区中国大陆上海市上海市杨浦区民府路678号抖音新江湾广场t3号楼,邮编:200082


职位发布者
HR
北京字节跳动网络技术有限公司


-
互联网·电子商务
-
1000人以上
-
私营·民营企业
-
北京市海淀区知春路甲48号2号楼10A室