Drimva

TikTok

Machine Learning Engineer Graduate (E-Commerce Content Recommendation - Generative & Large Recommendation Model) - 2027 Start (PhD)

San Jose, California · Full-time · Posted Oct 28, 2026

Education
PhD in Computer Science, AI, Mathematics, Statistics, or a related discipline; completing or recently completed.

About the job

Build and research production-scale generative recommendation systems for TikTok Shop content, spanning retrieval, ranking, and serving.

What you'll do

  • Develop and scale recommendation models and generative retrieval for short-video, livestream, and product scenarios.
  • Optimize model training and inference, including kernels, distributed systems, and millisecond-latency serving.
  • Research recommendation problems and build agent-assisted R&D workflows; deploy iterations to production.

What they ask for

  • Completing or recently completed a PhD in Computer Science, AI, Mathematics, Statistics, or a related discipline.
  • Strong machine-learning and engineering fundamentals, including algorithms, data structures, and clean, efficient, reproducible code.
  • Deep research or engineering practice in at least one of LLMs/foundation models, NLP, CV, RL, or recommendation/search/ads.
  • Genuine enthusiasm for production use of LLM/LRM techniques 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 related projects.

Tools

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

AI focus

Build LLM-based and generative recommendation models, apply large-model knowledge and reasoning to user-interest discovery, and use coding agents to automate parts of R&D.

Model-written summary · the original posting is authoritative

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