训练基建工程师

职位描述

工作职责：

1、负责具身智能训练平台的基础架构设计与核心开发，覆盖数据管线、训练调度、推理部署三大模块

2、基于 Isaac Lab 等仿真框架，搭建和优化大规模并行强化学习训练流水线，支撑运控策略的高效迭代

3、设计并实现训练数据管理系统，涵盖数据采集、清洗、版本管理与高效读写，支持 sim-to-real 数据闭环

4、开发模型推理与部署服务，实现策略模型在仿真环境与实机硬件上的低延迟推理与热更新

5、建设实验管理与可观测性体系，包括训练指标追踪、实验对比、模型版本管理与 artifact 存储

6、编写清晰、规范的技术文档（架构设计、接口规范、部署手册、变更记录等）

7、参与代码评审，推动工程规范与基础设施质量持续提升

任职要求

任职要求：

1、本科及以上学历，计算机科学、软件工程、自动化或相关专业

2、扎实的计算机基础，熟悉常用数据结构、算法及常见设计模式

3、熟练掌握 Python，具备 C/C++ 开发能力；熟悉 PyTorch 核心机制及 GPU 编程基础

4、了解强化学习训练流程，具备 PPO、SAC 等算法的工程落地经验

5、熟悉 Docker，具备容器化开发与部署的实践经验

6、熟练使用 Git 工作流，适应代码评审与 CI/CD 驱动的团队协作模式

7、具备良好的英文技术文档阅读与写作能力

加分项：

1、熟悉 Isaac Lab / Isaac Gym 等 NVIDIA 仿真训练框架，有具身智能训练平台相关经验

2、具备分布式训练经验，了解多卡 / 多节点训练的通信机制（NCCL、Ring-AllReduce 等）

3、熟悉 Kubernetes，有 GPU 集群资源调度与编排经验

4、有 MLOps 平台建设经验，熟悉 MLflow、Weights & Biases、Ray 等实验管理 / 调度工具

5、熟悉实时推理优化，有 TensorRT / ONNX Runtime 模型部署或边缘端推理经验

6、有机器人运控、MuJoCo、URDF/MJCF 模型处理等相关经验