
超威半导体
AI Framework Eng.(VLLM CI)
AI Framework Eng.(VLLM CI)
发布于 1 小时前普通员工/个人贡献者
全职员工仅现场办公AI算法
AI算法
Cuda/Hip
Llm推理
Moe
Rocm
Sglang
Vllm
分布式系统
性能优化
值不值得投
82综合评分
综合评分 82
顶尖AI框架研发岗,技术前沿性强,成长空间巨大,薪酬优厚,但技术要求极高。
从薪资福利、成长发展、工作生活和使命价值四个维度综合评估,方便横向比较。
薪资怎么样
9.0万
比AI算法中位数高
发布情况
新岗位 · 今天发布
公司情况
这家公司招聘很活跃
97 个在招 · 其中AI算法 2 个
市场机会
选择面较广
上海 · AI算法 · 78 个在招
职位详情
关于这个职位
作为LLM推理框架团队的高级成员,负责构建和优化基于AMD GPU的大语言模型单节点及分布式推理运行时
重点在于提升吞吐量、延迟和内存效率,实现张量/流水线并行等大规模推理能力,并将优化成果开源至vLLM等项目,确立AMD在LLM服务领域的领先地位
最低要求
系统思维的机器学习工程师,关注吞吐量、延迟、内存移动和调度
能够阅读和修改大型推理框架,调试GPU和节点间的性能问题
与内核、编译器和网络团队协作以缩小端到端性能差距
热衷于开源工作并推动推理平台架构级改进
计算机科学、计算机工程、电气工程或相关领域的硕士或博士学位
工作职责
推理框架与运行时:基于内部LLM引擎或vLLM、SGLang、llm-d等开源栈,构建和优化分布式LLM推理运行时
混合执行设计:设计和改进TP/PP/EP(MoE)混合执行,包括KV-cache管理、注意力分发和Token调度
多节点管道:使用RCCL、RDMA和集体通信执行实施和优化多节点推理管道
性能与可扩展性:驱动单GPU和多GPU集群的吞吐量、延迟和内存效率
持续批处理优化:优化连续批处理、推测解码、KV-cache分页、前缀缓存和多轮服务
GPU与后端集成:与AMD GPU库(AITER, HIPBLAS-LT, RCCL, ROCm runtime)合作,确保推理框架高效使用FP8/FP4 GEMM和FlashAttention/MLA
编译器协作:与编译器团队(Triton, LLVM, ROCm)合作消除框架级性能瓶颈
开源与客户赋能:将特性和性能修复上游提交至vLLM、SGLang和llm-d
启用客户PoC和生产部署
构建和维护基准级推理管道
优先资格
推理栈知识:熟悉vLLM、SGLang或类似推理栈,有分布式推理扩展经验及上游开源贡献记录
深度学习集成:有将优化的GPU性能集成到PyTorch/TensorFlow等框架的经验
内核与推理框架:具备NVIDIA/AMD等GPU架构和内核开发背景
软件工程:精通Python,熟悉C/C++,具备大规模系统的调试、性能调优和测试设计能力
高性能计算:有在异构GPU集群上运行大规模工作负载并优化效率的经验
编译器与运行时优化:理解LLVM、ROCm及GPU代码生成
AI 洞察
优缺点分析
优点
- 处于AI基础设施最前沿,直接参与定义LLM推理标准,技术含金量极高
- 依托AMD全球研发资源,接触最顶级的GPU硬件和编译器工具链
- 开源导向文化,个人技术成果可直接转化为行业影响力,简历加分巨大
- 跨国巨头平台,薪酬福利体系完善,职业稳定性高
- 技术门槛极高,需同时掌握上层框架逻辑和底层硬件特性,学习曲线陡峭
- 性能优化工作枯燥且复杂,需面对海量的调试数据和细微的性能差异
- 跨部门协作难度大,需协调编译器、内核、网络等多个团队,沟通成本高
- 适合拥有深厚计算机系统背景、热爱底层优化、追求技术极致且在开源社区有影响力的资深工程师或研究人员
缺点 / 挑战
- 对学历和过往项目经验要求严苛,竞争压力极大
角色解读
- 成为LLM推理基础设施领域的权威专家,主导制定AMD在AI推理领域的技术路线
- 向首席科学家或技术总监方向发展,带领团队攻克下一代AI芯片的软件生态难题
- 在开源社区获得极高影响力,成为vLLM等项目的核心Maintainer
- 负责大语言模型(LLM)在AMD GPU上的推理引擎开发与性能优化,涉及分布式训练/推理的核心运行时
- 深入底层,通过优化KV-cache、连续批处理、MoE路由等机制,提升推理吞吐量和降低延迟
- 将自研优化成果反向贡献给vLLM、SGLang等主流开源项目,建立行业标准
- 与编译器、内核团队合作,打通从软件框架到硬件底层的性能瓶颈
- 精通Python和C/C++,具备大规模系统调试和性能调优能力
- 深刻理解LLM推理原理,熟练掌握vLLM、SGLang等框架源码
- 熟悉GPU架构(AMD ROCm/NVIDIA CUDA),了解HIP/AIET等底层库
- 具备分布式系统开发经验,熟悉RCCL/RDMA等通信协议
- 拥有计算机科学/工程硕士或博士学位,有顶级会议论文或开源核心贡献者优先
申请策略
- 准备一份详细的技术博客或GitHub仓库,展示你对推理框架源码的理解
- 面试中要展现出对“为什么这样做”的深层思考,而不仅仅是“怎么做”
- 关注AMD最新的AI战略和产品路线图,在面试中体现对公司业务的认同
- 突出LLM推理相关的实战项目,特别是针对vLLM/SGLang的优化案例或贡献PR
- 量化性能提升成果,如“将吞吐量提升X%”、“延迟降低Y ms”,体现数据驱动思维
- 强调C++/Python混合编程能力及Linux环境下的高性能计算经验
- 列出相关的学术论文、专利或GitHub高星项目链接
- 深入研究Transformer架构细节,特别是Attention机制和MoE路由算法
面试指南
- 采用STAR法则(情境、任务、行动、结果)描述项目经历,重点突出技术决策的依据
- 对于原理性问题,从数学公式推导到代码实现层层递进,展示理论深度
- 对于系统设计题,先界定边界条件,再提出核心架构,最后讨论权衡取舍(Trade-off)
- 展现对开源社区的尊重和理解,说明如何通过协作推动项目进展
- 请详细描述你优化过的一个LLM推理场景,遇到了什么性能瓶颈?你是如何定位和解决的?
- 解释一下Continuous Batching的工作原理,以及它在vLLM中是如何实现的?
- 在分布式推理中,如何处理GPU间的数据同步和通信开销?RCCL相比NCCL有哪些异同?
- 如果让你设计一个支持MoE模型的推理引擎,你会考虑哪些关键模块?
职位点评
薪资福利85
成长发展95
工作生活60
使命价值80
薪资福利
85较高
作为已上市的跨国巨头,提供具有市场竞争力的薪酬和完善的全方位福利保障,满足度高。
薪资信号偏高 (6.0千–12.0万/月)
福利待遇Benefits offered are described: AMD benefits at a glance.
成长发展
95较高
处于AI前沿技术领域,直接接触LLM推理核心架构,技术成长空间极大,且鼓励开源贡献。
技术前沿前沿/新兴技术
技术栈LLM、vLLM、SGLang、ROCm、MoE、Tensor Parallelism、C++、Python
成长机会ADVANCE YOUR CAREER、Upstream features... into open-source inference frameworks、driving architecture-level improvements
业务类型profit_center
工作生活
60中等
虽为外企,但高级研发岗位通常伴随较高的技术压力和交付节奏,WLB取决于具体团队文化,JD未明确提及弹性工作制。
工作模式未明确
办公地点科技园/产业园
加班情况未提及(无法判断)
使命价值
80较高
公司使命明确指向利用技术解决世界重要挑战,AI领域具有显著的社会价值和行业前景。
行业发展高速增长赛道
社会影响正向社会影响力较高
使命信号technology has the power to solve the world’s most important challenges、shaping the future、technology that moves the world forward
创新程度积极采用新技术
超威半导体 的其他在招职位
相似职位推荐
Industrial Engineering ES
德科斯米尔 · 沈阳市
AI 估算 · 2.0–3.5万21 分钟前发布
IT Intern - 数字员工平台应用工程师
康明斯 · 北京市
AI 估算 · 4.3–6.5千37 分钟前发布
IT Intern - 大模型应用工程师
康明斯 · 北京市
AI 估算 · 0.2–0.3千37 分钟前发布
高德-Agent应用算法工程师-北京
阿里巴巴 · 北京市
AI 估算 · 3.0–5.0万1 小时前发布
乌鸫科技-AI agent应用工程师-平台技术
阿里巴巴 · 杭州市
AI 估算 · 2.5–4.0万1 小时前发布
阿里控股-AI研发工程师-杭州
阿里巴巴 · 杭州市
AI 估算 · 3.0–6.0万1 小时前发布
研究型实习生-强化学习训练框架优化
阿里巴巴 · 北京市
AI 估算 · 0.3–0.4千1 小时前发布
阿里国际-AI算法工程师(视频生成 / 创意生成方向)-日常实习生
阿里巴巴 · 杭州市
AI 估算 · 0.2–0.3千1 小时前发布
Watch Jobs