小米
算法架构工程师 - 训练框架方向
算法架构工程师 - 训练框架方向
发布于 大约 2 小时前普通员工/个人贡献者
北京市
中级经验
全职员工
仅现场办公
本科
算法优化工程师
Pytorch
Tensorflow
分布式训练
异构计算
性能优化
推荐系统
AI 估算 · 40k–70k
北京一线大厂，算法架构方向，技术难度高，薪资竞争力强，参考市场水平。
职位详情
关于这个职位
作为小米广告、搜索、推荐业务的算法架构工程师，你将专注于训练与推理框架的研发和性能优化，探索百亿流量场景下的异构计算与分布式技术
你将参与前沿技术调研，推动更大规模模型在业务中的落地，适合对技术有极致追求、热爱挑战的工程师
最低要求
熟练掌C++/Rust/Python编程语言，熟悉Linux开发环境
具备扎实的编程能力，熟悉常用算法和数据结构，对代码有洁癖、追求极致者优先
了解机器学习基础理论知识、熟悉TensorFlow、PyTorch、有主流大规模推荐/搜索/广告在线服务的开发经验优先
追求挑战，具备较强学习能力和实践精神，有强烈的求知欲、好奇心，能及时关注和学习业界最新技术
工作职责
负责异构计算（GPU/CPU）、分布式训练、分布式推理、机器学习、生成式推荐等前沿技术架构迭代方向，参与训练、推理框架全链路流程开发、性能优化
参与到前沿技术的调研讨论，致力于通过模型训练、推理框架的能力迭代及性能的优化，支撑更多样的模型结构，更大参数规模的模型在业务场景的应用落地
AI 洞察
优缺点分析
优点
- 岗位涉及分布式训练、推理优化等热门方向，技能积累价值高
- 团队氛围注重技术追求，有良好的成长空间
- 对技术深度和广度要求高，需要持续学习
缺点 / 挑战
- 小米作为大型互联网公司，业务规模大，技术挑战高，能接触前沿技术
- 工作强度可能较高，需要应对复杂的技术问题和业务压力
- 适合对技术有极致追求、热爱挑战、具备扎实编程基础和机器学习知识的工程师
角色解读
- 技术专家路线：深耕训练/推理框架，成为分布式系统与性能优化专家
- 架构师路线：参与系统架构设计，逐步成长为技术架构师
- 管理路线：带团队负责核心模块，向技术管理方向发展
- 负责广告、搜索、推荐业务的训练和推理框架开发，包括分布式训练、异构计算等
- 参与全链路流程的性能优化，提升模型训练和推理效率
- 调研前沿技术，推动新模型结构和更大参数规模模型在业务中的应用
- 精通C++/Rust/Python，熟悉Linux环境，具备扎实的编程和数据结构基础
- 了解机器学习基础，熟悉TensorFlow或PyTorch等主流框架
- 有大规模推荐/搜索/广告在线服务开发经验者优先
申请策略
- 了解小米互联网业务，特别是广告、搜索、推荐场景
- 准备展示对技术细节的追求，如代码优化案例
- 突出C++/Rust/Python编程能力，展示相关项目经验
- 强调分布式系统、性能优化或大规模推荐系统相关经历
- 展示对机器学习框架（TensorFlow/PyTorch）的使用和优化经验
- 深入学习分布式训练原理，如参数服务器、AllReduce等
- 熟悉GPU编程（CUDA）或异构计算优化
- 了解最新生成式推荐技术
面试指南
- 结合项目经验，用STAR法则回答，突出技术难点和解决方案
- 展示对底层原理的理解，如分布式通信、GPU架构
- 强调数据驱动和实验验证的思维
- 如何优化大规模分布式训练的性能？
- 描述一次你进行性能调优的经历
- C++中内存管理有哪些注意事项？
- 如何设计一个高效的推理服务？
- 复习C++/Rust/Python基础，特别是内存管理、并发
职位点评
72
综合评分
小米算法架构工程师，技术前沿，薪资高，但工作强度可能较大。
从薪资福利、成长空间、工作节奏和岗位方向综合评估，方便横向比较。
更适合这类人
适合追求技术成长、热爱挑战、对分布式系统和性能优化有浓厚兴趣的工程师。
表现最好
成长发展
相对薄弱
工作生活
薪资福利75
成长发展90
工作生活50
使命价值60
薪资福利
75中等
薪资水平较高，福利待遇好，但JD未明确具体薪资和福利细节。
薪资信号未披露（AI估算：40K-70K/月）
成长发展
90较高
技术前沿，涉及分布式训练、异构计算等，成长空间大，公司平台好。
技术前沿前沿/新兴技术
技术栈C++、Rust、Python、TensorFlow、PyTorch、分布式训练、异构计算
业务类型profit_center
工作生活
50较低
工作地点在北京，未提及远程或弹性办公，可能工作强度较大。
工作模式仅现场办公
办公地点未明确
加班情况未提及（无法判断）
使命价值
60中等
公司业务规模大，技术影响力强，但JD未强调社会价值。
行业发展稳定成熟行业
社会影响中性/一般
创新程度积极采用新技术
小米 的其他在招职位
相似职位推荐
Watch Jobs