小米的VLA训练infra算法工程师薪资是多少？

该职位薪资范围为 35k–60k（人民币/月）。

VLA训练infra算法工程师的工作地点在哪里？

该职位工作地点位于北京市。工作形式为仅现场办公。

该职位要求学历未注明学历及高级经验工作经验。

🤖 AI 估测：¥35K-60K

发布时间：24 天前

这是一个专注于大规模视觉语言模型（VLA）训练基础设施开发的算法工程师岗位

你将负责设计和实现基于PyTorch生态的分布式训练框架，优化训练性能与效率，并构建高吞吐的数据处理流水线

核心目标是构建稳定、高效的训练系统，以支持前沿多模态模型的研发

基于 PyTorch 生态（FSDP / DeepSpeed / Megatron 等）设计并实现 VLA 模型的分布式训练方案（DP / TP / PP / MoE），构建稳定高效的训练框架

推动混合精度（BF16 / FP8）与算子融合（FlashAttention / Triton kernel）

构建高吞吐数据pipeline，设计数据格式与 shard 策略，实现高效的数据加载

支持大规模实验追踪、管理、指标可视化

扎实的 Python / C++ 基础，熟悉 Linux 系统

深入理解 PyTorch 分布式训练，熟悉至少一种分布式训练优化框架（FSDP / DeepSpeed / Megatron）

深入理解 GPU / CPU / 内存 / 网络基本原理

有大规模训练经验

有多模态 / VLA / 机器人 / 自动驾驶大规模训练经验

熟悉 NCCL / RDMA / InfiniBand 调优

有自定义 CUDA / Triton kernel 经验

VLA训练infra算法工程师

🤖 AI 估测：¥35K-60K

发布时间：24 天前