返回岗位列表

深度学习框架图编译工程师

查看原岗位

地点：北京薪资：30-60K 日期：2026-03-03

岗位摘要

优化Torch Compile模式下各类神经网络的性能；进行网络性能瓶颈分析；优化编译耗时；优化codegen性能；紧跟PyTorch社区最新技术进展；将前沿性能优化技术迁移到MLU平台；有较强的C++和Python功底

岗位职责

优化Torch Compile模式下各类神经网络的性能
进行网络性能瓶颈分析
优化编译耗时
优化codegen性能
紧跟PyTorch社区最新技术进展
将前沿性能优化技术迁移到MLU平台

任职要求

有较强的C++和Python功底
有扎实的数据结构和算法基础
学习能力强，有强烈的求知欲和自我驱动能力
加分项：熟悉PyTorch/TensorFlow等主流深度学习框架，熟悉框架内部的模块、机制
加分项：熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术

加分项

1：熟悉PyTorch/TensorFlow等主流深度学习框架，熟悉框架内部的模块、机制
2：熟悉torch inductor、Triton、XLA、MLIR、TVM等深度学习编译技术
芦先生刚刚活跃

工作地址

北京海淀区北航致真大厦D座