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

职位描述

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等业界主流分布框架的使用和原理。