AgiBot logo
智元机器人
端侧AI infra工程师

端侧AI infra工程师

发布于 大约 14 小时前

普通员工/个人贡献者

北京市 / 上海市
中级经验
全职员工
仅现场办公
学历未注明
信息技术与基础设施
Ai Infra
Gr00T
Jax
Lerobot
Pytorch
Torch Compile
剪枝
模型微调
模型部署

AI 估算 · 25k–45k

端侧AI infra方向技术难度高,市场人才稀缺,结合B轮公司竞争力,薪资处于行业中上水平。

职位详情

关于这个职位

该职位负责端侧AI模型的微调、推理优化和部署,将算法落地到真实业务场景中

你需要熟悉具身智能框架(如lerobot、gr00t)和主流深度学习框架,并掌握多种训练优化手段
适合对AI infra和模型部署感兴趣、具备较强工程能力的工程师

最低要求

熟悉掌握常见有监督、无监督等算法模型的原理、优缺点、适用场景等基础知识

熟悉lerobot、gr00t等具身框架中的训练机制,包括但不限于BC,RL,HIL等
熟悉Jax、Pytorch等主流深度学习框架,并有实际的模型训练、调优的项目经验
熟悉torch profiler,torch compile,tirtion,cudagraph等训练优化手段,并可以快速赋能训练过程
熟悉transformer engine,model opt,torchao等算法优化框架,可以配合模型量化、剪枝、蒸馏等工作加速算法推理效果
良好的沟通能力、解决问题能力
参与AI平台的研发,提升系统性能和稳定性
与团队合作,共同解决技术难题,推进项目进展

工作职责

优化端侧微调+推理流程,加速端侧部署

负责算法在真实业务场景中的落地,紧贴业务需求,不断改进算法在业务中的效果
对接模型优化、模型部署、应用框架等多种能力,为算法交付提供有力支撑
积极跟进AI学术界和业界的最新动态,优化内部算法模型,改进模型部署效率

AI 洞察

优缺点分析

优点

  • 涉及多种先进优化工具和框架,能显著提升工程和技术广度
  • 公司处于机器人赛道热门方向,发展前景良好
  • 端侧部署受设备资源限制,优化难度高,可能面临高强度工作
  • 技术栈更新快,需保持对多种工具和框架的熟练度
  • 适合热爱底层优化、有扎实工程能力、对AI infra有浓厚兴趣的工程师,尤其是希望在具身智能领域深耕的求职者

缺点 / 挑战

  • 接触具身智能这一前沿领域,技术挑战性高,成长空间大
  • 需持续跟进AI和工程领域的最新进展,学习压力较大

角色解读

  • 向AI infra架构师方向深化,负责大规模AI系统的设计与优化
  • 成为模型优化专家,专注于端侧推理效率的极致提升
  • 结合具身智能领域发展,转型为机器人AI算法专家或技术负责人
  • 负责端侧AI模型的微调与推理流程优化,加速模型在终端设备上的部署
  • 将算法落地到具体业务场景,根据业务需求持续改进算法效果
  • 对接模型优化、部署和应用框架,确保算法交付的整体效率和稳定性
  • 跟踪AI领域最新进展,将前沿技术引入内部模型优化和部署流程
  • 熟练掌握深度学习框架(如Jax、Pytorch)和模型训练调优经验
  • 熟悉具身智能框架(lerobot、gr00t)及其训练机制(BC、RL、HIL)
  • 掌握训练优化工具(torch profiler、torch compile、triton、cudagraph)和算法优化框架(transformer engine、modelopt、torchao)
  • 具备模型量化、剪枝、蒸馏等加速推理的知识和实践经验

申请策略

  • 提前了解智元机器人的具身智能产品和业务方向,在面试中展示技术热情
  • 可以准备一段关于端侧推理优化整体方案的思考,体现系统思维
  • 突出模型优化和部署项目经验,特别是端侧或资源受限场景的实战案例
  • 强调对Jax、Pytorch等框架的熟练度以及训练调优的具体成果
  • 列出熟悉的优化工具(如triton、cudagraph)和算法压缩技术(量化、剪枝)
  • 若参与过AI平台或infra建设,详细描述系统性能提升的量化指标
  • 补全对torch compile、triton等工具的代码级理解,争取有实际优化案例
  • 学习具身智能框架lerobot、gr00t,可通过开源项目快速上手

面试指南

  • 首先分析瓶颈(计算、内存、带宽),然后选择合适的优化技术(如算子融合、量化、剪枝),并验证效果
  • 结合具体案例,说明优化前后的性能对比(耗时、内存占用),强调权衡精度与速度
  • 对于框架问题,体现从原理到实践的理解,提及遇到的挑战和解决方案
  • 请描述你如何优化一个端侧模型的推理速度?会考虑哪些因素?
  • 你使用过哪些训练优化工具?举例说明它们如何提升训练效率
  • 解释模型量化、剪枝、蒸馏的原理及适用场景
  • 你有具身智能框架(如lerobot)的使用经验吗?描述一个相关项目
  • 如何处理一个在端侧设备上部署的模型精度下降问题?

职位点评

70
综合评分

前沿AI infra岗位,技术栈新颖,成长性强,但工作地点固定且WLB不明。

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

更适合这类人
最看重技术成长和前沿性,对工作地点和具体福利要求不高的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利70
成长发展85
工作生活50
使命价值70

薪资福利

70中等

薪资未明确披露,但基于职位技术难度和公司融资阶段,薪资应有一定竞争力,福利信息缺失,补偿性动机满足程度中等。

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

成长发展

85较高

技术栈前沿,涉及多种优化工具和具身智能框架,成长机会多,但JD未明确提及晋升通道或培训体系。

技术前沿前沿/新兴技术
技术栈端侧推理、模型微调、模型部署、Jax、Pytorch、lerobot、gr00t、torch compile、triton、cudagraph、transformer engine、modelopt、torchao、模型量化、剪枝、蒸馏、AI infra
业务类型ambiguous

工作生活

50较低

工作地点为北京或上海,仅现场办公,未提及弹性工作或远程选项,WLB信号不明显,生活化动机满足程度较低。

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

使命价值

70中等

机器人行业处于高速增长赛道,具身智能具有社会影响力,但JD未强调使命感或社会价值信号。

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