小米|算法架构工程师 - 训练框架方向

小米 · 北京 · 更新时间:2026-09-11

岗位职责

1、小米广告、搜索、推荐业务的算法架构工程团队,通过技术及模型能力迭代支撑互联网业务发展,探索百亿流量场景下的训练/推理框架研发与性能优化; 2、负责异构计算(GPU/CPU)、分布式训练、分布式推理、机器学习、生成式推荐等前沿技术架构迭代方向,参与训练、推理框架全链路流程开发、性能优化; 3、参与到前沿技术的调研讨论,致力于通过模型训练、推理框架的能力迭代及性能的优化,支撑更多样的模型结构,更大参数规模的模型在业务场景的应用落地。

任职要求

1、熟练掌C++/Rust/Python编程语言,熟悉Linux开发环境; 2、具备扎实的编程能力,熟悉常用算法和数据结构,对代码有洁癖、追求极致者优先; 3、了解机器学习基础理论知识、熟悉TensorFlow、PyTorch、有主流大规模推荐/搜索/广告在线服务的开发经验优先; 4、追求挑战,具备较强学习能力和实践精神,有强烈的求知欲、好奇心,能及时关注和学习业界最新技术。

小米校园招聘:算法架构工程师 - 训练框架方向岗位来自小米招聘官网,具体内容以小米招聘官网为准。