返回岗位列表

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

查看原岗位

地点：深圳薪资：30-50K 日期：2026-05-08

岗位摘要

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

岗位职责

优化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等深度学习编译技术

福利待遇

提供具有竞争力的薪酬：30-50K·15薪
硕士学历要求匹配优质岗位定位
工作地点位于深圳南山区田厦金牛广场，交通便利
寒武纪平台技术资源支持，职业发展空间广阔

工作地址

深圳南山区田厦金牛广场3406