Drimva

TikTok

Software Engineer Intern (ML Infra) - 2027 Start (PhD)

San Jose, California · Internship · Posted Oct 28, 2026

Education
Currently pursuing a PhD degree in Computer Science, AI, Machine Learning or a related discipline.

About the job

Build and optimize infrastructure, data pipelines, and machine-learning platforms for TikTok’s advertising recommendation systems.

What you'll do

  • Develop and optimize advertising recommendation models and large-scale data pipelines.
  • Build ML platforms for distributed training, model serving, and lifecycle management.
  • Explore MLLM applications and AI agents for advertising workflows and operations.

What they ask for

  • Currently pursuing a PhD in Computer Science, AI, Machine Learning, or a related discipline.
  • Proficiency in at least one programming language, such as C++, Python, Go, or Java.
  • Solid knowledge of data structures, algorithms, machine learning, and computer systems.
  • Familiarity with recommendation systems, MLLMs, distributed systems, ML platforms, or large-scale data processing.

Nice to have

  • Experience with deep learning frameworks, distributed training, GPU scheduling, AI agents, or cloud-native technologies.

Tools

C++, Python, Go, Java

AI focus

Develop advertising recommendation models; explore MLLM data production, training, fine-tuning, and inference optimization; apply AI agents to workflow orchestration, resource scheduling, failure diagnosis, and automated operations.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)