Software Engineer Intern (TikTok AI Search & Visual Search Infra Team) - 2027 Summer
TikTok, San Jose, California
What you will do
- 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.