AI Infrastructure Engineer Intern (TikTok Live Recommendation Architecture) - 2027 Start

TikTok; Singapore; On-site; SGD 17,000 - 23,000; Part time

Job summary

TikTok is offering an internship in its Live Recommendation Architecture Team in Singapore. The role focuses on building scalable, high-performance systems for live broadcast recommendations while gaining hands-on experience with cutting-edge GPU-accelerated training pipelines.

Interns will work on optimizing memory usage, reducing inference latency, and profiling GPU kernels for real-time workloads, alongside ML engineers. Rolling applications and a 3-month minimum commitment are expected.

Qualifications

• Pursuing a Bachelor's or Master's degree in Computer Science, Computer Engineering, Information Systems, or a related technical discipline
• Strong programming skills in C++/CUDA/Python; basic understanding of GPU architecture and distributed training
• Familiarity with deep learning frameworks and basic knowledge of recommendation systems

Responsibilities

• Optimize recommendation model memory usage and inference latency on GPU-based heterogeneous computing systems for live scenarios
• Design and implement high-throughput distributed training and inference pipelines for live broadcast recommendation models
• Perform GPU kernel profiling and performance tuning to optimize real-time inference throughput for live streaming workloads

Skills

C++/CUDA/Python

GPU architecture

Distributed training

Deep learning frameworks

Recommendation basics

Education

Bachelor's/Master's in CS/CE/IS

Tools

NCCL

Horovod

DeepSpeed

FSDP

Job description

Responsibilities

Our Live Recommendation Architecture Team is responsible for building up and optimizing the architecture for live broadcast recommendation system to provide the most stable and best experience for our users. The team is responsible for system stability and high availability, online services and offline data flow performance optimization, solving system bottlenecks, reducing cost overhead, building data and service mid-platform, realizing flexible and scalable high-performance storage and computing systems. We work closely with applied machine learning engineers and build scalable systems to support all kinds of innovative algorithms and techniques.

We are looking for talented individuals to join us for an internship. Our internship program offers students hands-on experience, industry exposure, and opportunities to apply their knowledge to real-world challenges while building a strong foundation for personal and professional growth.

Interns will gain practical experience, explore potential career paths, and participate in social events, learning programs, and development workshops alongside industry professionals.

Candidates may apply to a maximum of two positions across Our Company and its affiliates globally. Applications will be considered in the order they are submitted.

Applications are reviewed on a rolling basis, so we encourage you to apply early. Please clearly state your availability in your resume, including your start and end dates.

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

Internship Responsibilities

• Optimize recommendation model memory usage and inference latency on GPU-based heterogeneous computing systems for live scenarios
• Design and implement high-throughput distributed training and inference pipelines for live broadcast recommendation models
• Perform GPU kernel profiling and performance tuning to optimize real-time inference throughput for live streaming workloads

Mandatory Obligations: Minimum 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 and basic knowledge of recommendation systems

Mandatory Obligations: Preferred Qualifications

• Experience building production-grade training and inference systems for large-scale models, especially for recommendation or live streaming scenarios
• Hands-on experience optimizing recommendation models for memory efficiency, latency, and throughput improvements
• Familiarity with distributed training frameworks (NCCL, Horovod, DeepSpeed, FSDP)

Job Information

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