机器学习研发工程师-搜索架构
Core
Design and implement machine learning and deep learning architectures for TikTok's domestic and international short video search scenarios, focusing on text relevance, multi-modal, Wide&Deep, and LLM models.
Role type
Senior IC machine learning engineer (search architecture)
Builds
High-performance search models and operator libraries for short video platforms
Domain
Internet / Search / Deep Learning
Deliverable
production ML models
Required skills
Computer science fundamentals, data structures, algorithms, operating systems, deep learning frameworks (PyTorch, TensorFlow), GPU programming (CUDA, Triton), distributed training frameworks (FSDP, DeepSpeed, Megatron), large model inference (vLLM, TRT-LLM)
Preferred skills
GPU throughput optimization, high-performance operator library design, parallel programming techniques
Technologies
PyTorch, TensorFlow, CUDA, Triton, vLLM, TRT-LLM, FSDP, DeepSpeed, Megatron
Responsibilities
Optimize model training efficiency and inference performance, design high-performance operator libraries, research GPU computing technologies for model deployment
Seniority
Senior, hands-on IC