Machine Learning Engineer, Recommendation - E-Commerce

TikTok is seeking a Machine Learning Engineer for E-Commerce recommendations to advance scalable optimization, modeling, and tooling for the TikTok Shop recommendation system. This onsite role is based in San Jose, California, with a salary range of USD 150,000 to 316,800 per year.

Compensation

Annual salary range: USD 150,000 to 316,800.

Responsibilities

• Design and implement optimization strategies for a large-scale e-commerce recommendation pipeline that handles tens of millions of products or creators' content.
• Develop long and short term user interest models by analyzing diverse data and creating algorithms to efficiently uncover latent interests.
• Prototype, evaluate, and iterate predictive models for candidate generation and ranking, including real-time data pipelines, feature engineering, model optimization, and innovative approaches for CTR and CR prediction.
• Create supporting and debugging tools as needed.

Requirements

• Bachelor's degree in computer science or a related field.
• At least 3 years of experience with a strong foundation in data structures and algorithm design, proficient in programming languages such as Python, Java, C++, or R.
• Familiarity with common machine learning, deep learning, causal inference, and operational optimization algorithms, including classification, regression, clustering methods, as well as mathematical programming and heuristic methods.
• Experience with at least one framework such as TensorFlow, PyTorch, or MXNet, including training and deployment details, as well as training acceleration methods like mixed precision and distributed training.
• Exposure to big data frameworks; familiarity with MapReduce or Spark is preferred.

Technologies

• Python, Java, C++, R
• TensorFlow, PyTorch, MXNet
• MapReduce, Spark

Benefits

• Medical, dental, and vision insurance
• 401(k) savings plan with company match
• Paid parental leave
• Short-term and long-term disability coverage
• Life insurance
• Wellbeing benefits
• 10 paid holidays per year
• 10 paid sick days per year
• 17 days of Paid Personal Time, prorated upon hire with tenure-based accruals

About the Teams

• E-commerce - Recommendation and Marketing
• E-commerce - Alliance
• E-commerce - Search
• E-commerce - Search Growth

Notice for Los Angeles County applicants

Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act.

• Criminal history may negatively impact duties such as interacting with internal or external clients and colleagues, sometimes in unsupervised contexts.
• Criminal history may affect the handling of confidential information and access to information technology systems.
• The position involves exercising sound judgment in performing responsibilities.