返回岗位列表

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

查看原岗位

地点：上海薪资：25-50K 日期：2026-03-03

岗位摘要

持续开发torch_mlu，集成PyTorch最新技术，保持torch_mlu技术竞争力和创新性；解决网络推理和训练精度、性能和分布式问题；协助分析和解决torch_mlu应用问题；较强的C++开发经验和一定Python开发能力，扎实的数据结构和算法基础

岗位职责

持续开发torch_mlu，集成PyTorch最新技术，保持torch_mlu技术竞争力和创新性
解决网络推理和训练精度、性能和分布式问题
协助分析和解决torch_mlu应用问题

任职要求

较强的C++开发经验和一定Python开发能力，扎实的数据结构和算法基础
精通一种主流框架开发：PyTorch/TensorFlow/vLLM等
解决过网络精度、性能问题
熟悉分布式训练的原理和调优
学习能力强，有强烈的求知欲和自我驱动能力，良好的沟通和团队协作能力
计算机及相关专业毕业
加分项1：参与过AI框架相关社区开发，提交过代码
加分项2：熟悉Torch Inductor、Triton、XLA、MLIR、TVM等
加分项3：有解决分布式训练问题的经验

加分项

1：参与过AI框架相关社区开发，提交过代码
2：熟悉torch inductor、Triton、XLA、MLIR、TVM等
3：解决分布式训练问题的经验
陶先生刚刚活跃

工作地址

上海浦东新区展想广场1805室