Machine Learning Engineer - AI Search (ByteDance Singapore)

ByteDance
Singapore
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningExperience: 5+ yearsSkills: ["Problem-solving","Communication","Collaboration"]

Build and improve AI Search and AI Bot experiences using LLM-powered agents, including planning, memory, and tool-use. Work on query understanding for multi-turn conversations and semantic analysis to ensure accurate, objective results. Optimize RAG and LLM pipelines with retrieval/ranking improvements and fine-tuning/inference optimization. Translate agent/LLM research into production with cross-functional teams to deliver upgraded products.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
ByteDance
ByteDance
1 month ago

Machine Learning Engineer - AI Search (ByteDance Singapore)

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 30 days agoStatus: Live
Reposted: similar role first listed 1 month ago

Job Summary

Build and improve AI Search and AI Bot experiences using LLM-powered agents, including planning, memory, and tool-use. Work on query understanding for multi-turn conversations and semantic analysis to ensure accurate, objective results. Optimize RAG and LLM pipelines with retrieval/ranking improvements and fine-tuning/inference optimization. Translate agent/LLM research into production with cross-functional teams to deliver upgraded products.
Location: Singapore
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Build AI Search agents with planning, memory, and tool-use capabilities, applying reinforcement learning and knowledge representation to solve problems end-to-end.
  • •Enhance LLM-based query understanding by handling multi-turn conversations and performing semantic analysis for accurate, objective search results.
  • •Optimize RAG and LLM pipelines by improving retrieval and ranking, and training/optimizing LLMs for answer generation.
  • •Improve the relevance and structured presentation of search results.
  • •Translate agent/LLM research into production by collaborating with cross-functional teams to drive product upgrades.

Key Requirements

  • •5+ years of experience in machine learning in a fast-paced, high-growth environment.
  • •Proficiency in C++ and/or Python, with strong algorithms and data structures skills (e.g., retrieval, graph algorithms).
  • •Basic understanding of LLM fundamentals such as prompt engineering and fine-tuning workflows.
  • •Strong problem-solving skills with a track record of delivering technical projects.
  • •Hands-on experience with LLM/agent technologies such as RAG optimization, RLHF/SFT, or agent framework development.
Experience:5+ yearsLLMRAGMachine learning
Skills:Problem-solvingCommunicationCollaboration
Tech Stack:C++PythonLLMAI SearchAI BotLLM ChatBotAgentRAGPrompt engineeringFine-tuningInference optimizationRLHFSFT

Company Brief

ByteDance
Develops consumer internet and content platforms, including TikTok and other apps for short-form video, news, and entertainment. It also builds advertising, commerce, and creator tools that connect audiences, brands, and publishers across global markets.
Industry: Digital Media
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Established Company
Headquarters: Beijing, China
Founded: 2012
WebsiteLinkedIn