Drimva

TikTok

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

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

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

About the job

Research and develop generative, LLM-based, and agentic recommendation systems for TikTok e-commerce, improving personalization, efficiency, long-term value, and user experience.

What you'll do

  • Research generative recommendation methods, including retrieval and re-ranking, and optimize model training and inference on GPUs.
  • Apply LLMs, reinforcement learning, and agents to improve recommendation understanding, reasoning, and real-time personalization.
  • Develop algorithms for long-term value and user experience modeling in e-commerce.

What they ask for

  • Currently pursuing a PhD in Computer Science, Electrical Engineering, Mathematics, Statistics, or a related discipline.
  • Foundation in machine learning and deep learning, with research or project experience in at least one of: large language models, reinforcement learning, generative models, recommender systems, or information retrieval.
  • Proficient in Python and at least one mainstream deep learning framework.
  • Strong problem-solving skills and interest in complex, open-ended research problems.

Nice to have

  • Experience in recommendation systems or related large-scale consumer products; generative recommendation, retrieval, ranking, or recommendation architecture upgrades.
  • Experience with LLMs or multimodal foundation models, long-term value modeling, causal inference, user behavior modeling, distributed training, or relevant publications or competition achievements.

Tools

Python, PyTorch, TensorFlow, JAX, LLMs, RL, GPUs

AI focus

Research and build generative, LLM-powered, and agentic recommendation systems, including improving model efficiency and optimizing long-term value and user experience.

Model-written summary · the original posting is authoritative

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