AI Infrastructure Engineer Graduate (TikTok Recommendation Architecture) - 2027 Start
Core
Build and optimize distributed training and inference systems for TikTok's recommendation engine and Large Language Models (LLMs) at massive scale.
Role type
Graduate IC AI Infrastructure Engineer
Builds
Distributed training and serving infrastructure for recommendation models and LLMs
Domain
AI Infrastructure / Large-Scale Systems / GPU Computing
Deliverable
infrastructure
Required skills
C++, Python, data structures, algorithms, computer systems, PyTorch, TensorFlow, Transformer architectures, LLMs
Preferred skills
LLM training/inference experience, distributed training (DP, TP, PP, FSDP, ZeRO), GPU programming (CUDA, Triton), LLM serving (KV Cache, Continuous Batching, FlashAttention), open-source contributions
Technologies
PyTorch, TensorFlow, CUDA, Triton
Responsibilities
Build and optimize infrastructure for large-scale model training and online inference; Develop distributed systems supporting large recommendation models and LLMs; Improve training and inference performance through GPU optimization and efficient communication; Collaborate with researchers to develop and deploy LLM training and serving solutions; Analyze system bottlenecks and implement performance optimizations
Seniority
New Grad, hands-on IC