AgiBot logo
智元机器人
AI编译器开发工程师

AI编译器开发工程师

发布于 大约 16 小时前

普通员工/个人贡献者

北京市 / 上海市
高级经验
全职员工
仅现场办公
学历未注明
软件工程
Ai编译器
Pytorch
Tensorrt
Tvm
Vla
Vlm
Ast语法树

AI 估算 · 40k–70k

5年以上AI编译经验稀缺,B轮中大型企业一线城市,薪资竞争力强,月薪40k-70k合理。

职位详情

关于这个职位

该职位负责开发自研AI编译器,支持VLA/VLM等前沿模型的多芯片部署,持续优化编译器性能对齐业界SOTA

适合有5年以上AI编译经验、掌握TVM/TensorRT等框架的技术专家,能够独立交付算法并解决部署难题

最低要求

年以上AI编译经验,独立部署过vla,vlm类型模型

理解AI框架及常见的AI模型
熟练掌握python以及C/C++编程
对AI编译技术栈有深度理解,前端、优化、后端有基本认知
熟悉transformers,hugginface,pytorch等框架的使用并了解算法搭建过程
深度理解torch.compile, torch.trace, jax.jit过程原理,有能力进行AST语法树解析
熟练使用TVM,TensorRT等框架并有实际部署经验
了解算子优化与集成常用技术,可以使用算子注册加速模型

工作职责

全栈开发自研编译器,辅助算法业务落地

跟进算法交付,在规定时间内交付可部署的算法,并支持多芯片部署
持续优化编译器各层能力,保持编译效果对齐业界SOTA

AI 洞察

优缺点分析

优点

  • 公司B轮融资,发展前景好,可参与前沿AI模型部署
  • 多芯片部署经验有助于在AI基础设施领域建立壁垒
  • 多芯片适配工作繁琐,需快速跟进算法迭代
  • 技术栈深广,需要持续学习新框架和芯片特性
  • 适合有扎实编译基础、热爱底层优化、能独立解决复杂问题的AI系统工程师

缺点 / 挑战

  • 自研编译器项目技术挑战高,能深入编译器底层,积累稀缺经验
  • 年以上AI编译经验的要求较高,且需要独立交付能力强

角色解读

  • 可向编译器架构师或AI系统专家方向发展
  • 积累芯片适配经验后,可转向AI芯片设计或系统优化
  • 若涉足管理,可成为编译器团队负责人或技术总监
  • 设计并实现自研AI编译器,支持VLA/VLM等模型的推理与部署
  • 参与算法交付全流程,确保编译后的模型在多种芯片上高效运行
  • 持续优化编译器前/中/后端,提升编译质量和性能
  • 精通Python和C/C++编程,能进行AST解析和算子开发
  • 深入理解TVM、TensorRT等编译框架,具备实际部署经验
  • 熟悉PyTorch、Transformers等框架,理解模型结构与推理过程
  • 掌握torch.compile、jax.jit等动态编译原理

申请策略

  • 了解智元机器人的业务方向(可能是具身智能),在面试中体现对该场景编译器需求的思考
  • 突出展示5年以上AI编译相关项目的交付经验,特别是VLA/VLM模型部署
  • 详细描述使用TVM/TensorRT等框架的优化案例及性能提升数据
  • 列举掌握的编程语言和编译器工具链,如AST解析、算子注册等
  • 提前熟悉TVM/TensorRT的最新版本,了解MLIR等新趋势
  • 练习VLA/VLM模型的导出与优化流程
  • 复习编译原理基础,提升AST解析和Pass开发能力

面试指南

  • STAR法则:描述情境、任务、行动、结果,突出技术难点和量化成果
  • 从通用原理到具体实践:先讲清楚底层原理,再结合项目案例说明
  • 请描述一次使用TVM/TensorRT部署VLA模型的经历,遇到了哪些性能瓶颈?
  • 解释torch.compile和jax.jit的原理差异,并给出优化思路
  • 如何在前端进行AST语法树优化?举例说明
  • 如果要在新芯片上部署一个transformer模型,你的步骤是什么?
  • 在编译器后端,如何设计算子融合策略?
  • 准备几个完整的项目案例,包括编译器优化前后性能对比

职位点评

70
综合评分

技术前沿、成长性极强但薪资与WLB信息不足的AI编译器岗位。

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

更适合这类人
适合追求技术深度与前沿发展、能接受一定工作不确定性的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利65
成长发展90
工作生活50
使命价值70

薪资福利

65中等

薪资面议,但作为B轮公司核心技术岗,薪资应有竞争力,但福利未明确提及。

薪资信号面议 (40K-70K/月)

成长发展

90较高

自研编译器项目技术前沿,涉及多芯片适配,成长空间大,但未明确晋升通道。

技术前沿前沿/新兴技术
技术栈AI编译器、VLA、VLM、TVM、TensorRT、PyTorch、AST
业务类型ambiguous

工作生活

50较低

未提及办公方式与WLB,北京/上海一线城市通勤压力大,但可能弹性。

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

使命价值

70中等

AI编译器是AI基础设施关键环节,对社会技术发展有推动,但具体使命未提。

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