Skip to main contentA logo with &quat;the muse&quat; in dark blue text.

AI Machine Learning Engineer, TikTok Search

3 days ago Singapore

Responsibilities

Our Search Team is responsible for building and owning TikTok's search engine which provides our users the best search experience. On the TikTok Search Team, you'll have the opportunity to build a full-stack search engine system and combine information retrieval technology with modern machine learning methods from related fields such as NLP, Computer Vision, Multimodal, and Recommender Systems. We embrace a culture of self-direction, intellectual curiosity, openness, and problem-solving.

Responsible for the Search Top-1 Card and AI Assistant features in our product. Leverage in-platform content and users' historical conversations as contextual inputs to deliver a personalized, highly authentic tool-based product through conversational interaction.In the conversational search paradigm, accurately identify user intent by combining real-time dialogue context and historical chat memory. Integrate high-quality in-platform UGC content with authoritative global knowledge to generate authentic, traceable AI-powered results.

Core Responsibilities:
1. End-to-End AI Agent System Construction: Lead the full lifecycle delivery of AI Agent applications, from offline MVP prototyping to production-grade closed-loop systems. Own the design and implementation of foundational data pipelines, core workflow architectures, service execution layers, and observability frameworks to ensure system scalability, stability, and performance.
2. Business Outcome-Driven Engineering Iteration: Drive algorithmic and strategic optimizations aligned with business objectives. Oversee core business pipeline launches and iterations via offline batch runs, A/B testing, and online canary deployments, continuously improving user experience, system efficiency, and key business metrics.
3. Quality Assurance & Evaluation System Establishment: Build end-to-end quality validation mechanisms covering data processing, reasoning, and generation. Develop offline evaluation frameworks, bad case taxonomies, and automated assessment pipelines to guarantee the accuracy, consistency, and compliance of business outputs.
4. Intelligent Capability Evolution & Cross-Scenario Expansion: Own the iteration and optimization of core intelligent capabilities including multi-step reasoning, RAG, and personalized memory systems. Drive the reuse and deployment of Agent technologies across vertical business scenarios to support long-term intelligent business upgrades.

Want more jobs like this?

Get Data and Analytics jobs in Singapore delivered to your inbox every week.

Job alert subscription


Qualifications

Minimum Qualifications
1. Bachelor's degree or above in a computer-related major, with excellent coding ability and proficient in programming languages such as C++/Python, and familiar with common data structures and programming algorithms.
2. Proficient in C++/Python; strong command of algorithms and data structures (e.g., retrieval, graph algorithms).
3. Basic understanding of LLM fundamentals (e.g., prompt engineering, fine-tuning workflows).
4. Strong problem-solving skills with a track record of delivering technical projects.

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

Client-provided location(s): Singapore
Job ID: TikTok-7633426598587173173
Employment Type: OTHER
Posted: 2026-04-27T20:46:45

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.