Machine Learning Engineer — AI Search & Generative Recommendations

TikTok USDS Joint Venture; San Jose (CA); On-site; USD 137,000 - 360,000; Full time

Job summary

TikTok USDS Joint Venture LLC is seeking an independent developer for the Search team to advance its scalable, personalized search and recommendation systems using cutting-edge ML, LLMs and multi-modal tech.

The role emphasizes independent execution, LLM integration in the recommendation pipeline, and collaboration with product managers, data scientists and infrastructure engineers to deliver engaging user experiences for a broad global audience.

Qualifications

• Bachelor's or advanced degree in Computer Science, ML, or a related field.
• Proficient in ML frameworks (TensorFlow, PyTorch) and programming languages (Python, Java, C++).
• 2+ years in developing large-scale search or recommendation systems.
• Hands-on experience deploying LLMs for real-world applications, incl. prompt engineering & RLHF.

Responsibilities

• Independent development, experimentation, analysis and deployment of large-scale personalized search & recommendation systems.
• Research and integrate LLMs to enhance content understanding, user intent, and personalization.
• Explore generative recommender approaches (GenRec) with LLMs/VLMs to generate relevant recommendations.
• Collaborate with product, data science & infra teams to enable search personalization strategies.

Skills

ML frameworks

Python

Java

C++

Large-scale search

LLM experience

Education

Bachelor's or advanced degree in Computer Science / ML

Tools

TensorFlow

PyTorch

Job description

TikTok USDS Joint Venture LLC is seeking an independent developer for the Search team to advance its scalable, personalized search and recommendation systems using cutting-edge ML, LLMs and multi-modal tech.

The role emphasizes independent execution, LLM integration in the recommendation pipeline, and collaboration with product managers, data scientists and infrastructure engineers to deliver engaging user experiences for a broad global audience.