ML Engineer - Generative Search & LLM Integration

TikTok Inc.; San Jose (CA); On-site; USD 137,000 - 360,000; Full time

Benefits offered by this job

Medical, dental, vision insurance

401(k) with company match

Paid parental leave

Short/long-term disability coverage

Life insurance

Wellbeing benefits

10 paid holidays per year

10 paid sick days per year

Paid Personal Time

Job summary

TikTok USDS Joint Venture LLC is seeking an independent developer for the Search team to advance TikTok's search capabilities by integrating cutting-edge recommendation algorithms and large language models (LLMs). This role emphasizes scalable, robust systems and close collaboration with product managers and engineers.

Responsibilities include building scalable search and recommendation pipelines, integrating LLMs, exploring generative approaches, and partnering with cross-functional teams to

Qualifications

• Bachelor's degree in CS, ML, or related field.
• Proficient with ML frameworks and programming languages.
• 2+ years in large-scale search or recommendation systems.
• Hands-on experience deploying LLMs in real-world applications.

Responsibilities

• Independent development and deployment of large-scale personalized search and recommendation systems.
• Research and integrate LLMs within the recommendation pipeline to enhance understanding and personalization.
• Explore generative recommendation approaches using GenRec-like frameworks with LLMs/VLMs.
• Collaborate with product, data science, and infrastructure teams to implement search personalization strategies.

Skills

Large-scale ML

Python

C++

Java

TensorFlow

PyTorch

Education

Bachelor's or higher in CS/ML

Tools

TensorFlow

PyTorch

Python

Java

C++

Job description

TikTok USDS Joint Venture LLC is seeking an independent developer for the Search team to advance TikTok's search capabilities by integrating cutting-edge recommendation algorithms and large language models (LLMs). This role emphasizes scalable, robust systems and close collaboration with product managers and engineers.

Responsibilities include building scalable search and recommendation pipelines, integrating LLMs, exploring generative approaches, and partnering with cross-functional teams to