
小米
顶尖应届-大模型AI算力研发工程师-基础技术平台
顶尖应届-大模型AI算力研发工程师-基础技术平台
发布于 大约 2 个月前普通员工/个人贡献者
北京市
无经验要求
全职员工
仅现场办公
硕士
软件工程
Ai算力
Megatron
Volcano
大模型
CUDA
GO
GPU
PyTorch
SGLang
AI 估算 · 20k–30k
AI算力研发方向热门,大厂应届硕士起薪较高,结合北京生活成本,预计月薪2-3万。
职位详情
关于这个职位
作为小米基础技术平台的大模型AI算力研发工程师,你将聚焦于GPU国产卡适配和大模型训推场景的性能优化,与顶尖团队一起攻克算力瓶颈
该岗位面向应届硕士及以上学历,要求熟悉CUDA生态和Kubernetes技术,适合对AI基础设施充满热情的技术人才
最低要求
计算机相关专业,硕士及以上学历
熟悉主流 GPU 训练和推理芯片,熟悉 CUDA 生态,有 CUDA 开发和性能调优经验优先
熟悉容器及 Kubernetes 技术,具备基于 Kubernetes 的 GPU 算力管理经验者优先
熟练 Go/Python/C++至少一门语言,有 PyTorch/Volcano/Megatron/SGLang 框架熟悉者优先
工作职责
GPU 国产卡的适配开发
大模型训推场景 GPU 相关性能优化和问题定位
AI 洞察
优缺点分析
优点
- 聚焦前沿的大模型算力方向,技术壁垒高,积累的经验在行业内有持续价值
- 小米作为大型科技公司,平台资源丰富,能够接触到大规模GPU集群和实际业务场景
- 团队属于基础技术平台,技术氛围浓厚,有利于快速成长
- 适合对AI基础设施和底层性能优化有浓厚兴趣,具备较强动手能力和计算机基础的应届硕士生
缺点 / 挑战
- 技术挑战大,需要深入理解GPU底层和分布式系统,学习曲线陡峭
- 国产卡适配和性能优化工作可能涉及较多调试和问题排查,工作强度较高
角色解读
- 深耕AI算力领域,成为GPU优化专家或算力平台架构师
- 向AI系统全栈方向发展,结合算法与工程,参与更核心的基础设施建设
- 未来可晋升为技术Leader,带领团队攻克算力难题
- 负责GPU国产卡的适配开发,确保国产芯片在AI训练推理场景中稳定高效运行
- 针对大模型训练和推理场景,进行GPU性能优化和问题定位,提升算力利用率
- 与团队协作,打造基于Kubernetes的算力管理平台,支撑大规模分布式训练
- 扎实的计算机基础,熟悉GPU架构和CUDA生态,具备性能调优经验
- 掌握容器化和Kubernetes技术,能够管理GPU算力资源
- 熟练使用Go/Python/C++至少一种语言,了解PyTorch、Volcano等框架
申请策略
- 在面试中展示对AI算力领域的热情,可针对小米的AI业务(如小爱、自动驾驶)准备相关理解
- 关注小米基础技术平台的技术博客或开源项目,增强技术共鸣
- 突出GPU相关项目经验,如CUDA编程、性能优化、并行计算等
- 展示容器化或Kubernetes实践经验,如GPU集群管理、资源调度
- 强调编程语言能力,特别是Go/Python/C++的熟练度,以及PyTorch等框架使用经历
- 提前学习CUDA编程和GPU性能分析工具(如Nsight、nvprof)
- 熟悉Kubernetes GPU调度原理,尝试搭建小规模GPU集群
面试指南
- 项目经验类:用STAR法则(情境-任务-行动-结果)清晰描述,突出技术深度和量化成果
- 技术原理类:先阐述基础概念,再结合实际问题给出分析思路,最后提出优化方案
- 请介绍你在CUDA编程或GPU优化方面的项目经验
- 如何优化大模型训练中的GPU显存利用率?
- Kubernetes如何管理GPU资源?谈谈你对device plugin的理解
- 比较不同GPU架构(如NVIDIA vs 国产卡)的特点和挑战
- 在分布式训练中如何解决通信瓶颈?
- 复习CUDA编程模型、内存管理、流并行等核心知识,准备一个完整的优化案例
职位点评
68
综合评分
大厂AI算力核心岗位,前沿技术栈,薪资有竞争力,但可能加班较多。
从起薪待遇、成长路径、工作节奏和岗位方向综合评估,方便比较职业起点。
更适合这类人
适合极度重视技术成长、愿意投入高强度学习工作的求职者,对工作生活平衡要求不高。
表现最好
成长发展
相对薄弱
工作生活
薪资福利60
成长发展90
工作生活40
使命价值70
薪资福利
60中等
薪资水平预计市场领先,但JD未明确具体薪酬福利,稳定性中等。
薪资信号未披露(AI估算:20K-30K/月)
成长发展
90较高
技术栈前沿,大模型算力方向成长空间大,但JD未提及明确晋升路径。
技术前沿前沿/新兴技术
技术栈GPU、CUDA、Kubernetes、PyTorch、Megatron、SGLang
业务类型ambiguous
工作生活
40较低
仅现场办公,地点北京可能通勤压力大,JD未提及工作生活平衡。
工作模式仅现场办公
办公地点未明确
加班情况未提及(无法判断)
使命价值
70中等
AI算力是数字经济基础设施,行业前景好,但直接社会影响力有限。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
小米 的其他在招职位
相似职位推荐
Watch Jobs