返回岗位列表

大模型训练优化工程师

查看原岗位

地点：北京薪资：70-100K 日期：2026-06-26

岗位摘要

负责大语言模型训练基础设施的设计与开发，包括大规模数据加载优化、训练数据构建和数据处理管道；负责大规模训练数据格式与存储方案设计，优化数据IO和分布式训练场景下的数据吞吐能力；负责模型工具链建设，包括模型格式转换、权重切分与合并、跨框架模型迁移

岗位职责

负责大语言模型训练基础设施的设计与开发，包括大规模数据加载优化、训练数据构建和数据处理管道
负责大规模训练数据格式与存储方案设计，优化数据IO和分布式训练场景下的数据吞吐能力
负责模型工具链建设，包括模型格式转换、权重切分与合并、跨框架模型迁移
参与大模型训练系统开发与优化，支持预训练、SFT、RLHF等不同训练流程，提高训练效率和GPU利用率
参与大模型推理基础设施开发，建设高性能推理平台，支持在线推理服务
跟踪大模型训练与推理领域的前沿技术，持续推动系统能力升级

任职要求

计算机科学、人工智能、软件工程或相关专业，硕士及以上学历，2年以上AI基础设施相关工作经验
精通Python/Go/C++中至少两门语言，能编写高质量代码
有良好的算法和数理基础，熟悉常用算法
熟练掌握深度学习框架（PyTorch）、分布式训练技术（5D并行）及相关框架（Megatron-LM），有实际模型训练经验，对LLM训练各环节有深入研究
熟练使用分布式计算系统如Ray、Spark，有实际使用和应用调优经验
擅长性能优化，对性能和稳定性有极致追求，能在压力下快速定位性能瓶颈并落地优化方案
具备良好的问题分析与解决能力、团队合作精神和沟通能力，能编写高质量文档

加分项

在国内外知名会议发表过相关领域的高质量论文
在ACM、PAC等编程竞赛中取得过较好成绩
知名开源项目的core contributor或maintainer
许先生刚刚活跃

福利待遇

薪资范围：70-100K·16薪
工作地点：北京海淀区大恒科技大厦南座F9层
参与前沿大模型技术研发
与优秀团队合作，持续推动技术升级

工作地址

北京海淀区大恒科技大厦南座F9层