【留用实习】大模型AI Infra底软/硬件加速工程师当面告知

北京快手科技有限公司
2026-11-15 19:45:07
其他27届28届实习

岗位详情

地区:

北京
·
上海

专业要求:

人工智能计算机科学与技术软件工程

岗位职责:

1、负责视频生成大模型/多模态大模型在最新硬件平台上的训练/推理性能分析与优化,目标达成该硬件平台上的极限性能; 2、负责分析并提升大规模分布式训练系统的集群性能,完成对大规模训练任务故障的根因定位与稳定性提升。

招聘要求:

1、掌握Python/C++编程语言,熟练使用Pytorch训练框架或SGlang/vLLM等大模型推理框架; 2、掌握大模型领域基础算法知识,熟悉常规Transformer/SD模型结构,以及对应的常用性能优化方法; 3、深刻理解GPU硬件体系结构,能熟练编写高性能cuda kernel。 加分项: 1、有分布式大模型训练框架(Megatron/FSDP)或分布式推理引擎开发经验者优先; 2、有在超节点硬件环境上,做过训练/推理性能优化者优先; 3、有GPU高性能算子库(Cutlass/Cute、PTX、汇编)、集合通信库(xCCL)、AI编译器(XLA/Dynamo)开发经验者优先。

公司介绍

北京快手科技有限公司

北京快手科技有限公司

北京快手科技有限公司,成立于2015-03-20,位于北京,法定代表人:银鑫,登记状态:存续,注册资本10101.01万人民币,公司类型:民营/私企

公司信息:


公司名称:北京快手科技有限公司


公司类型:媒体/出版/影视/文化传播


公司性质:5000以上


公司地点:


公司规模:--


涉及领域:--


公司网址: