训练平台性能优化工程师实习生当面告知
小米科技有限责任公司
2027-01-28 12:08:13
28届29届实习本科硕士
岗位详情
地区：
北京
专业要求：
人工智能计算机科学与技术软件工程
岗位职责：
深度优化训练流程
主导模型训练全链路性能分析与优化，设计GPU资源弹性调度策略
开发自动化训练加速工具链，构建可扩展的云端训练框架
研发混合精度训练、梯度压缩等前沿技术，突破训练吞吐瓶颈
构建训练优化体系
制定标准化训练效能评估体系，建立成本-效率量化模型
设计可复用的训练加速组件库，沉淀最佳实践方法论
开发训练过程性能分析平台，实现性能问题智能诊断
赋能业务研发
优化多任务资源调度策略，提升GPU集群整体利用率
为算法团队提供训练加速解决方案，缩短模型迭代周期
招聘要求：
1.精通深度学习训练加速技术，熟悉分布式训练框架设计
2.掌握CUDA编程及GPU性能分析工具（nsys/torch profiler）
3.熟练使用PyTorch框架，具备训练流程优化实战经验
4.熟悉常见模型压缩技术（量化/剪枝/蒸馏）及落地应用
5.具备大规模集群资源调度系统开发经验者优先
【技术加分项】
1.有mmcv//deepspeed/megatron/ray等训练框架开发经验
2.算子优化经验,triton/cuda等开发经验.
3.熟悉MPI/NCCL等分布式通信协议
4.在MLSys/ICLR等顶会发表过训练优化相关论文
公司介绍
小米科技有限责任公司
小米科技有限责任公司，成立于2010-03-03，位于北京，法定代表人：雷军，登记状态：存续，注册资本185000万人民币，公司类型：民营/私企
"小米科技有限责任公司成立于2010年3月3日，是专注于智能硬件和电子产品研发的全球化移动互联网企业，同时也是一家专注于智能手机、智能电动汽车、互联网电视及智能家居生态链建设的创新型科技企业。小米公司创造了用互联网模式开发手机操作系统、发烧友参与开发改进的模式。“为发烧而生”是小米的产品概念。“让每个人都能享受科技的乐趣”是小米公司的愿景。小米公司应用了互联网开发模式开发产品，用极客精神做产品，用互联网模式干掉中间环节，致力让全球每个人，都能享用来自中国的优质科技产品。小米已经建成了全球最大消费类IoT物联网平台，连接超过4.78亿台智能设备，进入全球100多个国家和地区。MIUI全球月活跃用户达到5.5亿。小米系投资的公司超500家，覆盖智能硬件、生活消费用品、教育、游戏、社交网络、文化娱乐、医疗健康、汽车交通、金融等领域。"