T

TikTok

Short-form video platform with ads

AI Infrastructure Optimization Engineer Intern - Global E-Commerce Recommendation & Search Architecture

Natural Language Processing & LLMsConfirmed live in the last 24 hours
Internship · Summer 2027SingaporeNo salary listedIn-person

About the Job

Requirements

  • Currently pursuing a Bachelor's or Master's degree in Computer Science, Computer Engineering, Information Systems, or a related technical discipline.
  • Have solid programming skills in C++, CUDA, and Python, with a basic understanding of GPU architecture and distributed training.
  • Be familiar with deep learning frameworks such as PyTorch and TensorFlow, and have basic knowledge of large language models.

Responsibilities

  • Research and implement large-language-model-based recommendation foundation model training and inference optimization technologies for e-commerce scenarios.
  • Build model optimization toolchains, including quantization, pruning, and distillation, for deep learning frameworks.
  • Design and implement distributed inference serving architectures that support large-scale large language model deployment for recommendation use cases.

Desired Qualifications

  • Have hands-on experience with large language model inference acceleration technologies, including model quantization, key-value cache optimization, and FlashAttention.
  • Have practical experience using large language model systems such as Megatron-LM, DeepSpeed, vLLM, or TensorRT-LLM.
  • Have an interest in recommendation systems and generative artificial intelligence technologies.

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