T

TikTok

Short-form video platform with ads

Software Engineer Graduate - TikTok AI Search & Visual Search Infra Team

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USAIn-person

About the Job

Requirements

  • Complete or have recently completed a Bachelor's degree in Computer Science or a related discipline.
  • Have strong programming skills and a solid foundation in data structures and basic algorithms.
  • Be familiar with a Linux development environment and proficient in at least one programming language such as C++, Java, Python, or Go.
  • Have strong analytical and problem-solving abilities and an interest in solving complex engineering problems at scale.
  • Have effective communication and collaboration skills, a strong sense of ownership, and willingness to learn.

Responsibilities

  • Design, build, and optimize large-scale infrastructure for TikTok AI Search and 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, retrieval-augmented generation indexing and serving, and context and memory management.
  • Deploy and optimize large language model and vision-language model post-training and inference, including GPU inference acceleration, supervised fine-tuning and reinforcement learning efficiency, kernel and CUDA optimizations, and offline throughput improvement.
  • Build and improve search engine components such as offline document processing pipelines, distributed indexing systems for billion- or trillion-document collections, feature extraction services, and multi-stage retrieval and 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.

Desired Qualifications

  • Have internship, research, course project, or open-source experience in one or more of the following areas: large language model or vision-language model serving, retrieval-augmented generation, or AI agents; distributed systems, storage systems, and large-scale data processing; search engines, information retrieval, recommendation systems, or advertising systems.
  • Understand search engine concepts such as query understanding, indexing, retrieval, ranking, and result aggregation.
  • Be familiar with large language model serving frameworks such as vLLM, SGLang, or TensorRT-LLM, or agent development frameworks such as LangGraph.
  • Have experience analyzing or optimizing system performance, including latency, throughput, concurrency, memory utilization, or resource efficiency.
  • Be able to learn new technologies quickly and apply them to practical engineering or research problems.

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs