AML机器学习编译优化工程师-上海

字节跳动

岗位职责

1、参与AI编译器相关项目的技术方案规划,设计在离线全系统的计算图优化链路; 2、针对搜索/推荐/广告场景,优化模型训练/推理的计算图执行效率; 3、与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。

任职要求

1、熟悉开源ML编译器代码(如XLA/MLIR/TVM),并有相关开发经验; 2、具有 CPU/GPU 下的算子Kernel 开发和性能优化经验; 3、具有扎实的C/C++编程能力和数据结构和算法知识; 4、具有独立解决问题的能力,能够对业务逻辑进行合理的抽象和拆分,良好的团队合作精神。 加分项: 1、有软硬件联合设计的经验; 2、深入研究过至少一种机器学习框架(Tensorflow / PyTorch / MxNet 或其他自研框架)的底层架构和机制。
岗位信息来源于公开招聘渠道,仅作信息聚合展示。收录于 2026/7/23岗位编号 bytedance_social-7319787987527797043
AML机器学习编译优化工程师-上海 - 字节跳动 | 好简历招聘岗位库