職位描述

工作職責:
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室