AI Infrastructure Optimization Engineer Intern (TikTok Global E-Commerce Recommendation & Search Architecture) - 2027 Start

TikTok
Singapore
InternshipFunction: Architecture & Urban PlanningEducation: bachelorsSkills: []

Work on the E-commerce Recommendation Infra team to research and implement LLM-based training and inference optimization for recommendation scenarios. Build model optimization toolchains (quantization, pruning, distillation) and design distributed inference serving architectures to support large-scale LLM deployment. Collaborate with applied ML engineers to scale innovative recommendation algorithms, including generative recommendation approaches, while contributing to real-world infrastructure improvements.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

AI Infrastructure Optimization Engineer Intern (TikTok Global E-Commerce Recommendation & Search Architecture) - 2027 Start

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Work on the E-commerce Recommendation Infra team to research and implement LLM-based training and inference optimization for recommendation scenarios. Build model optimization toolchains (quantization, pruning, distillation) and design distributed inference serving architectures to support large-scale LLM deployment. Collaborate with applied ML engineers to scale innovative recommendation algorithms, including generative recommendation approaches, while contributing to real-world infrastructure improvements.
Location: Singapore
Employment Type: Internship
Job Function: Architecture & Urban Planning
Seniority: Intern level

Key Responsibilities

  • •Research and implement LLM-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 supporting large-scale LLM deployment for recommendation use cases.

Key Requirements

  • •Currently pursuing a Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Information Systems, or a related technical discipline.
  • •Solid programming skills in C++/CUDA/Python, with basic understanding of GPU architecture and distributed training.
  • •Familiarity with deep learning frameworks (PyTorch, TensorFlow) and basic knowledge of LLMs.
  • •Hands-on experience with LLM inference acceleration technologies such as model quantization, KV cache optimization, and FlashAttention.
  • •Practical experience using LLM systems such as Megatron-LM, DeepSpeed, vLLM, or TensorRT-LLM, and interest in recommendation systems and generative AI technologies.
Experience:E-commerceLLMGenerative AIRecommendation systems
Education:Bachelor's in Computer Science, Computer Engineering, Information Systems, or a related technical discipline
Tech Stack:C++CUDAPythonGPU architectureDistributed trainingPyTorchTensorFlowLLMsModel quantizationKV cache optimizationFlashAttentionMegatron-LMDeepSpeedVLLMTensorRT-LLM

Company Brief

TikTok
Short-form video platform that lets users create, share, and discover entertainment content through algorithmic recommendations. It also offers advertising and creator tools for brands, influencers, and businesses.
Industry: Digital Media
Company Size: Enterprise (1,001+ employees)
Growth: Scaleup
Headquarters: Singapore, Singapore
Founded: 2016
WebsiteLinkedIn