机器学习研发工程师-搜索架构

关于这个机会

1、参与抖音&国际化短视频搜索场景下机器学习和深度学习架构的设计与实现； 2、关注文本相关性模型、多模态、Wide&Deep模型、LLM模型等深度学习模型在搜索场景中的性能和成本优化； 3、提高模型训练效率，加速推理性能，优化GPU吞吐； 4、设计和开发高性能算子库，研究最新的GPU计算技术和优化方法，充分利用GPU并行计算能力，支持各种模型在不同硬件上的构建和部署。

任职要求

1、掌握扎实的计算机基础知识，深入理解数据结构、算法和操作系统知识； 2、有强烈的求知欲，快速学习和沟通能力； 3、编程能力扎实，有良好的编程习惯。 加分项： 1、熟悉主流的深度学习框架（Pytorch、Tensorflow）； 2、熟悉主流高性能并行编程技术，有GPU编程经验（CUDA或Triton）； 3、熟悉vLLM，TRT-LLM等大模型推理框架，熟悉大模型并发推理原理； 4、熟悉FSDP、DeepSpeed、Megatron等业界主流分布框架的使用和原理。

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。