AgiBot logo
智元机器人
具身智能算法工程师/专家(训练效率方向)

具身智能算法工程师/专家(训练效率方向)

发布于 大约 24 小时前

普通员工/个人贡献者

上海市
高级经验
全职员工
仅现场办公
硕士
研究与开发 (研发)
Cuda
Diffusion
Flow Matching
Gpu优化
Nccl
Pytorch Profiler
具身智能
分布式训练
训练效率

AI 估算 · 35k–55k

具身智能赛道火热,上海高级算法工程师薪资竞争力强,但B轮公司现金部分偏保守,搭配期权可能,月薪35-55K合理。

职位详情

关于这个职位

该职位专注于提升具身智能模型的训练效率,涉及从预训练到后训练的全链路优化

你需要深入理解 Transformer、扩散模型等架构及 IL/RL 等算法,利用 GPU 编程和分布式通信技术,系统性解决训练瓶颈
适合对高性能计算和算法优化有浓厚兴趣、希望在机器人 AI 前沿领域深耕的工程师

最低要求

计算机、AI 等相关专业硕士及以上学历

具备扎实的 Python/C++/CUDA 编程能力,有 GPU kernel 开发或性能调优经验
对深度学习模型架构与训练算法有扎实理解,能基于算法计算特性定位瓶颈并设计优化方案
熟练掌握性能分析工具(Nsight Systems / PyTorch Profiler / NVIDIA Profiler 等),能系统性定位训练瓶颈
熟悉分布式通信机制(NCCL / MPI / gRPC 等),理解通信拓扑、带宽、延迟对训练效率的影响
对训练全链路(数据加载 → 前向 → 反向 → 梯度同步 → 参数更新)有整体理解,能从算法和系统两个视角做联合优化

工作职责

负责具身智能训练效率的度量、分析与系统性优化,覆盖预训练与后训练全链路

深入理解不同模型架构(Transformer / Diffusion / Flow Matching 等)与训练算法(IL/RL/BC 等)的计算特性,针对性设计优化方案
构建训练效率监控体系,包括 GPU 利用率、训练吞吐、通信效率、数据加载延迟等核心指标的可视化与自动化追踪
模型训练效率优化:算子融合、CUDA graph、kernel 调优等,提升单卡和多卡计算效率
并行与通信效率优化:梯度同步、云边通信、权重分发等通信开销分析与优化,支撑百台规模训练不退化
数据加载效率优化:数据预处理 pipeline、IO 吞吐优化,消除数据侧瓶颈

优先资格

加分项

有 CUDA graph / 算子融合 / Triton kernel 开发经验
有大规模集群(百卡以上)训练效率优化实战经验
有训练效率监控平台或 profiling 工具的开发经验
有顶会论文或开源项目贡献经历

AI 洞察

优缺点分析

优点

  • 身处具身智能这一前沿赛道,技术领先性和市场价值高
  • 工作内容硬核,能深度锻炼 GPU 底层优化和分布式系统能力
  • 公司处于 B 轮融资,技术氛围浓厚,成长空间大
  • 训练效率优化对系统知识要求极高,需要同时理解算法和硬件
  • 面临百卡规模集群的复杂问题,调试和排查难度大
  • 行业竞争激烈,需持续跟进最新技术和论文

缺点 / 挑战

  • 适合对 GPU 底层技术有热情、享受性能调优挑战、愿意在具身智能领域深耕的技术型工程师

角色解读

  • 技术深钻:成为训练效率领域专家,主导下一代训练框架设计
  • 架构扩展:从单卡优化走向大规模集群,成为分布式系统专家
  • 跨领域融合:结合具身智能算法与系统优化,转型为 AI 系统架构师
  • 量化并分析训练效率瓶颈,从数据加载到梯度同步全链路监控
  • 针对不同模型架构和算法,设计算子融合、CUDA graph 等优化方案
  • 优化分布式通信效率,确保百台规模训练性能不退化
  • 开发自动化 profiling 工具,持续提升训练吞吐
  • 精通 Python/C++/CUDA 编程,具备 GPU kernel 开发经验
  • 深入理解深度学习模型(Transformer/Diffusion)和训练算法(IL/RL)
  • 熟练使用 Nsight Systems、PyTorch Profiler 等性能分析工具
  • 熟悉 NCCL/MPI 等分布式通信机制及集群拓扑

申请策略

  • 关注智元机器人发布的技术博客或开源项目,面试中展现对公司的了解
  • 准备一个完整的训练效率优化案例,从瓶颈定位到方案落地
  • 突出 GPU kernel 开发或性能调优的具体项目,量化提升效果(如训练速度提升 X%)
  • 展示分布式训练优化经验,包括集群规模、使用的框架和工具
  • 如果有开源贡献或顶会论文(如 MLSys、SC 等),务必重点列出
  • 学习 CUDA graph 和 Triton Kernel 编程,动手实践算子融合
  • 熟悉 PyTorch Distributed 和 NCCL 通信库,搭建小集群实验
  • 阅读主流训练框架(Megatron、DeepSpeed)的优化源码

面试指南

  • 结构化回答:先明确问题现象(如 GPU 利用率低),再分析可能原因(通信、计算、IO),然后提出针对性优化(如梯度压缩、算子融合)
  • STAR 法则:描述背景、任务、行动和结果,突出量化指标
  • 如何诊断并解决大规模分布式训练中的通信瓶颈?
  • 请解释 CUDA graph 的工作原理及适用场景
  • 你在训练效率优化中遇到最难的问题是什么?如何解决的?
  • Transformer 训练时,如何优化前向和反向计算效率?
  • 如何设计一个训练效率监控系统?关键指标有哪些?
  • 复习 CUDA 编程和 GPU 架构,熟悉常用性能分析工具

职位点评

68
综合评分

前沿技术栈、高成长性,但WLB一般,薪资中等偏上。

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

更适合这类人
适合以技术成长为核心诉求、愿意接受高强度工作以换取前沿技能积累的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利65
成长发展90
工作生活40
使命价值70

薪资福利

65中等

薪资处于行业中等偏上水平,B轮公司可能搭配期权,但现金部分不如大厂有竞争力。福利未在JD中明确,补偿性动机满足度一般。

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

成长发展

90较高

职位涉及前沿技术(具身智能、GPU底层优化、分布式系统),技术栈新且挑战大,能够显著提升个人技术深度和广度。成长路径清晰,但JD未明确提及晋升通道。

技术前沿前沿/新兴技术
技术栈CUDA、Transformer、Diffusion、NCCL、GPU kernel、训练效率
业务类型profit_center

工作生活

40较低

工作地点仅在上海,未提及远程或弹性办公。训练优化工作可能涉及紧急问题排查,需要高强度投入,工作生活平衡一般。

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

使命价值

70中等

具身智能属于高速增长赛道,对社会智能化有积极影响。但公司尚处于B轮阶段,产品落地和长期影响力尚不确定。

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