TheJobsMap
Xiaomi logo
Xiaomi

871 open roles

AI基建-大模型训练框架开发工程师

BeijingFull-timePosted Jan 16, 2026

Job description

  • 基于 PyTorch、DeepSpeed、Megatron-LM 等技术，参与大模型训练框架的设计、开发与优化，提升模型训练的效率、稳定性与扩展性，支撑大规模模型的高效训练和部署。
  • 解决万卡集群中大模型训练中的关键技术难题，包括分布式训练通信、内存显存优化、数据加载与预处理加速等，保障训练高效稳定，降低资源消耗。
  • 开展大模型框架性能评估和调优工作，构建并完善性能监控体系，通过实时监测训练指标，定位性能瓶颈，提出优化方案，确保在不同硬件上达到最优性能。

Description copied from Xiaomi's careers page. Read the full posting before you apply.

More jobs at Xiaomi

See all openings at Xiaomi

More jobs in Beijing

See all jobs in Beijing
Apply on Xiaomi's site