【留用实习】大模型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以上
公司地点:
公司规模:--
涉及领域:--
公司网址: