高性能计算研发工程师-Data语音
字节跳动
北京校招字节跳动校园招聘(校招应届)
岗位职责
1、负责构建新一代大模型推理引擎,优化多模态语音理解/语音生成等多模态生成大模型在GPU集群上的推理性能,实现低延迟、高吞吐的工业级部署;
2、主导模型推理性能优化,涵盖CUDA/Triton算子开发、vLLM/SGLang框架升级、分布式推理策略优化、量化/稀疏化等模型efficiency加速优化等;
3、研发GPU推理加速技术栈,设计最佳分布式通算结合方案,优化PCIe通信与高并发推理架构;
4、负责高性能方案前瞻性建设,构建基于C++/Python研发的高性能推理系统;
5、与上下游部门深度合作,分析性能瓶颈,通过软硬结合提升模型训推效率,优化和部署语音大模型,支持AI工具链和技术生态建设,推动字节跳动AI关键业务发展;负责语音多模态场景下高性能推理系统的开发,支撑各业务场景下的性能优化需求并推动业务落地。
任职要求
1、精通Python,熟悉C++特性,具备高性能代码开发能力和相关经验;
2、至少具备以下一个领域经验:GPU编程(CUDA/Triton/AscendC/TileLang开发)、模型量化/稀疏化/蒸馏、基于vLLM的框架研发,并行计算的通算结合(多卡/多机推理优化);
3、符合以下条件者优先:拥有推理系统经验、具备vLLM/SGLang开发经验、Tilelang/Tritton开发经验,深入了解Transformer架构,有量化/稀疏化等相关技术落地或者论文发表等相关经验。