[Skip to main content](#main-content)

[OpenNash Jobs](https://jobs.opennash.com/)
[Hiring report](/reports/hiring/) · [All Xiaomi jobs](https://jobs.opennash.com/companies/xiaomi/) · [Search jobs](/search/)

Xiaomi · 北京

# 顶尖应届-大模型训练框架优化工程师-自动驾驶

北京 

Listed on Xiaomi ’s official careers site. Posted June 4, 2026. Check that it is still open and apply there.
[Apply on Xiaomi ’s careers site →](https://xiaomi.jobs.f.mioffice.cn/toptalent/position/7646711716172040467/detail)

[OpenNash Jobs](https://jobs.opennash.com/) / [Xiaomi ](https://jobs.opennash.com/companies/xiaomi/) / 顶尖应届-大模型训练框架优化工程师-自动驾驶

## Job description
- 负责基于现有大模型训练框架分析自动驾驶与机器人模型训练链路，包括数据加载、分布式并行、显存占用、通信开销和算子执行等环节，定位性能瓶颈；
- 围绕训练吞吐、GPU 利用率和模型迭代效率开展优化，推动训练周期缩短和训练资源成本下降；
- 结合 PyTorch、Megatron-LM、DeepSpeed 等技术栈，开展分布式训练策略、混合精度、算子融合、通信优化和 CUDA 高性能算子优化；
- 推进训练框架与 AI 平台、调度系统和监控体系协同，建立训练效率、集群算力利用率和任务稳定性的评估机制；
- 探索 AI Agent 在自动化 profiling、瓶颈定位和代码优化建议中的应用，沉淀可复用的训练优化工具链。

## Qualifications
- 本科及以上学历，扎实的编程基础（Python/C++）与良好的工程习惯；
- 精通深度学习框架底层原理（PyTorch 核心机制如 torch.compile、Autograd、TorchScript、XLA）；
- 熟悉分布式训练框架（Horovod、Megatron-LM、DeepSpeed 等）的实现逻辑；
- 熟悉 CUDA 编程及 GPU 架构，能独立开发高性能自定义算子（CUDA、CUTLASS、TileLang、Triton）；
- 理解大模型训练框架工程化实践，具备在现有框架基础上进行性能分析、模块扩展或业务适配的能力；
- 有深度学习编译技术栈（TVM、MLIR 等）、AI 开源社区贡献、训练性能优化或大规模模型训练经验者优先。

## At a glance
- Posted by the employer: Jun 4, 2026
- First seen by OpenNash: Aug 1, 2026
- Last checked: Oct 19, 2026
- Listed for: 137 days
- Location: 北京[Apply on Xiaomi ’s careers site →](https://xiaomi.jobs.f.mioffice.cn/toptalent/position/7646711716172040467/detail)

## More jobs at Xiaomi 

### [27届招聘实习生-国际职能平台](https://jobs.opennash.com/companies/xiaomi/jobs/6d4d527313cd96474e53/)

北京

### [AI Agent 产品经理](https://jobs.opennash.com/companies/xiaomi/jobs/fc2f52865a7e2d84fbd7/)

北京

### [AI Agent 架构师 / Tech Lead](https://jobs.opennash.com/companies/xiaomi/jobs/35dcfe12c39db2a6a88a/)

北京

### [AI Agent产品经理](https://jobs.opennash.com/companies/xiaomi/jobs/1b2859d295836fa011a9/)

北京

### [AI Agent研究员实习生-小爱](https://jobs.opennash.com/companies/xiaomi/jobs/0c7ddc4025af5dc78885/)

北京

## Similar jobs at other companies

### [自动驾驶-感知-信号灯检测算法实习生](https://jobs.opennash.com/companies/meituan/jobs/94eb8c82a9d00e0638de/)

Meituan · 深圳市

### [算法工程师-自动驾驶](https://jobs.opennash.com/companies/jd/jobs/10da2acb9a5d8541ec64/)

JD.com · 北京市-北京市

### [算法工程师-自动驾驶](https://jobs.opennash.com/companies/jd/jobs/50200ecdb6943189c1dc/)

JD.com · 北京市-北京市 +1 more

### [主管成本分析师（数字化&自动驾驶）](https://jobs.opennash.com/companies/chery-automobile/jobs/46daec5778e7aeb7c19b/)

Chery Automobile · 安徽省, 芜湖市

### [主任成本分析师（数字化&自动驾驶）](https://jobs.opennash.com/companies/chery-automobile/jobs/0bcff52ae17769ddb9f9/)

Chery Automobile · 安徽省, 芜湖市

OpenNash lists jobs from employers’ own careers sites and does not handle applications.
Dates come from the employer where stated; “first seen” and “last checked” are our own observations.

OpenNash Jobs · Official listings with dates and links to the employer. [Weekly hiring report](/reports/hiring/) · [Methodology](/methodology/)