Software Engineer Graduate (TikTok AI Search & Visual Search Infra Team) - 2027 Start

TikTok is a short-form mobile video platform seeking a graduate Software Engineer for its AI and Visual Search Infrastructure team. The role involves building large-scale search infrastructure, AI agent and RAG platforms, LLM/VLM training and inference systems, distributed data pipelines, and multimodal search capabilities.

Responsibilities:

• Design, build, and optimize large-scale infrastructure for TikTok AI Search & Visual Search, supporting billions of online search requests and large-volume offline data processing
• Evolve TikTok’s search agent platform, including multi-agent orchestration, ReAct and tool-calling workflows, RAG indexing & serving engine, and context/memory management
• Deploy and optimize LLM/VLM post-training and inference, including GPU inference acceleration, SFT/RL efficiency, kernel/CUDA optimizations, and offline throughput improvement
• Build and improve search engine components such as offline document processing pipelines, distributed indexing systems for billion/trillion documents , feature extraction services, and multi-stage retrieval/ranking services
• Develop scalable data pipelines and distributed systems for processing billion- or trillion-scale text, image, and video datasets, covering offline computation, stream processing, task scheduling, storage, and data quality
• Explore emerging technologies in personalized search, generative search, multimodal understanding, reinforcement learning, model alignment, and evaluation
• Collaborate with machine learning, product, and other engineering teams to deliver accurate, responsive, and engaging AI-powered and visual search experiences