logo of mi

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

校招全职地点:北京状态:招聘

工作描述


任职要求
1、本科及以上学历,扎实的编程基础(Python/C++)与良好的工程习惯;
2、精通深度学习框架底层原理(PyTorch 核心机制如 torch.compile、Autograd、TorchScript、XLA);
3、熟悉分布式训练框架(Horovod、Megatron-LM、DeepSpeed 等)的实现逻辑;
4、熟悉 CUDA 编程及 GPU 架构,能独立开发高性能自定义算子(CUDA、CUTLASS、Tile…
登录查看完整工作描述
微信扫码,1秒登录

包括英文材料
学历+
Python+
C+++
深度学习+
还有更多 •••
相关职位

logo of mi
校招

1、精通机器学习(深度学习),具备卓越的创新研究能力,充满对未知领域的好奇心,热衷于探索前沿技术边界; 2、编程能力出色,熟练掌握至少两种编程语言,精通Pytorch/Tensorflow,能够将创造

更新于 2026-06-02北京
logo of mi
校招

1、硕士及以上学历,计算机科学、软件工程、人工智能等相关专业,具备大模型训练Infra、推理加速或分布式系统研发经验。 2、熟悉Megatron-LM、DeepSpeed、FSDP等主流分布式训练框

更新于 2026-04-03北京
logo of mi
校招

1、计算机科学、人工智能、车辆工程等; 2、对 SFT、PEFT、强化学习、prompt engineering、RAG、Agent等大模型相关技术有深入了解,在此领域有至少一年以上的工作经验; 3、

更新于 2025-12-01南京
logo of mi
校招

1、计算机科学、人工智能、数据科学、知识工程、车辆工程、交通工程、可靠性工程等相关专业,博士优先; 2、熟悉知识图谱、因果推断、小样本学习、可解释AI、多源数据融合等方向; 3、熟悉Python、Py

更新于 2026-06-02南京