Ivyea Jobs 14753 roles · 250 companies · 刚刚 帮公司接 AI
启元机器人 · 具身智能

AI Infra工程师(预训练方向)

职位描述

1.负责感知导航团队AI基础设施建设,支撑大规模端到端模型的数据、训练与推理全链路高效运转。 2.数据管线:构建机器人多模态数据(图像、点云、轨迹、语言指令等)的采集、清洗、标注、存储与管理平台,打造高效的数据闭环。 3.训练基建:搭建并优化大规模分布式训练框架,支持多机多卡训练、混合精度、并行策略(DP/TP/PP/FSDP),提升训练吞吐与资源利用率。 4.推理基建:负责模型在机端/边缘/云端的高性能部署,实现模型量化、剪枝、蒸馏、算子优化与推理引擎集成(TensorRT/ONNX/TVM等)。 5.建设实验管理、模型版本管理与CI/CD流程,提升算法团队整体研发效率。

任职要求

1.负责具身智能统一预训练框架的设计与迭代,支撑VLA、大模型、世界模型、大小脑等多种模型架构的高效训练。 2.深入理解VLA、多模态大模型、世界模型、视频生成等前沿算法,基于算法特性设计框架层面的训练策略与优化方案。 3.设计并实现多模态数据加载体系,支持图片、视频、点云、触觉等多种数据格式的高效接入与预处理。 4.设计并实现多种分布式并行策略(数据并行、模型并行、流水线并行、混合并行等),适配从百亿到千亿级参数模型的训练需求。 5.构建数据全生命周期闭环能力,包括数据版本管理、质量筛选、配比策略,打通从数据采集到模型训练的完整链路。 6.跟进预训练领域前沿进展,具备快速复现并工程化落地新算法的能力。
启元机器人去官方页面投递 →