顶尖应届-大模型训练框架研发工程师-AI 基建当面告知
小米科技有限责任公司
2026-11-23 18:43:32
校招博士26届27届硕士
岗位详情
地区：
北京
专业要求：
软件工程
岗位职责：
1、基于 PyTorch、DeepSpeed、Megatron-LM 等，设计、优化大模型训练框架，提升效率、稳定性与扩展性，支撑大规模高效训练与部署。
2、攻克分布式通信、内存显存优化、数据加载加速等关键技术，保障高效稳定训练，降低资源消耗。
3、开展性能评估与调优，构建监控体系，实时定位瓶颈并提出优化方案，确保跨硬件最优性能。
招聘要求：
1、硕士及以上学历，计算机、软件工程、数学等相关专业，具备扎实计算机理论基础和丰富编程经验；
2、熟悉 C++ / Python，掌握至少一种深度学习框架（如 PyTorch），理解Transformer系统深度学习原理算法；
3、熟悉 GPU 和其他加速硬件的使用，有基本的性能调优经验，有CUDA优化、Triton经验优先；
4、了解大模型训练框架，熟悉分布式训练原理和技术，能针对大模型训练性能瓶颈进行优化优先；
5、有良好团队协作与沟通能力，有强烈的上进心和自我驱动，学习适应能力强，乐观自信，能挑战自我不断追求卓越。
公司介绍
小米科技有限责任公司
小米科技有限责任公司，成立于2010-03-03，位于北京，法定代表人：雷军，登记状态：存续，注册资本185000万人民币，公司类型：民营/私企
"小米科技有限责任公司成立于2010年3月3日，是专注于智能硬件和电子产品研发的全球化移动互联网企业，同时也是一家专注于智能手机、智能电动汽车、互联网电视及智能家居生态链建设的创新型科技企业。小米公司创造了用互联网模式开发手机操作系统、发烧友参与开发改进的模式。“为发烧而生”是小米的产品概念。“让每个人都能享受科技的乐趣”是小米公司的愿景。小米公司应用了互联网开发模式开发产品，用极客精神做产品，用互联网模式干掉中间环节，致力让全球每个人，都能享用来自中国的优质科技产品。小米已经建成了全球最大消费类IoT物联网平台，连接超过4.78亿台智能设备，进入全球100多个国家和地区。MIUI全球月活跃用户达到5.5亿。小米系投资的公司超500家，覆盖智能硬件、生活消费用品、教育、游戏、社交网络、文化娱乐、医疗健康、汽车交通、金融等领域。"