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

TikTok; Singapore; On-site; SGD 13,000 - 20,000; Full time

Job summary

TikTok is seeking an AI Infrastructure Optimization Engineer Intern for a 2027 start in Singapore. You will research and implement optimization techniques for large-scale LLM-based e-commerce recommendations, and help build end-to-end inference pipelines.

You will work with C++/CUDA/Python, PyTorch/TensorFlow, and distributed systems, gaining hands-on experience with Megatron-LM, DeepSpeed, and other LLM tools while contributing to real-world deployment.

Qualifications

• Pursuing a Bachelor’s or Master’s degree in CS/CE/IS.
• Strong programming in C++/CUDA/Python; basic GPU knowledge.
• Familiarity with PyTorch and TensorFlow; knowledge of LLMs.

Responsibilities

• Research and implement LLM-based optimization for recommendation use cases.
• Build model optimization toolchains including quantization, pruning, distillation.
• Design distributed inference serving architectures for large-scale LLMs.

Skills

C++

CUDA

Python

GPU architecture

Distributed training

PyTorch

TensorFlow

LLM acceleration

Education

Bachelor's degree in CS/CE/IS

Master's degree in CS/CE/IS

Tools

Megatron-LM

DeepSpeed

vLLM

TensorRT-LLM

Job description

Discover a career that energizes and excites you every day.

@2026 TikTok

Technology

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

Location:

Employment Type:

Intern

Job Code:

A27925

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

Successful candidates must be able to commit to at least 3 months long internship period.

Qualifications

• 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 (model quantization, KV cache optimization, FlashAttention)
• Practical experience using LLM systems such as Megatron-LM, DeepSpeed, vLLM, or TensorRT-LLM
• Interest in recommendation systems and generative AI technologies

Job Information

About TikTok

TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Why Join Us

Inspiring creativity is at the core of TikTok's mission. Our innovative product is built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and bring joy - a mission we work towards every day.

We strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team. We're resilient and embrace challenges as they come. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.