Machine Learning Engineer, AI Search (ByteDance Singapore)

ByteDance
Singapore
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningSkills: ["Problem-solving","Communication","Collaboration"]

Build LLM-powered AI Search and AI assistant experiences for real-time question-answering. You’ll develop AI search agents with planning, memory, and tool-use, improve multi-turn query understanding and semantic analysis, and optimize RAG and LLM pipelines for retrieval, ranking, and answer generation (fine-tuning and inference). Partner with cross-functional teams to turn agent/LLM research into production upgrades.

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: 31 days agoStatus: Live
Reposted: similar role first listed 1 month ago

Job Summary

Build LLM-powered AI Search and AI assistant experiences for real-time question-answering. You’ll develop AI search agents with planning, memory, and tool-use, improve multi-turn query understanding and semantic analysis, and optimize RAG and LLM pipelines for retrieval, ranking, and answer generation (fine-tuning and inference). Partner with cross-functional teams to turn agent/LLM research into production upgrades.
Location: Singapore
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

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, performing semantic analysis on queries and documents, and ensuring accurate search results via LLM reasoning.
  • •Optimize RAG and LLM pipelines by improving retrieval and ranking algorithms and training/optimizing LLMs for answer generation (fine-tuning and inference optimization).
  • •Improve relevance and structured presentation of search results produced by the LLM pipeline.
  • •Translate agent/LLM research into production by collaborating with cross-functional teams to drive product upgrades.

Key Requirements

  • •Proficient in C++ and/or Python, with strong command of algorithms and data structures (e.g., retrieval, graph algorithms).
  • •Basic understanding of LLM fundamentals, including prompt engineering and fine-tuning workflows.
  • •Ability to solve complex problems and deliver technical projects end-to-end.
  • •Hands-on experience with LLM/agent technologies such as RAG optimization, RLHF/SFT, or agent framework development (planning/memory modules).
  • •Strong communication and collaboration skills, with a track record of exploring new technologies and driving technical advancements with teams.
Skills:Problem-solvingCommunicationCollaboration
Tech Stack:C++PythonLLMLLM ChatBotAgentsRAGReinforcement learningKnowledge representationPrompt engineeringFine-tuningInference optimizationRLHFSFTRetrieval algorithmsGraph algorithmsSemantic analysis

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