模型训练平台开发工程师

岗位职责：
我们正在寻找一位专注于AI模型训练平台构建与优化的技术专家，负责模型训练平台及训练任务的优化，确保平台稳定性、高效率和可扩展。
1. 训练平台架构设计与构建。基于云原生技术设计和实现大规模AI模型训练平台，支持千卡规模以上的分布式训练任务；构建高效的资源调度和管理系统，优化GPU计算资源的利用率，确保平台的高性能和可扩展性。
2. 大规模训练稳定性优化。解决大规模训练过程中的稳定性问题，包括软硬件故障、通信瓶颈、数据同步与加速等；设计和实现容错机制，确保训练任务在硬件或网络故障时能够快速恢复。
3. 故障定位与诊断。设计和实现高效的故障定位系统，能够在大规模集群的复杂环境中快速定位故障点；利用日志分析、性能监控和分布式追踪技术，实时监控训练任务的运行状态，快速识别和修复问题；简化故障排查流程，减少人工干预。
4. 训练效率提升。利用大模型训练的优化技巧，结合自动驾驶场景的超大数据集，持续优化提升整体训练效率；

任职要求：

1. 本科及以上学历，计算机、软件相关专业，有AI模型训练平台开发或优化经验，尤其是大规模分布式训练；
2. 掌握PyTorch等深度学习框架，具备丰富的分布式训练优化经验，熟悉通信优化、混合精度训练、梯度压缩等技术。
3. 具备丰富的故障定位和诊断经验，能够快速识别和解决问题；
4. 扎实的编程技术，熟悉 python/Golang/C++等至少编程语言；
5. 具备良好的团队合作精神，能够与跨职能团队紧密合作。
加分项：
1. 有自动驾驶或相关领域的大规模模型训练经验；

【职位类别】软件序列
【招聘项目】平台技术
【所属公司】北京地平线信息技术有限公司