招聘信息分享 招聘信息分享 ← 返回岗位列表

大模型训练性能优化工程师(训练算子)(深圳/北京/上海/杭州)

腾讯📍北京TEG技术两年以上工作经验
产品线:公共
发布时间:

岗位职责

1.负责深度学习训练相关算子的设计、实现与优化( CUDA/CUTLASS/Triton ); 2.面向大模型训练场景,对算子进行端到端性能分析与调优,持续挖掘吞吐、延迟、显存利用率等指标的优化空间; 3.参与或主导 3D 并行(Data / Tensor / Pipeline Parallel 等)训练体系下的算子与通信方案设计与优化; 4.与分布式训练、系统、模型算法团队密切协作,共同提升大规模训练任务的整体效率与稳定性; 5.跟踪业界前沿的硬件架构与系统软件(GPU 架构、网络、编译器、库等),将最新技术转化为实际性能收益。
前往官网申请 →