锦鲤求职锦鲤求职

小米集团

顶尖应届-大模型训练框架研发工程师-AI 基建

让 AI 辅助填写这个岗位的官网网申

使用同一份求职档案填写教育、实习与项目经历;登录、验证码和缺失信息由你处理,最终提交前请本人核对确认。

当前公开岗位库收录 小米 819 个在招校招岗位。岗位数量与状态以企业官网为准。

也可以勾选下方同企业岗位,一起准备网申 →

是否提交成功以企业官网回执或申请记录为准。

岗位描述

1、基于 PyTorch、DeepSpeed、Megatron-LM 等,设计、优化大模型训练框架,提升效率、稳定性与扩展性,支撑大规模高效训练与部署。
2、攻克分布式通信、内存显存优化、数据加载加速等关键技术,保障高效稳定训练,降低资源消耗。
3、开展性能评估与调优,构建监控体系,实时定位瓶颈并提出优化方案,确保跨硬件最优性能。

1、硕士及以上学历,计算机、软件工程、数学等相关专业,具备扎实计算机理论基础和丰富编程经验;
2、熟悉 C++ / Python,掌握至少一种深度学习框架(如 PyTorch),理解Transformer系统深度学习原理算法;
3、熟悉 GPU 和其他加速硬件的使用,有基本的性能调优经验,有CUDA优化、Triton经验优先;
4、了解大模型训练框架,熟悉分布式训练原理和技术,能针对大模型训练性能瓶颈进行优化优先;
5、有良好团队协作与沟通能力,有强烈的上进心和自我驱动,学习适应能力强,乐观自信,能挑战自我不断追求卓越。

人工智能方向的申请准备

先核对岗位要求与自己的经历,再准备档案、投递和面试。

阅读求职步骤与示例 →