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

TIKTOK PTE. LTD.; Singapore; On-site; SGD 60,000 - 110,000; Full time

Benefits offered by this job

Yoga and fitness

Stock options

Competitive compensation

Flexible hours

Job summary

TikTok is hiring for a graduate/early-career position in Singapore to join our Live Recommendation Architecture Team. You will work on building and optimizing the architecture for live broadcast recommendations, ensuring stability, high availability, and efficient data flow.

You’ll collaborate with ML engineers to support innovative algorithms and scalable systems. As a recent graduate, you will explore bold ideas, tackle complex challenges, and have opportunities for growth in a fast-moving,

Qualifications

• Bachelor's or Master's in AI, CS, or related discipline.
• Solid programming skills in C++, CUDA, and Python.
• Familiarity with GPU architecture and distributed training is desirable.
• Experience in at least one language: C, C++, Java or Golang.
• Strong ownership, communication and initiative.

Responsibilities

• Optimize model performance and memory efficiency on GPU-based systems for Live recommendation.
• Deploy and scale high-throughput training and inference pipelines for live models.
• Develop tools, libraries, and compiler optimizations for DL workloads.
• Analyze system performance via GPU profiling and throughput tuning for real-time workloads.
• Build and maintain high-performance computing frameworks and storage for training/inference.

Skills

C++

CUDA

Python

Education

Bachelor's or Master's in AI / Computer Science / Software

Tools

NCCL

Horovod

DeepSpeed

TVM

LLVM

Job description

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.

Job highlights

• Yoga and fitness
• Stock options
• Positive team atmosphere
• Career growth opportunity
• Paid leave
• Flat organization
• 100+ mil users
• Industry experts
• Competitive compensation
• Flexible hours

Responsibilities

Team Introduction:

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 our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.

Responsibilities

• Optimize model performance and memory efficiency on GPU-based systems for TikTok Live recommendation scenarios
• Deploy and scale high-throughput training and inference pipelines for live broadcast recommendation models
• Develop tools, libraries, and compiler optimizations to accelerate deep learning workloads for live streaming use cases
• Analyze system performance through GPU profiling, kernel analysis, and throughput tuning for real-time workloads
• Build and maintain high-performance computing frameworks and storage systems for live recommendation model training and inference

Qualifications

Minimum Qualifications:

• Individuals who are completing or have recently completed a Bachelor's or Master's degree in Artificial Intelligence, Software Development, Computer Science, Computer Engineering or a related discipline.
• Solid programming skills in C++/CUDA/Trition/Python.
• Familiarity with GPU architecture and distributed training is highly desirable.
• Experience programming in at least one of the following programming languages: C, C++, Java or Golang.
• Effective communication skills and a sense of ownership and drive.

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, including memory efficiency, latency, and throughput improvements.
• Knowledge of distributed training frameworks (e.g., NCCL, Horovod, DeepSpeed, FSDP) is a plus.
• Familiarity with deep learning compiler frameworks such as TVM or LLVM, and understanding of their underlying principles.
• Experience in livestream or real‑time system related areas is a plus.
• Contributions to open‑source projects or relevant research publications.
• Agile, quick self learner, highly self‑motivated with strong sense of product ownership and creative problem solver.
• Good collaborator and team player, comfortable working in a fast moving, culturally diverse and globally distributed team environment.