职位描述
岗位职责:
1、实现与优化具身智能大模型的预训练代码(PyTorch/JAX),解决分布式训练、混合精度加速等工程问题;构建高效的数据处理Pipeline(多模态数据并行加载、增强)。
2、将预训练模型适配下游任务(如机器人抓取、导航),开发轻量化微调工具链;
3、优化模型在边缘设备(如机械臂、AGV)上的推理效率(TensorRT、ONNX)。
4、开发可视化调试工具,监控预训练过程中的模态对齐、损失收敛等关键指标。
任职要求:1、硕士学历,专业方向:计算机科学、自动化、电子信息。
2、熟练使用PyTorch,有大模型训练/微调经验(如LLaMA、Stable Diffusion);熟悉Linux环境、分布式训练(FSDP/Deepspeed)、CUDA优化;
3、了解机器人操作系统(ROS)或仿真工具(PyBullet)。
4、有具身智能相关研究经历或相关领域顶会文章发布加分。
以担保或任何理由索取财物,扣押证照,均涉嫌违法,请提高警惕