招聘中 社招

VLA训练infra算法工程师 - XiaomiRobotics

小米

  • 北京

收录时间

岗位职责

  1. 基于 PyTorch 生态(FSDP / DeepSpeed / Megatron 等)设计并实现 VLA 模型的分布式训练方案(DP / TP / PP / MoE),构建稳定高效的训练框架
  2. 推动混合精度(BF16 / FP8)与算子融合(FlashAttention / Triton kernel)
  3. 构建高吞吐数据pipeline,设计数据格式与 shard 策略,实现高效的数据加载
  4. 支持大规模实验追踪、管理、指标可视化

职位要求

  1. 扎实的 Python / C++ 基础,熟悉 Linux 系统
  2. 深入理解 PyTorch 分布式训练, 熟悉至少一种分布式训练优化框架(FSDP / DeepSpeed / Megatron)
  3. 深入理解 GPU / CPU / 内存 / 网络 基本原理
  4. 有大规模训练经验 加分项
  5. 有 多模态 / VLA / 机器人 / 自动驾驶 大规模训练经验
  6. 熟悉 NCCL / RDMA / InfiniBand 调优
  7. 有自定义 CUDA / Triton kernel 经验

相似岗位

有些机会,只在官网短暂出现

真正值得关注的岗位,常常只在企业官网短暂开放,可能两三天后就下线,也未必会同步到综合招聘平台。没有持续关注,你甚至不会知道它曾经出现。职先机持续聚合并核验官网岗位,帮你抓住职场先机,快人一步。

微信小程序 / 当前岗位

在微信里继续看这个岗位

小米VLA训练infra算法工程师 - XiaomiRobotics

扫码直达当前岗位收藏、浏览记录和下线提醒留在微信里
电脑端可直接微信扫码;手机端可保存小程序码后在微信中识别。
微信扫码在职先机小程序查看VLA训练infra算法工程师 - XiaomiRobotics正在准备岗位码
职先机微信小程序一岗一码 · 正式版直达保存小程序码

岗位提醒 · 01

收藏当前岗位,下线及时通知

当前岗位VLA训练infra算法工程师 - XiaomiRobotics小米 · 北京

01

扫码进入当前岗位无需重新搜索,直接打开当前岗位详情。

02

收藏并开启提醒在小程序内收藏后,岗位下线或变化时及时通知。

03

继续查看相似机会原岗位变化时,可继续浏览相关在招岗位。

职先机会持续核验公开岗位;岗位状态与最终招聘结果仍以企业招聘官网为准。

微信小程序扫码打开当前岗位

扫码打开当前岗位,完成收藏并开启下线提醒。

微信扫码在职先机小程序收藏VLA训练infra算法工程师 - XiaomiRobotics正在准备岗位码

微信小程序 · 当前岗位

请前往微信小程序提交内推申请

小米VLA训练infra算法工程师 - XiaomiRobotics社招 · 北京

01进入当前岗位扫码或打开后直达本岗位内推申请页。

02提交 PDF 简历简历替换、处理进度和消息提醒在小程序内同步。

手机端将尝试直接打开微信;电脑端请使用微信扫码。

微信扫码进入VLA训练infra算法工程师 - XiaomiRobotics内推申请页正在准备岗位码
当前岗位专属码扫码后无需重新搜索岗位保存小程序码

内推提示

暂不支持该岗位内推

如需投递,建议前往官网进行投递。