AI 工程师（工具链）

负责大模型预训练上下游工具链的设计、开发与维护，覆盖模型转换、评测、可视化分析、实验管理等关键环节，提升训练研发效率与系统可靠性。与预训练、后训练、评测等团队协作，构建稳定、可扩展、易维护的基础设施平台，为大规模模型研发提供高质量工程支撑。

岗位职责

负责预训练上下游工具链的设计、开发与维护，包括但不限于：模型转换工具链、评测工具链、实验结果可视化与监控工具链、自动化测试与验证系统。

参与模型训练框架与生态适配工作。

建立工具链质量保障体系，完善单元测试、集成测试、回归测试和 CI/CD 流程。

优化工具链易用性与可维护性，提升研究人员开发效率。

参与容器化环境建设与维护，包括 Docker 镜像构建、验证、版本管理等。

与预训练、后训练、评测团队协同推进工具链标准化与工程化建设。

任职要求

计算机相关专业本科及以上学历。

扎实的软件工程基础，具备良好的代码设计能力和工程规范意识。

熟练掌握 Python，具备较强的系统开发能力。

具有大型软件项目开发经验，熟悉预训练工具链的测试体系建设。

熟悉 Linux 开发环境及常见工程工具链。

熟悉 Docker、容器化部署及镜像管理流程。

加分项

具有 FSDP、Megatron-LM、Megatron-Bridge、Transformer Engine 等生态开发经验。

具有 ML Infra、LLM Infra 或 AI 平台开发经验。

具有 SwanLab、WanDB、MLFlow 等实验管理工具开发经验。

具有 CI/CD 平台建设经验。