顶尖应届-具身大模型训练推理工程师-机器人事业部当面告知
小米科技有限责任公司
2026-11-23 18:42:22
其他校招26届27届
岗位详情
地区：
北京
专业要求：
人工智能计算机科学与技术
岗位职责：
1、基于 PyTorch 生态（FSDP / DeepSpeed / Megatron 等）设计并实现具身大模型分布式训练方案（DP / TP / PP / MoE）；
2、推动混合精度（BF16 / FP8）与算子融合（FlashAttention / Triton kernel）等性能优化，构建高吞吐数据 pipeline 与 shard 策略；
3、负责模型推理加速与部署优化，设计 Causal-DiT / MoE-DiT / AR-Diffusion 等混合架构提升吞吐与生成质量；
4、构建模型数据的自蒸馏 / 自进化框架，让模型能高效输出数据并持续优化；
5、支持大规模实验追踪、管理、指标可视化，与算法团队协同完成模型高效迭代。
招聘要求：
1、扎实的 Python / C++ 基础，熟悉 Linux；
2、深入理解 PyTorch 分布式训练，熟悉至少一种分布式训练优化框架（FSDP / DeepSpeed / Megatron-LM）；
3、深入理解 GPU / CPU / 内存 / 网络 基本原理；理解 Transformer / Diffusion / MoE 等主流结构的性能瓶颈与优化方法。
加分项
1、多模态 / VLA / 世界模型 / 视频生成 大规模训练经验；
2、熟悉 NCCL / RDMA / InfiniBand 调优；有自定义 CUDA / Triton kernel 经验；
3、熟悉 TensorRT / ONNX Runtime / Triton 等推理框架，具备量化、算子融合、显存优化实践经验；
4、在 NeurIPS / ICLR / ICML / CVPR / ICCV / ECCV / RSS / CoRL / ICRA 等顶会有相关成果。
公司介绍
小米科技有限责任公司
小米科技有限责任公司，成立于2010-03-03，位于北京，法定代表人：雷军，登记状态：存续，注册资本185000万人民币，公司类型：民营/私企
"小米科技有限责任公司成立于2010年3月3日，是专注于智能硬件和电子产品研发的全球化移动互联网企业，同时也是一家专注于智能手机、智能电动汽车、互联网电视及智能家居生态链建设的创新型科技企业。小米公司创造了用互联网模式开发手机操作系统、发烧友参与开发改进的模式。“为发烧而生”是小米的产品概念。“让每个人都能享受科技的乐趣”是小米公司的愿景。小米公司应用了互联网开发模式开发产品，用极客精神做产品，用互联网模式干掉中间环节，致力让全球每个人，都能享用来自中国的优质科技产品。小米已经建成了全球最大消费类IoT物联网平台，连接超过4.78亿台智能设备，进入全球100多个国家和地区。MIUI全球月活跃用户达到5.5亿。小米系投资的公司超500家，覆盖智能硬件、生活消费用品、教育、游戏、社交网络、文化娱乐、医疗健康、汽车交通、金融等领域。"