ZTE logo
中兴通讯
AI编译器开发资深专家

AI编译器开发资深专家

发布于 大约 3 小时前

普通员工/个人贡献者

上海市
专家级经验
全职员工
仅现场办公
硕士
软件工程
Cuda
Gpu
Llvm
Mlir
Npu
Openxla
Pytorch
Rocm
Tensorflow

AI 估算 · 45k–70k

AI编译器是稀缺方向,资深专家在沪薪资高企,大厂平台有溢价空间。

职位详情

关于这个职位

这是一个负责AI编译器底层研发的资深专家岗位,主要基于MLIR、TVM、OpenXLA等框架,为GPU、NPU等硬件做模型编译优化和部署

工作内容涉及计算图优化、IR转换、代码生成等核心模块,需要深厚的编译原理功底和软硬件协同优化能力
适合对编译技术有热情、追求技术深度的人

最低要求

计算机、电子工程、人工智能等相关专业硕士及以上学历,8年以上工作经验

熟悉至少一种主流AI编译框架(如 OpenXLA、TVM、torch.compile、MLIR 等)
有至少基于一种GPU、NPU后端架构的端到端编译器方案设计、工具链流程开发经验,多模型适配和优化经验,AI编译器领域从业经验5-8年以上
扎实的编译原理基础,熟悉 LLVM IR、MLIR Dialects、Pass机制、CodeGen流程等
熟悉 AI框架(TensorFlow / PyTorch / ONNX Runtime) 的模型执行流程
熟悉 CUDA / OpenCL / ROCm / NPU指令集 / 硬件调度机制中的一种或多种
精通 C++ / Python 编程,具备良好的代码结构与调试能力
具备良好的跨团队沟通能力与技术文档编写能力

工作职责

负责将 OpenXLA、TVM、MLIR 等 AI 编译器框架适配至多种GPU、NPU后端芯片架构,实现端到端模型编译、优化与部署

参与算子(Operator)定义、调度(Scheduling)、融合(Fusion)、内存规划(Memory Planning) 等核心模块的设计与实现
负责模型编译的计算图优化(Graph Optimization)、中间表示(IR)转换、代码生成(Codegen)等模块的设计与实现,以提升模型在GPU、NPU上的性能与稳定性
参与建立AI模型性能分析工具链,包括算子性能评估、编译日志可视化、profiling工具等
与硬件架构团队、AI算法团队深度协作,分析模型特性与硬件瓶颈,持续改进编译优化策略
跟踪 LLVM、MLIR、OpenXLA、TVM、TorchInductor等社区动态,探索AI编译器技术前沿并落地到产品中

AI 洞察

优缺点分析

优点

  • AI 编译器是当前极稀缺的技术方向,积累的经验具有高价值和长期市场竞争力
  • 工作涉及软硬件协同优化,能极大提升系统级技术视野和问题解决能力
  • 中兴通讯作为大平台,提供丰富的硬件资源和场景,有助于技术落地和深度探索
  • 技术难度高,需要同时理解编译器、AI 框架和硬件架构,学习曲线陡峭
  • 领域变化快,需持续跟进社区最新动态,对自我驱动要求高
  • 适合对编译技术和底层系统有浓厚兴趣,喜欢攻坚克难、乐于深耕技术细节的工程师

缺点 / 挑战

  • 跨团队协作多,沟通成本和技术权衡复杂度可能较高

角色解读

  • 可以成长为编译器领域的资深专家,主导大型编译优化项目
  • 有潜力转向 AI 基础设施架构师,负责整体软硬件栈设计
  • 也可拓展为 AI 框架优化负责人,结合前沿技术推动产品落地
  • 你将负责将 MLIR、TVM、OpenXLA 等主流 AI 编译框架适配到不同的 GPU、NPU 上,完成模型从训练到部署的编译优化链路
  • 深入参与算子定义、调度、融合、内存规划等核心模块开发,解决软硬件协同中的性能瓶颈
  • 与硬件架构和算法团队紧密合作,通过 profiling 和分析工具不断迭代编译策略,提升模型运行效率
  • 扎实的编译原理功底,熟悉 LLVM IR、MLIR Dialects、Pass 机制和代码生成流程
  • 熟练掌握至少一种主流 AI 编译框架,并有端到端工具链开发和优化经验
  • 精通 C++/Python,对硬件架构(GPU/NPU)有深入理解,熟悉 CUDA、ROCm 或 NPU 指令集

申请策略

  • 了解中兴通讯在 AI 基础设施方面的布局,结合公司业务方向展示你的价值
  • 面试前准备好个人技术博客或项目总结,体现对技术的热情和深度
  • 突出端到端编译器项目的经历,例如你如何设计并实现了基于某框架的编译流程
  • 量化性能优化成果,如编译时间缩短、模型推理速度提升的具体数字
  • 强调对 MLIR/TVM 等框架的源码级理解,以及独立实现的 pass 或 CodeGen 模块
  • 展示跨团队协作案例,说明你如何与硬件或算法团队共同推进问题解决
  • 若有时间,可深入阅读 MLIR 官方文档和源码,理解 Dialect 和 Pass 机制
  • 学习 GPU 架构基础(如 CUDA 内存模型、调度机制),有助于面试中展现系统思维

面试指南

  • 先从原理入手,说明技术概念和设计思路,再结合自身项目经验给出具体实践案例
  • 采用问题拆解的方式,分步骤阐述从分析、设计到实施和验证的过程
  • 突出评估权衡,说明在不同约束下如何做出技术选型
  • 请解释一下 MLIR 中 Dialect 和 Pass 的作用,以及它们如何协作?
  • 你如何设计一个编译器 pass 来优化算子融合?请结合具体案例
  • 遇到模型在 GPU 上性能不佳时,你会怎么定位和分析瓶颈?
  • 如何将一个新的硬件后端集成到 TVM 或 MLIR 中?需要哪些步骤?
  • 你如何看待 AI 编译器与硬件协同发展的趋势?

职位点评

67
综合评分

AI编译器前沿岗位,技术成长性极强,薪资有竞争力但WLB未知。

从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。

更适合这类人
最适合追求前沿技术深度、对编译优化有强烈兴趣,且不太在意工作地点灵活度的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利65
成长发展85
工作生活45
使命价值70

薪资福利

65中等

该职位未披露具体薪资,但资深专家级别通常具有较高薪酬水平,且大公司提供稳定的福利保障。

薪资信号未披露(AI估算:45K-70K/月)

成长发展

85较高

AI编译器处于技术前沿,涉及的MLIR、TVM等技术极具挑战性和成长性,能持续拓展技术深度。

技术前沿前沿/新兴技术
技术栈MLIR、TVM、OpenXLA、LLVM、GPU、NPU、CUDA、ROCm
业务类型ambiguous

工作生活

45较低

工作地点固定在上海,但JD未提及弹性工作或远程选项,生活工作平衡情况不明确。

工作模式未明确
办公地点未明确
加班情况未提及(无法判断)

使命价值

70中等

AI编译器推动AI基础设施发展,技术价值高,但中兴作为传统通信厂商,使命感一般。

行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
Watch Jobs