企业招聘
职位
关于禾蛙
合伙人计划
联系客服快速发单
扫码添加企业微信
客服电话
400-7777-5125
猎企入驻
400-777-5125
免费注册
登录
**********************
AI Infra 算子优化专家
接单
收藏职位
分享职位
100-150万
上海
研究生
5-8年
招聘 1 人
预计佣金
170.1K
1天前刷新/2天前发布
迅致直营
反馈快
72小时新发
职位亮点
反馈快、回款快
JD基本信息
岗位职责
1. 高性能算子交付。负责生产级高性能算子的开发与持续优化,覆盖 Hopper / Blackwell 等 NVIDIA GPU 架构,充分利用 Tensor Core、TMA、WGMMA 等硬件特性,保障算子在不同代际 GPU 上的性能领先与稳定性。 2. 算法迭代中的算子瓶颈定位与演进式支持。在算法模型结构演进过程中(如 Attention 变体等),快速定位性能瓶颈算子,评估算子融合可能性、收益与代价,输出兼顾性能、精度和可维护性的演进式算子优化路线图,确保算法迭代不被算子性能阻塞 3. 算子 Benchmark 与选型决策。建立核心算子的标准化 Benchmark 流程,系统性对比 cuBLAS、TE、Liger-Kernel 等主流实现,在不同尺寸和精度等条件下给出数据驱动的选型结论,为训推框架的算子使用策略提供依据
任职要求
1. 学历与经验:计算机、软件工程、人工智能等相关专业,硕士及以上学历,3-8 年 AI 算子或 CUDA Kernel 开发经验 2. LLM 核心算子实战经验:有 Flash-Attention、RMSNorm、AdamW 等 LLM 关键算子的性能优化或生产级实现经验;熟悉 cuBLAS、TE、Liger-Kernel 等主流算子库的区别和适用场景 3. CUDA 体系化能力:精通 CUDA 编程模型和执行机制,深入理解不同代际 GPU 架构(Ampere / Hopper / Blackwell)的特性差异;熟练使用 CUTLASS / CuTe / Triton 等算子开发框架;熟练使用 ncu Profiler 工具 4. Pytorch 后端与通信:熟悉 PyTorch 框架的算子调度与后端实现机制,理解 ATen 及 CUDA 后端链路,能够分析 torch 原生算子的性能瓶颈,并做出引入其他算子库替换的工程决策。熟悉 NCCL 集合通信原语和常见问题,能在算子设计中考虑通算 Overlap 可能性,为大规模 scale-up 训练提供支撑 5. 融合算子与精度分析:熟悉计算图优化主流方案(如 torch.compile 或 CUDA graph)和使用限制,具备 GEMM-based 融合算子的开发能力;深入理解常见精度的数值特性、误差来源和适用边界,在算法迭代中能快速定位并解释数值不稳定根因并提出解决方案 6. 编程语言与 Native 工程能力:精通 C/C++,Linux 系统底层基础扎实,熟悉现代 Native 开发体系 7. 软素质:工作细心踏实不浮躁;沟通顺畅,与算法 / Infra 其他角色高效协作;具备 owner 意识,对算子精度,数据参数,性能结果高度敏感 加分项: 1. 有国产 AI 芯片(如沐曦、昇腾、寒武纪等)算子移植、调优和交付经验 2. 熟悉主流 LLM 训练/推理框架(Megatron-LM / vLLM 等),熟悉训练/推理流程,并熟悉内部算子调用路径 3. 向主流算子库开源项目(如 CUTLASS、TE、Flash-Attention 等)提交过被主分支合并的关键 Patch
所属行业:
计算机软件
职能分类:
算法工程师
工作城市:
上海,招聘1人,详细地址:上海市静安区江宁路 699 号moho
职位要求
学历要求:
研究生·统招
工作年限:
5-8年
技能/证书:
-
薪资福利
年薪范围:
100-150万*15薪
薪资福利:
五险一金、当地人力引进政策
团队架构
所属部门:
研发部
下属人数:
-
部门架构:
无
汇报对象:
部门经理
职级职称:
无
面试信息
面试轮次:
4轮
面试流程:
初面(业务部门)- 领导评价 - 高层面试 - HR
视频面试:
可以接受
举报
为你推荐
职能类别
IT互联网技术
电子/通信/半导体
销售/客服
制药/医疗器械/医疗护理
高级管理
关于
禾蛙动态
合伙人计划
猎企入驻
帮助中心
禾蛙协议
禾蛙海外版
联系我们
电话:400-777-5125(服务时间:工作日9:00至18:00)
网址:www.hewa.cn
地址:中国(上海)自由贸易试验区滨江大道99弄6号第11层
服务支持
接单方小程序
发单方小程序
微信公众号
©版权所有
上海禾蛙科技有限公司
沪公网安备31011502406070号
备案号:沪ICP备2026021218号-7
|
营业执照
|
服务许可证