T

TikTok

Short-form video platform with ads

Graduate Machine Learning Engineer - E-Commerce Recommendation Video

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USAIn-person

About the Job

Requirements

  • Complete or recently complete a Bachelor's degree in Computer Science, Electrical Engineering, Mathematics, Statistics, or a related discipline.
  • Have solid fundamentals in algorithms and data structures and strong coding ability, including writing clean, efficient, and reproducible code.
  • Have solid foundations in machine learning, probability, and statistics, including understanding model assumptions, limitations, and why methods work or fail.
  • Be fluent in a deep-learning framework such as PyTorch or TensorFlow and at least one large-scale data-processing tool such as Spark, Flink, or SQL.
  • Be proficient in Python and have working skills in C++ and Linux.
  • Be able to investigate the causes of metric changes and understand A/B test design, statistical confidence, and common experimental pitfalls.
  • Commit to an onboarding date by the end of the year and state availability and graduation date clearly in the resume.

Responsibilities

  • Optimize recommendation models across the full funnel for TikTok Shop products, short videos, and livestreams, covering retrieval, pre-ranking, ranking, re-ranking, and multi-queue blending, and ship iterations to real traffic.
  • Iterate ranking and retrieval model architectures, including multi-task and multi-objective learning from clicks through conversion and GMV; multi-scenario and multi-format joint modeling; and sample, label, and debiasing design.
  • Model user interests in ultra-long behavior sequences, balancing positional-encoding extrapolation, attention cost, online latency, and storage; separate stable preferences from seasonal demand, momentary impulses, and satisfied needs; and mine implicit negative feedback throughout the funnel.
  • Combine product images, text, and video content with behavioral data to create high-quality item and user representations.
  • Make list-level re-ranking and blending decisions across content types, control diversity and repetition, and develop exploration mechanisms that surface unexpressed interests while reducing recommendation feedback loops.
  • Design cold-start and content-ecosystem mechanisms for new products, livestream hosts, and creators.
  • Conduct original work on recommendation problems including long-term value modeling, repurchase and retention, transaction attribution, fatigue modeling, new-user recommendation, incremental value modeling, and LLM4Rec.

Desired Qualifications

  • Research or engineering experience in recommendation, search, advertising, information retrieval, natural language processing, or large-scale machine-learning systems.
  • Publications at KDD, NeurIPS, WWW, SIGIR, WSDM, ICML, ICLR, RecSys, CIKM, or comparable venues, or high-quality open-source contributions.
  • Awards in Kaggle, Tianchi, or RecSys Challenge, or an ACM-ICPC or NOI competition background.
  • Hands-on experience with causal inference, online learning or bandits, graph neural networks, sequence modeling, or large-scale distributed training.
  • Experience using AI coding and agentic workflows to build systems and optimize models.

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