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

TikTok
Singapore
Workplace: OnsiteInternshipFunction: DevOps, Cloud & InfrastructureSkills: ["Problem-solving","Passion for building large-scale AI systems"]

Build and optimize distributed training and low-latency online inference infrastructure for large-scale recommendation models and LLMs. Work on high-performance GPU kernel implementations, inter-node communication, and compiler optimization/operator fusion to accelerate deep learning execution. Collaborate with engineers and researchers to turn advanced AI techniques into production systems powering personalized recommendations for hundreds of millions of users.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 hour ago

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

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 1 hour agoStatus: Live

Job Summary

Build and optimize distributed training and low-latency online inference infrastructure for large-scale recommendation models and LLMs. Work on high-performance GPU kernel implementations, inter-node communication, and compiler optimization/operator fusion to accelerate deep learning execution. Collaborate with engineers and researchers to turn advanced AI techniques into production systems powering personalized recommendations for hundreds of millions of users.
Location: Singapore
Workplace: Onsite
Employment Type: Internship
Job Function: DevOps, Cloud & Infrastructure
Seniority: Intern level

Key Responsibilities

  • •Design, build, and optimize distributed training infrastructure and low-latency online inference systems for large-scale recommendation models and LLMs.
  • •Develop high-performance GPU kernel implementations and efficient inter-node communication primitives to improve training and inference efficiency.
  • •Build compiler optimization passes and operator fusion technologies to accelerate model execution.

Key Requirements

  • •Currently 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++ or Python, with a solid understanding of data structures and algorithms.
  • •Familiarity with PyTorch or TensorFlow, plus basic knowledge of Transformer architectures and LLMs.
  • •Hands-on experience with LLM training or inference via academic projects, internships, or open-source contributions.
  • •Familiarity with distributed training concepts (DP, TP, PP, FSDP, ZeRO) or GPU programming technologies (CUDA, Triton).
Education:
Skills:Problem-solvingPassion for building large-scale AI systems
Tech Stack:C++PythonPyTorchTensorFlowTransformersLLMsCUDATritonDPTPPPFSDPZeRO

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