硬件加速算子实现研发工程师-AI工具链
字节跳动
北京校招字节跳动社会招聘(社招)
岗位职责
1、负责深度学习模型算子分析工作;
2、负责后端算子开发工作;
3、负责算子优化Auto-tuning工作;
4、负责算子以及模型编译精度验证工作。
任职要求
1、熟悉视觉/语音/NLP/推荐中一种以上的常用模型;
2、熟悉TensorFlow,PyTorch等深度学习框架;
3、对常见算子(ONNX等)的底层计算过程有深入了解和分析;
4、熟练掌握C/C++、Python语言等相关编程语言;
5、熟悉数据中心内深度学习相关业务,有模型硬件加速应用和精度评估经验优先;
6、熟悉处理器体系结构或者AI加速硬件的优先。