T

TikTok

Short-form video platform with ads

Infrastructure Engineer Intern - TikTok Recommendation Architecture

AI ResearchConfirmed live in the last 24 hours
Internship · Summer 2027SingaporeNo salary listedIn-person

About the Job

Requirements

  • Currently pursuing a PhD in Computer Science, engineering, or a quantitative field.
  • Strong programming skills in C++, CUDA, and Python, with a solid understanding of deep learning frameworks such as PyTorch and TensorFlow.
  • Published research in systems, machine learning, or related areas at venues including SOSP, OSDI, NSDI, ICML, NeurIPS, ICLR, or equivalent conferences.
  • Must commit to an internship lasting at least three months.

Responsibilities

  • Conduct novel research on distributed training and inference system optimization for large-scale recommendation models and large language models.
  • Design and implement high-performance GPU kernel architectures and communication primitives to accelerate deep learning workloads.
  • Publish original research findings at top-tier academic conferences and collaborate with cross-functional teams to translate research into production impact.

Desired Qualifications

  • Hands-on experience with distributed training frameworks such as DeepSpeed, FSDP, or Megatron-LM, or GPU kernel optimization.
  • Experience with large language model inference/serving systems such as vLLM, TensorRT-LLM, or SGLang.
  • Strong background in parallel computing, computer architecture, or machine learning systems.

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs