小米集团招聘大模型训练框架开发工程师，要求 1–5 年分布式训练经验

小米集团招聘全职大模型训练框架开发工程师，基于 PyTorch、DeepSpeed、Megatron-LM 参与训练框架设计与优化，需解决万卡集群分布式训练通信、显存优化等问题。岗位要求 1–5 年高性能计算/分布式训练/深度学习系统研发经验，熟练修改核心源码并用 nsight/NCCL profiler 定位性能瓶颈。