返回岗位列表

小米算法架构师

查看原岗位

地点：北京薪资：薪资未公开日期：2026-01-29

岗位摘要

通过技术及模型能力迭代支撑搜广推业务，持续提升广告/推荐场景点击率、转化率等业务指标；负责分布式的训练框架及推理框架的研发及性能优化工作，实现大规模稀疏模型的训练及推理能力落地；负责特征DSL相关能力建设，持续优化特征工程、模型迭代效率

岗位职责

通过技术及模型能力迭代支撑搜广推业务，持续提升广告/推荐场景点击率、转化率等业务指标
负责分布式的训练框架及推理框架的研发及性能优化工作，实现大规模稀疏模型的训练及推理能力落地
负责特征DSL相关能力建设，持续优化特征工程、模型迭代效率
探索容器化、异构计算、大模型训练/推理等前沿技术架构方向，参与一站式分布式机器学习平台的研发
持续优化代码性能，提升模型迭代效率，为公司节省计算成本

任职要求

掌握机器学习/深度学习基础理论知识，了解常见的个性化推荐算法；熟悉TensorFlow/PyTorch等深度学习框架设计原理、阅读过相关源码者优先
熟练掌握C++/Rust，会使用Python、Java等编程语言，或者可以在一个月内熟练掌握一门语言
掌握GPU编程，熟悉CUDA、并行计算等技术者优先，有算子开发及优化经验者优先
有主流大规模推荐/搜索/广告引擎架构经验者优先
追求挑战，较强学习能力和实践精神，有强烈的求知欲、好奇心，能及时关注和学习业界最新技术
对代码有执念，追求极致，具备一定影响力的开源项目者优先