
阿里巴巴
平头哥-边缘AI芯片推理框架开发工程师-上海
平头哥-边缘AI芯片推理框架开发工程师-上海
发布于 大约 14 小时前普通员工/个人贡献者
上海市
高级经验
全职员工
仅现场办公
硕士
机器学习工程
Cuda
Pytorch
Sglang
Tensorrt
Tensorrt-Llm
Vllm
边缘Ai
Llm推理优化
AI 估算 · 35k–65k
上海大厂AI推理优化岗,技术门槛高,资深人才稀缺,薪资竞争力强,通常15薪左右。
职位详情
关于这个职位
该职位聚焦边缘AI芯片上的大模型推理优化,负责将LLM/VLM等模型高效部署到阿里平头哥芯片上,并解决性能、精度问题
核心工作涉及CUDA编程、TensorRT等推理框架的深度调优,以及软硬件协同加速方案的探索
适合有并行计算和推理优化经验、热爱底层性能优化的工程师
最低要求
电子工程,计算机等相关专业硕士及以上学历
具备3年以上AI推理优化相关工作经验,深刻理解并行计算和CUDA编程,熟悉TensorRT和TensorRT-LLM的模型部署和优化
熟练掌握Python、C/C++编程语言,掌握AI Vibe Coding技术,良好的逻辑思维与问题分析能力
熟悉主流深度学习框架Pytorch,推理框架vLLM、SGlang,TensorRT, TensorRT-LLM 等
责任心强,能够跨团队/部门协作分析与解决业务场景中AI软硬件全栈的功能、精度、性能问题达成产品化目标
工作职责
负责AI模型以及大模型(LLM、VLM、VLA) 的部署和推理优化,结合AI软硬件特性实现优化计算和推理效率优化,包括但不限于多模型部署、KV Cache 管理、 Inflight Batching 等技术
在边缘AI上适配SOTA开源框架和SOTA模型,分析解决适配过程中发现的功能、性能与精度问题,为客户提供问题支持和解决方案
深入挖掘边缘AI软件栈和系统性能瓶颈,提出软硬件的加速解决方案
AI 洞察
优缺点分析
优点
- 阿里平头哥深耕半导体和AI芯片,技术积累雄厚,平台资源丰富
- 岗位涉及大模型推理优化,技术前沿,个人成长快
- 可接触到从芯片到软件栈的全栈问题,提升综合能力
- 上海地区薪资竞争力强,大厂福利完善
- 需要快速适配新模型和开源框架,工作节奏可能较快
- 跨团队协作要求高,需要处理软硬件复杂的定位问题
缺点 / 挑战
- 技术门槛高,需要同时掌握CUDA、多种推理框架和芯片架构,学习压力大
- 适合对AI系统优化有浓厚兴趣、具备扎实并行计算基础、喜欢挑战技术难题的工程师
角色解读
- 成为AI推理优化领域的专家,深入掌握软硬件协同设计,向技术专家或架构师方向发展
- 可转向芯片软件栈的架构设计,或带领团队负责边缘AI产品的整体性能优化
- 有机会接触前沿大模型技术,未来可向AI系统研究员或技术管理职位发展
- 负责AI模型(包括LLM/VLM/VLA)在边缘芯片上的部署与推理优化,提升计算效率和推理速度
- 通过KV Cache管理、Inflight Batching等技术优化多模型部署和并发推理性能
- 适配并维护开源推理框架(TensorRT、vLLM等),解决实际部署中的功能、性能和精度问题
- 深入分析边缘AI软件栈性能瓶颈,提出软硬件协同加速方案
- 精通CUDA编程和并行计算,熟悉TensorRT/TensorRT-LLM模型部署与优化
- 熟练使用Python和C/C++,掌握PyTorch等深度学习框架,熟悉vLLM、SGlang等推理框架
- 理解AI芯片架构和边缘计算场景,具备系统性能分析和调优能力
- 具备跨团队协作和问题解决能力,责任感强
申请策略
- 了解平头哥的产品线和技术方向,在面试中展示对边缘AI场景的理解
- 准备一个完整的性能优化案例,展示发现问题、分析瓶颈、解决方案的闭环
- 突出在AI推理优化方面的项目经验,特别是使用TensorRT、CUDA进行加速的实际案例
- 强调对大模型部署的实践经验,如参与过LLM推理服务开发或优化
- 展示对底层硬件和系统性能的理解,如CPU/GPU/NPU架构分析
- 如有开源贡献或性能调优成果,可重点列出
- 深入掌握TensorRT-LLM和vLLM的源码级优化,熟悉KV Cache等机制
- 学习边缘AI芯片(如含光系列)的软件栈,了解NPU指令集
面试指南
- 对于技术题,先讲原理,再结合实践,给出量化结果
- 对于案例题,按照STAR法则(情境、任务、行动、结果)结构回答
- 对于系统设计题,从目标出发,分析瓶颈,提出方案并权衡取舍
- 解释一下TensorRT的模型优化流程,如何减少精度损失?
- 大模型部署时,如何处理KV Cache的内存管理?
- 在边缘设备上部署模型的瓶颈是什么?如何通过软硬件协同优化?
- 如何比较vLLM与SGlang的调度策略?
- 描述一次你优化推理性能的完整经历
职位点评
77
综合评分
上海大厂AI推理优化岗,技术前沿薪资高,但办公灵活性一般。
从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。
更适合这类人
最适合追求技术深度和高速成长、对高薪有期待,且能接受一定工作强度的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利85
成长发展90
工作生活50
使命价值70
薪资福利
85较高
上海大厂高薪职位,薪资福利具有较强吸引力,虽未明确薪资金额,但行业竞争力突出。
薪资信号未披露(AI估算:35K-65K/月)
成长发展
90较高
技术栈前沿,涉及大模型和AI芯片优化,成长空间大,能持续学习前沿技术。
技术前沿前沿/新兴技术
技术栈CUDA、TensorRT、TensorRT-LLM、vLLM、LLM、边缘AI
业务类型ambiguous
工作生活
50较低
仅现场办公,无弹性工作安排明确说明,大厂节奏可能较快,生活自由度一般。
工作模式仅现场办公
办公地点未明确
加班情况未提及(无法判断)
使命价值
70中等
边缘AI芯片属于高速增长赛道,对行业技术进步有贡献,但直接社会价值体现不明显。
行业发展高速增长赛道
社会影响中性/一般
创新程度积极采用新技术
阿里巴巴 的其他在招职位
相似职位推荐
Watch Jobs