Machine Learning Engineer (AI Search ) - TikTok Search

Machine Learning Engineer (AI Search ) - TikTok Search at TikTok in Singapore.

Responsibilities

Our Search Team focuses on LLM application algorithms for TikTok Search, with core scenarios including AI Search and TikTok AI Bot. We build search question-answer cards via LLM-powered generation and develop TikTok's official AI assistant, leveraging LLM ChatBot, Agent, and RAG technologies to deliver accurate, real-time responses. By closely integrating technical innovation with business scenarios, we bridge users with information and services, driving TikTok's business growth.

• Build AI Search Agents with planning, memory, and tool-use capabilities. Apply reinforcement learning, knowledge representation, and automated decision-making to solve complex problems end-to-end.
• Enhance LLM-based query understanding: process multi-turn conversations, conduct semantic analysis on queries and documents, and ensure accurate, objective search results via LLM reasoning.
• Optimize RAG and LLM pipelines: improve retrieval and ranking algorithms, train and optimize LLMs for answer generation (e.g., fine-tuning, inference optimization), and enhance the relevance and structured presentation of search results.
• Translate research (e.g., Agent/LLM innovations) into production, collaborating with cross-functional teams to drive product upgrades.

Requirements

• 5 years of experience in Machine Learning, preferably in a fast-paced, high-growth environment.
• Proficient in C++/Python; strong command of algorithms and data structures (e.g., retrieval, graph algorithms).
• Basic understanding of LLM fundamentals (e.g., prompt engineering, fine-tuning workflows).
• Strong problem-solving skills with a track record of delivering technical projects.

Preferred Qualifications

• Hands-on experience with LLM/Agent technologies: RAG optimization, RLHF/SFT for LLMs, or Agent framework development (e.g., planning/memory modules).
• Excellent communication and collaboration skills, with a track record of exploring new technologies and driving technical advancements with teams.

About the Company

The Global Business Solutions (GBS) team is responsible for the revenue growth of the TikTok business, and our teams include Sales, Marketing, Ops, Account Managers, Agency and partnerships, as well as Marketing Science.

At TikTok, our Global Business Solutions (GBS) team plays a key role in generating revenue by promoting our advertising solutions, onboarding new clients, driving ad campaigns, and more. As the TikTok community grows at an unprecedented speed around the world, our GBS team leads groundbreaking projects that are changing the landscape of the advertising industry in real time.

We're seeking an analytically driven, and detail-oriented Client Solutions Manager (CSM) Intern to join our Ecommerce Team. As a CSM, you will partner closely with Client Partners and Client Solutions Managers to drive revenue by identifying opportunities, leveraging data insights, and delivering consultative solutions for advertisers.

This role centers on client education, relationship growth, data analysis, and campaign success. You will provide strategic recommendations to both clients and internal teams, ensuring campaigns achieve business objectives while optimizing long-term partnerships. Success in this role requires strong data analytics skills, adaptability in a fast-paced environment, and a test-and-learn mindset to uncover the best solutions.