T

TikTok

Short-form video platform with ads

Machine Learning Engineer Intern

AI ResearchPosted on Nov 11
InternshipSan Jose, CA, USAIn-person

About the Job

Requirements

  • Currently pursuing an undergraduate degree in Software Development, Computer Science, Computer Engineering, or a related discipline.
  • Have a solid foundation in machine learning and deep learning, with research or project experience in at least one of the following areas: large language models, reinforcement learning, generative models, recommender systems, or information retrieval.
  • Be proficient in Python and at least one mainstream deep learning framework, such as PyTorch, TensorFlow, or JAX.
  • Have strong problem-solving skills and an interest in tackling complex, open-ended research problems.

Responsibilities

  • Drive the evolution of recommender systems from discriminative to generative paradigms; explore generative retrieval and generative re-ranking or blending; improve personalization while optimizing generative-model training and inference efficiency on GPUs.
  • Use large language models, reinforcement learning, and related techniques to improve recommender-system semantic understanding and reasoning, addressing e-commerce challenges such as cold start, long-tail item distribution, and user intent understanding.
  • Explore constructing self-evolving agents and using agents to optimize recommender systems; advance agentic architectures that perceive user context and make real-time, personalized decisions and adjustments.
  • Explore replacing traditional heuristic rule-based systems with large language model and agent capabilities; build algorithms to measure and optimize long-term value and user experience, supporting sustainable platform ecosystem growth.

Desired Qualifications

  • Have experience in recommendation systems, especially in live commerce, e-commerce, search, advertising, or other large-scale consumer products.
  • Have experience with generative recommendation, large recommendation models, retrieval and ranking systems, or related recommendation architecture upgrades.
  • Have experience with large language models or multimodal foundation models, including pre-training, post-training, representation learning, contrastive learning, supervised fine-tuning, or reinforcement-learning-based optimization.
  • Have experience in cross-domain transfer learning, long-term value modeling, long-term value optimization, causal inference, or debiasing.
  • Have experience with long-sequence user behavior modeling, multi-task learning, multi-interest modeling, or large-scale distributed training and inference optimization.
  • Have publications in top-tier conferences such as NeurIPS, ICML, ICLR, KDD, ACL, CVPR, SIGIR, or RecSys, or strong achievements in major technical competitions.

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs