小米的模型训练优化工程师薪资是多少？

该职位薪资范围为 35k–60k（人民币/月）。

模型训练优化工程师的工作地点在哪里？

该职位工作地点位于北京市、武汉市。工作形式为仅现场办公。

该职位要求学历未注明学历及中级经验工作经验。

🤖 AI 估测：¥35K-60K

发布时间：29 天前

这是一个专注于深度学习模型训练性能优化的技术岗位

你将负责开发高性能GPU算子，优化主流分布式训练框架，并构建自动化性能诊断系统，旨在持续降低大规模AI模型训练的成本与时间

基于 CUDA/Triton/Numba 开发高性能自定义算子，针对 GPU 硬件特性做深度调优，消除计算瓶颈

深度优化 DeepSpeed、Megatron-LM、FSDP 等主流框架的训练配置与执行效率

构建基于 Agent 的自动化性能诊断系统，对训练代码进行智能分析并自动给出优化方案

建立训练任务效能评估体系，通过技术手段持续压缩 GPU 训练成本

参与硬件选型评估与集群资源调度优化

3年以上 GPU 计算优化或深度学习系统开发经验，精通 CUDA 编程

熟练使用 Triton、Numba 等工具进行算子开发，深入理解 GPU 架构与显存体系

有大规模模型训练实战经验，熟悉主流分布式训练框架的内部机制

扎实的 profiling 能力，能精准定位计算、内存、通信各层面的性能瓶颈

有自动化调优系统或 AI Agent 构建经验者优先

模型训练优化工程师

🤖 AI 估测：¥35K-60K

发布时间：29 天前