拓达教育 Touchdown Education 拓达求职支持办公室Career Support Office touchdown.org.cn ↗
JobFinder为留学生筛选身份匹配的岗位

大模型推理/训练优化工程师

快手 · 北京 · 上海 · 深圳 · 杭州 · 8月9日 · 3周前
评分
85
发布日期
8月9日 (3周前)
内推
暂无
公司规模
10k+
行业
互联网

岗位描述

1、参与大模型推理/训练优化。通过研发业界领先的AI Compiler 技术,支撑搜推场景在GPU上的训练计算性能优化;支持大模型推理优化技术在异构硬件上的落地; 2、参与各种大模型推理所需的功能性开发任务;相关编译优化功能开发,以图优化、算子融合、GPU高性能算子开发及自动Codegen等技术手段不断推高在不同卡型上的计算性能极限; 3、参与支持日常的大模型推理服务部署,参与内部日常提效工具的研发。 1、硕士及以上学历,计算机、电子、自动化等专业优先; 2、熟悉python/C++,熟悉linux使用,有扎实的算法与数据结构基础; 3、有较强的自驱力和学习力,有严谨的科研思维,沟通良好擅长与人合作; 4、了解AI infra 整体技术栈,有大模型相关训练或推理优化实战经验;有vLLM、TensorRT-LLM、MLC-LLM、Tensorflow、PyTorch等框架之一的实践经验。 加分项: 1、熟悉CUDA 或 ROCM,熟悉Nsight System/ Nsight Compute 工具的使用,有 GPU 或 ASIC 高性能算子开发经验; 2、熟悉深度学习编译优化或异构硬件,有 XLA/ TVM /MLIR 开发、优化经验,熟悉pass编写或代码生成原理和实践;或有传统编译器开发经验,熟悉LLVM原理和使用; 3、有相关领域顶会paper发表; 4、有二次开发能力或开源社区贡献经历。

加入我的投递 →

快手 的其他岗位

【留用实习】AI Agent 全栈工程师-音视频方向 · 北京 · 评分 90
游戏客户端开发工程师 · 杭州 · 中国 · 评分 88
智能媒体处理研发工程师 · 北京 · 评分 87
策略开发工程师-数据安全方向 · 北京 · 中国 · 评分 87
AI商业化产品经理-PDE · 北京 · 中国 · 评分 87
大模型数据引擎工程师 · 北京 · 评分 86

查看全部 →