腾讯集团
微信WeLM-大语言模型推理优化
RESPONSIBILITIES
岗位职责
WeLM 模型在模型结构上的创新,诸如Hidden Decoding/NGram Embedding等,对模型推理加速部分有算法创新上的要求。特别是在RL场景中,LowBatch推理降低推理Latency也制约了团队训练迭代的速度。WeLM推理会使用诸如整个模型单MegaKernel推理,MTP算法的创新性工作来加速推理场景。
REQUIREMENTS
任职要求
有推理框架开发经验,熟悉LLM推理的基本技巧与并行方式;
推理场景于高性能算子开发, 熟悉Hopper显卡的微架构;
了解RDMA/NCCL GIN/NVSHMEM等通信算子开发;
沟通协作能力强,有自驱力好奇心。
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。