Machine Learning Engineer, AI Search - USDS

TikTok's USDS Search team is hiring a Machine Learning Engineer to build and optimize large-scale personalized search and recommendation systems. The role focuses on integrating LLMs and generative recommendation approaches to enhance content understanding, user intent recognition, and personalization for over 1 billion monthly users. The engineer will collaborate with cross-functional teams and contribute to the full lifecycle of AI model development, from experimentation to deployment. This position requires on-site presence in San Jose, CA.

• Independently develop, experiment, analyze, and deploy large-scale personalized search and recommendation systems.
• Research and integrate LLMs within the recommendation pipeline to enhance content understanding, user intent recognition, personalization, and content generation.
• Drive generative recommendation approaches, leveraging models like LLMs/VLMs to generate personalized content recommendations beyond traditional ranking methods.
• Collaborate with product managers, data scientists, infrastructure engineers, and operations teams to align search personalization with product goals.

• Bachelor's or advanced degree in Computer Science, Machine Learning, or a related field.
• Proficiency in machine learning frameworks (e.g., TensorFlow, PyTorch) and programming languages (e.g., Python, Java, C++).
• 2+ years of experience developing large-scale search or recommendation systems (feed, search, advertisement, etc.).