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