Alibaba logo
阿里巴巴
阿里云智能-AI推理/训练优化专家-北京/杭州/上海

阿里云智能-AI推理/训练优化专家-北京/杭州/上海

发布于 4 小时前

普通员工/个人贡献者

AI 估算 · 4.0–7.0万 · AI 参考估算
中级经验
学历未注明
全职员工仅现场办公AI算法
AI算法
Ai推理优化
Cuda
Deepspeed
Megatron
Moe
Pytorch
Vllm
分布式训练
算子优化

值不值得投

76综合评分

综合评分 76

顶尖AI底层技术栈、高成长潜力、高压高回报的研发岗位

从薪资福利、成长发展、工作生活和使命价值四个维度综合评估,方便横向比较。

薪资怎么样

5.5万

比AI算法中位数高

发布情况

新岗位 · 今天发布

公司情况

这家公司招聘很活跃

约 4820 个在招 · 其中AI算法 66 个

市场机会

选择面较广

北京 · AI算法 · 约 140 个在招

职位详情

关于这个职位

负责阿里云智能主流深度学习框架及分布式训练/推理场景的性能调优,涵盖算子、通信、内存等关键指标

参与分布式训练框架建设(如MoE、模型并行)及推理引擎优化(如KV Cache管理、Attention-FFN分离),推动AI DSL编译优化与国产算力适配,提升集群整体运行效率

最低要求

年以上相关经验,熟悉深度学习框架(pytorch),分布式训练和推理技术框架(如deepseed、fsdp、megatron、vllm、sglang),AI系统性能分析(如Nsight、PyTorch Profiler等)

理解 Attention、KV Cache、MoE、PD、模型并行与请求调度,能够分析不同负载下的性能瓶颈
熟悉Tilelang、Triton、Cutlass等编译/算子优化,有AI DSL、手写汇编等相关经验者优先
对国产算力生态相关社区有贡献或实践、具备国产化芯片性能调优经验者优先
有强烈技术热情和好奇心,自驱力和学习力强
具备良好的分析与解决问题的能力、沟通以及团队合作能力
喜欢挑战性的技术研发工作,善于攻坚克难,有创新热情,积极乐观,坚韧抗压,结果导向,能够持续推动问题的解决和突破

工作职责

针对主流深度学习框架、分布式训练和模型部署场景等,进行性能调优,优化算子性能、通信性能、内存利用率等关键指标,提升集群整体运行效率

参与分布式训练框架建设和优化,包括模型并行、长上下文、MoE、低精度、数据读取加速等
参与推理引擎请求调度,多级KV Cache管理,投机采样算法优化与 Draft 模型训练适配,性能仿真等策略算法和工具的实现,优化cache命中率和资源利用率和推理性能
参与推理引擎Attention–FFN 分离,异构和跨数据中心PD分离等新型架构的设计与实现,优化 A/F 资源配比、专家资源共享、微批流水及分布式调度,优化跨节点传输与计算通信重叠
参与基于tilelang/triton的编译优化和调优、cutlass高性能算子实现和国产算力社区建设,推动模型day0适配
参与AI DSL编译优化方向,探索通算融合、megakernel、kernel design agent等前沿技术,推动其在训推场景的规模化落地

AI 洞察

优缺点分析

优点

  • 依托阿里云平台,拥有海量真实业务场景和顶级算力资源支持
  • 在国产算力生态建设中具有先发优势,行业影响力大
  • 技术迭代极快,需持续跟进最新论文与开源项目
  • 对底层系统原理要求极高,容错率低
  • 热衷于底层系统优化、对AI基础设施有深厚兴趣、具备极强自驱力和技术攻坚能力的资深工程师

缺点 / 挑战

  • 接触业界最前沿的AI大模型训练/推理技术栈,技术挑战性极高
  • 性能优化往往面临极端复杂的调试环境,压力较大

角色解读

  • 可向AI基础设施架构师、首席科学家方向发展,或在通用人工智能(AGI)底层系统优化领域成为行业顶尖专家,主导下一代AI训练/推理平台的技术演进
  • 深入优化深度学习框架(如PyTorch)及分布式训练/推理系统的性能,解决算子、通信、内存等瓶颈
  • 设计并实现高效的推理引擎策略(如KV Cache管理、MoE调度),推动AI DSL编译技术与国产算力适配
  • 精通PyTorch及DeepSpeed/Megatron/VLLM等主流框架
  • 掌握Triton/Cutlass等编译优化技术
  • 深刻理解Attention机制、模型并行、MoE架构原理
  • 具备C++/CUDA编程能力及性能剖析工具使用经验

申请策略

  • 准备详细的技术案例,说明如何解决具体的性能瓶颈问题
  • 了解阿里云在AI领域的战略布局,展现对云计算与AI结合的理解
  • 突出展示在深度学习框架二次开发、高性能算子编写(CUDA/Triton)、分布式训练稳定性优化方面的具体项目成果
  • 量化性能提升指标(如吞吐量提升%、延迟降低ms)
  • 深入研究MoE、PagedAttention等最新大模型架构细节
  • 加强C++/CUDA底层编程能力
  • 关注国产AI芯片(如华为昇腾、寒武纪)的软件栈适配经验

面试指南

  • 采用STAR法则(情境、任务、行动、结果)描述项目
  • 重点阐述问题分析思路、技术选型依据、遇到的困难及解决方案、最终量化的性能收益
  • 请详细描述你优化过的一个最复杂的深度学习算子或内核,使用了哪些技术手段?
  • 在分布式训练中,如何平衡通信开销与计算重叠?请举例说明
  • 针对MoE架构,你认为当前的调度策略有哪些痛点?你会如何改进?
  • 如何使用Nsight或PyTorch Profiler定位GPU内存泄漏或性能瓶颈?
  • 复习计算机体系结构、并行计算原理
  • 重温Transformer及其变体(如FlashAttention)的数学推导与工程实现

职位点评

薪资福利85
成长发展95
工作生活40
使命价值70

薪资福利

85较高

作为已上市巨头企业的核心研发岗,薪资水平通常处于行业高位,且福利体系完善,能满足较高的补偿性动机。

薪资信号未披露(AI估算:4.0千–7.0万/月)

成长发展

95较高

职位涉及AI推理/训练优化的最前沿技术领域,技术成长空间极大,能接触到行业顶尖的技术挑战和资源。

技术前沿前沿/新兴技术
技术栈PyTorch、DeepSpeed、Megatron、VLLM、Triton、Cutlass、MoE、KV Cache、AI DSL
业务类型profit_center

工作生活

40较低

互联网大厂核心研发岗通常工作强度较大,JD中提到的“坚韧抗压”暗示了较高的工作负荷,WLB相对一般。

工作模式未明确
办公地点市区核心地段
加班情况JD含高强度暗示词

使命价值

70中等

推动国产算力生态建设和AI技术落地,具有一定的行业价值和社会意义,但主要服务于商业产品。

行业发展高速增长赛道
社会影响中性/一般
创新程度开拓性创新(行业首创)
Watch Jobs