顶尖应届-大模型训练框架优化工程师-自动驾驶当面告知
小米科技有限责任公司
2026-11-23 18:41:54
校招26届27届本科硕士
岗位详情
地区：
北京
专业要求：
人工智能计算机科学与技术软件工程
岗位职责：
1、负责基于现有大模型训练框架分析自动驾驶与机器人模型训练链路，包括数据加载、分布式并行、显存占用、通信开销和算子执行等环节，定位性能瓶颈；
2、围绕训练吞吐、GPU 利用率和模型迭代效率开展优化，推动训练周期缩短和训练资源成本下降；
3、结合 PyTorch、Megatron-LM、DeepSpeed 等技术栈，开展分布式训练策略、混合精度、算子融合、通信优化和 CUDA 高性能算子优化；
4、推进训练框架与 AI 平台、调度系统和监控体系协同，建立训练效率、集群算力利用率和任务稳定性的评估机制；
5、探索 AI Agent 在自动化 profiling、瓶颈定位和代码优化建议中的应用，沉淀可复用的训练优化工具链。
招聘要求：
1、本科及以上学历，扎实的编程基础（Python/C++）与良好的工程习惯；
2、精通深度学习框架底层原理（PyTorch 核心机制如 torch.compile、Autograd、TorchScript、XLA）；
3、熟悉分布式训练框架（Horovod、Megatron-LM、DeepSpeed 等）的实现逻辑；
4、熟悉 CUDA 编程及 GPU 架构，能独立开发高性能自定义算子（CUDA、CUTLASS、TileLang、Triton）；
5、理解大模型训练框架工程化实践，具备在现有框架基础上进行性能分析、模块扩展或业务适配的能力；
6、有深度学习编译技术栈（TVM、MLIR 等）、AI 开源社区贡献、训练性能优化或大规模模型训练经验者优先。
公司介绍
小米科技有限责任公司
小米科技有限责任公司，成立于2010-03-03，位于北京，法定代表人：雷军，登记状态：存续，注册资本185000万人民币，公司类型：民营/私企
"小米科技有限责任公司成立于2010年3月3日，是专注于智能硬件和电子产品研发的全球化移动互联网企业，同时也是一家专注于智能手机、智能电动汽车、互联网电视及智能家居生态链建设的创新型科技企业。小米公司创造了用互联网模式开发手机操作系统、发烧友参与开发改进的模式。“为发烧而生”是小米的产品概念。“让每个人都能享受科技的乐趣”是小米公司的愿景。小米公司应用了互联网开发模式开发产品，用极客精神做产品，用互联网模式干掉中间环节，致力让全球每个人，都能享用来自中国的优质科技产品。小米已经建成了全球最大消费类IoT物联网平台，连接超过4.78亿台智能设备，进入全球100多个国家和地区。MIUI全球月活跃用户达到5.5亿。小米系投资的公司超500家，覆盖智能硬件、生活消费用品、教育、游戏、社交网络、文化娱乐、医疗健康、汽车交通、金融等领域。"