AML机器学习编译优化工程师-上海
字节跳动
岗位职责
1、参与AI编译器相关项目的技术方案规划,设计在离线全系统的计算图优化链路;
2、针对搜索/推荐/广告场景,优化模型训练/推理的计算图执行效率;
3、与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。
任职要求
1、熟悉开源ML编译器代码(如XLA/MLIR/TVM),并有相关开发经验;
2、具有 CPU/GPU 下的算子Kernel 开发和性能优化经验;
3、具有扎实的C/C++编程能力和数据结构和算法知识;
4、具有独立解决问题的能力,能够对业务逻辑进行合理的抽象和拆分,良好的团队合作精神。
加分项:
1、有软硬件联合设计的经验;
2、深入研究过至少一种机器学习框架(Tensorflow / PyTorch / MxNet 或其他自研框架)的底层架构和机制。