Drimva

TikTok

Machine Learning Engineer Intern (E-Commerce Recommendation Video) - 2027 Start (PhD)

Seattle, Washington · Internship · Posted Oct 28, 2026

Education
Currently pursuing a PhD in Computer Science, Electrical Engineering, Mathematics, Statistics, or a related discipline.

About the job

Build and research large-scale e-commerce recommendation systems for TikTok Shop, spanning retrieval, ranking, generative recommendation, model training, and inference.

What you'll do

  • Develop and scale recommendation models and generative retrieval for video, livestream, and product recommendations.
  • Optimize recommendation training and inference for performance and latency, including distributed systems and hardware-level acceleration.
  • Conduct original research on recommendation problems and build agent-assisted R&D workflows.

What they ask for

  • Currently pursuing a PhD in Computer Science, Electrical Engineering, Mathematics, Statistics, or a related discipline.
  • Strong machine-learning and engineering fundamentals, including algorithms and data structures and the ability to write clean, efficient, reproducible code.
  • Deep research or engineering practice in at least one of LLMs/foundation models, NLP, CV, RL, recommendation, search, or ads.
  • Enthusiasm for bringing LLM/LRM techniques into production and ability to define and decompose ambiguous problems.

Nice to have

  • Publications at listed venues or comparable venues, or high-quality open-source work.
  • Experience in CUDA/Triton, distributed training, high-performance inference, LLM post-training, agent systems, or relevant end-to-end projects.

Tools

LLMs, CUDA, Triton, RQ-VAE, SID, GRPO, SFT, RLHF

AI focus

Build LLM-based recommendation systems, including generative retrieval, large-scale recommendation models, LLM post-training, reasoning over user intent, and agent-assisted research and development.

Model-written summary · the original posting is authoritative

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