Maching Learning Engineer (AI Search ) - TikTok Search
Responsibilities
About the team
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.
Responsibilities:
- 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.
Want more jobs like this?
Get Software Engineering jobs in Singapore delivered to your inbox every week.
- 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.
Qualifications
Minimum Qualifications
- 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.
Perks and Benefits
Health and Wellness
- Health Insurance
- Dental Insurance
- Vision Insurance
- HSA
- Life Insurance
- Fitness Subsidies
- Short-Term Disability
- Long-Term Disability
- On-Site Gym
- Mental Health Benefits
- Virtual Fitness Classes
Parental Benefits
- Fertility Benefits
- Adoption Assistance Program
- Family Support Resources
Work Flexibility
- Flexible Work Hours
- Hybrid Work Opportunities
Office Life and Perks
- Casual Dress
- Snacks
- Pet-friendly Office
- Happy Hours
- Some Meals Provided
- Company Outings
- On-Site Cafeteria
- Holiday Events
Vacation and Time Off
- Paid Vacation
- Paid Holidays
- Personal/Sick Days
- Leave of Absence
Financial and Retirement
- 401(K) With Company Matching
- Performance Bonus
- Company Equity
Professional Development
- Promote From Within
- Access to Online Courses
- Leadership Training Program
- Associate or Rotational Training Program
- Mentor Program
Diversity and Inclusion
- Diversity, Equity, and Inclusion Program
- Employee Resource Groups (ERG)
Company Videos
Hear directly from employees about what it is like to work at TikTok.