AMD logo
超威半导体
AI Framework Eng.(VLLM CI)

AI Framework Eng.(VLLM CI)

发布于 1 小时前

普通员工/个人贡献者

AI 估算 · 6.0–12.0万 · AI 参考估算
高级经验
硕士
全职员工仅现场办公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
创新程度积极采用新技术
Watch Jobs