
百度
大模型训推框架优化工程师(J97863)
大模型训推框架优化工程师(J97863)
发布于 大约 15 小时前普通员工/个人贡献者
北京市 / 上海市
高级经验
全职员工
仅现场办公
学历未注明
软件工程
CUDA
大模型
推理引擎
DeepSpeed
SGLang
vLLM
训练框架
MoE
Megatron
AI 估算 · 30k–60k
百度高级工程师,一线城市,大模型赛道稀缺,技能难度高,市场竞争力强。
职位详情
关于这个职位
加入百度大模型训推框架优化团队,你将参与万卡级推理集群和千卡训练集群的深度优化,推动自研AI芯片规模化应用
工作涉及大模型推理引擎(MoE架构)、高性能KV Cache系统、全模态训练框架及芯片算子优化,是AI基础设施的核心技术岗位
最低要求
思维活跃,学习能力强、自驱力强,愿意主动捣腾事情
对大模型训推引擎或异构计算编程有浓厚的兴趣
拥有 vLLM、SGLang、Megatron、DeepSpeed 等主流框架的实战或者学习经验,或具备深入的 CUDA/高性能算子开发经历
对技术执着,不满足于“能跑”,致力于“更快”,愿意死磕每一个微小的优化点
具备良好的'Vibe Coding'素养,善于利用 AI 工具链提升工程效率
在大模型训推方向有优秀的论文产出
工作职责
大模型推理引擎优化:面向大规模 MoE 架构,从服务调度、并行策略、KV Cache 管理等维度进行极致优化
高性能 KV 系统开发:参与自研的生产级 KV Cache 池化系统的开发
全模态训练引擎研发:构建自研全模态训练框架,设计自适应切分与通算编排策略
自研芯片算子与适配:深入底层硬件,极致优化自研芯片算子性能,完成主流大模型框架的适配与迁移
AI 洞察
优缺点分析
优点
- 百度平台资源丰富,能深度参与开源社区(如SGLang),提升个人影响力
- 薪资竞争力强,互联网大厂待遇,且职位属于核心基础设施,稳定性高
- 技术更新快,需要持续学习新框架和硬件特性,保持技术敏锐度
- 适合对大模型训推有浓厚兴趣、热爱底层系统优化、追求技术极致的技术型人才,尤其是具有CUDA和高性能计算背景的工程师
缺点 / 挑战
- 身处AI大模型最前沿,接触万卡集群和自研芯片,技术挑战大,积累稀缺经验
- 工作强度较大,追求极致性能需要死磕细节,可能面临长时间调试压力
- 跨团队协作多,需要与芯片、框架、业务团队紧密配合,沟通成本较高
角色解读
- 技术专家路线:从单点优化到系统架构师,成为大模型训推领域权威
- 管理路线:主导核心项目,带领团队攻克技术难题,晋升为技术负责人
- 学术路线:将工程实践转化为高质量论文,推动行业技术进步
- 优化大规模MoE大模型推理引擎,提升吞吐和降低延迟,涉及服务调度、并行策略和KV Cache管理
- 开发高性能KV Cache池化系统,统一管理Prefill和Decode阶段的内存,突破显存限制
- 研发全模态训练框架,设计自适应切分与通算编排策略,提升LLM、VLM等模型的训练效率
- 深入底层硬件,优化自研AI芯片的算子性能,并完成主流框架的适配迁移
- 扎实的CUDA编程和高性能计算经验,能优化算子至极致性能
- 熟悉vLLM、SGLang、Megatron等主流大模型框架,有实际优化经验
- 对分布式系统、并行策略(如TP/PP/EP)和内存管理有深入理解
- 具备AI工具链使用能力,善于人机协同编程提升效率
申请策略
- 在面试中表达对技术的热情和执着,可以准备一个具体的性能优化案例进行深度讲解
- 关注百度大模型业务动态,了解文心一言、MaaS平台等产品,体现对业务价值的理解
- 重点突出在大模型推理/训练框架(如vLLM、Megatron)上的优化经历,附上具体性能提升数据
- 展示CUDA算子开发项目,说明如何通过优化显存、带宽等指标提升效率
- 如果有开源贡献或相关论文,务必列出,体现技术影响力
- 强调自驱力和对技术追求极致的案例,如“死磕某优化点实现x倍加速”
- 深入阅读vLLM、SGLang源代码,理解MoE调度和KV Cache管理实现
- 学习自研芯片的编程模型(如百度昆仑芯),提前熟悉其算子开发工具链
面试指南
- 针对性能优化问题,采用“问题分析-瓶颈定位-优化实现-效果验证”四步法,结合具体数据说明
- 对于系统设计题,先明确需求和约束,再提出架构方案,最后讨论权衡点
- 对于经验题,使用STAR(情境-任务-行动-结果)结构,强调你的贡献和学习收获
- 请解释MoE大模型推理中如何实现负载均衡,并设计一个优化方案
- 描述你如何优化一个CUDA kernel,使其达到接近理论峰值带宽
- 在分布式训练中,如何权衡数据并行、张量并行和流水线并行的组合?
- 如何设计KV Cache池化系统以支持长上下文推理?面临哪些挑战?
- 你使用过哪些AI工具辅助编程?请举例说明如何提升工程效率
职位点评
74
综合评分
大厂核心AI基础设施岗,前沿技术栈、高薪资,但工作强度大、现场办公。
更适合这类人
最适合追求技术极致、渴望系统性成长的发展型人才,对工作强度有较高承受力。
表现最好
成长发展
相对薄弱
工作生活
薪资福利85
成长发展95
工作生活40
使命价值75
薪资福利
85较高
该职位薪资水平处于行业高位,百度作为大厂福利完善,补偿性动机满足程度较高。
薪资信号市场水准 (30K-60K/月)
成长发展
95较高
接触大模型训推前沿技术、自研芯片和万卡集群,技能成长空间极大,发展性动机得到充分满足。
技术前沿前沿/新兴技术
技术栈大模型、MoE、KV Cache、CUDA、vLLM、SGLang、Megatron、DeepSpeed、自研芯片
业务类型profit_center
工作生活
40较低
未提及远程或弹性工作,大厂核心岗位通常要求现场办公且强度较大,生活化动机满足度有限。
工作模式仅现场办公
办公地点科技园/产业园
加班情况未提及(无法判断)
使命价值
75中等
大模型行业高速增长,但岗位偏向技术支持,社会影响力中性,意义感动机中等偏上。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
百度 的其他在招职位
相似职位推荐
Watch Jobs