机器学习研发工程师-搜索架构
Core
Design and implement machine learning and deep learning architectures for short video search scenarios on Douyin and international platforms.
Role type
Senior IC machine learning engineer (search architecture)
Builds
High-performance search models and operator libraries for Douyin and international 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 frameworks (vLLM, TRT-LLM)
Preferred skills
GPU throughput optimization, high-performance operator library design, model performance and cost optimization
Technologies
PyTorch, TensorFlow, CUDA, Triton, vLLM, TRT-LLM, FSDP, DeepSpeed, Megatron
Responsibilities
Design and implement ML/DL architectures for search, optimize model performance and cost, improve training efficiency and inference speed, develop high-performance operator libraries
Seniority
Senior, hands-on IC