机器学习研发工程师 - TikTok Search

关于这个机会

团队介绍：TikTok Search 团队面向TikTok搜索业务，支撑海量流量的高并发、高性能、高可用搜索基础架构与业务中台。我们长期深耕搜索引擎检索、相关性/个性化排序、算法服务与深度学习工程，打造了覆盖多地区、多数据中心的搜索系统架构。面向大模型时代，我们将生成式搜索、SVLM相关性排序、个性化推荐、多模态理解等多项前沿技术深度融入搜索全链路，持续构建高效、稳定、可扩展的大模型基础设施，同时也积极探索新的搜索范式如视搜、AI搜，和使用 AI Coding 大幅加速搜索系统重构与演进，为TikTok搜索业务提供强有力且可持续演进的技术基座。 1、参与TikTok搜索场景下，机器学习和深度学习架构的设计与实现； 2、关注在高吞吐、大规模场景下的文本相关性模型、多模态、LLM模型、生成式召回等深度学习模型，在搜索场景中的性能和成本优化； 3、提高模型训练效率，加速推理性能，优化GPU吞吐； 4、设计和开发高性能算子库，研究最新的GPU计算技术和优化方法，充分利用GPU并行计算能力，支持各种模型在不同硬件上的构建和部署。

任职要求

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

内推申请说明

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