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

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

Build and optimize the distributed training and online inference infrastructure behind TikTok’s large-scale recommendation systems and LLM foundation models. Work on high-performance GPU computing, scalable LLM infrastructure, and productionizing research with experienced engineers and researchers. Improve training/inference performance via GPU optimization and efficient communication, diagnose system bottlenecks, and implement performance improvements for Transformer-based architectures.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

AI Infrastructure Engineer Graduate (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: 13 hours agoStatus: Live

Job Summary

Build and optimize the distributed training and online inference infrastructure behind TikTok’s large-scale recommendation systems and LLM foundation models. Work on high-performance GPU computing, scalable LLM infrastructure, and productionizing research with experienced engineers and researchers. Improve training/inference performance via GPU optimization and efficient communication, diagnose system bottlenecks, and implement performance improvements for Transformer-based architectures.
Location: Singapore
Workplace: Onsite
Employment Type: Full time
Job Function: DevOps, Cloud & Infrastructure
Seniority: Graduate level

Key Responsibilities

  • •Build and optimize infrastructure for large-scale model training and online inference.
  • •Develop distributed systems supporting large recommendation models and LLMs.
  • •Improve training and inference performance through GPU optimization and efficient communication.
  • •Collaborate with researchers to develop and deploy LLM training and serving solutions.
  • •Analyze system bottlenecks and implement performance optimizations.

Key Requirements

  • •Complete or have recently completed a Bachelor’s or Master’s degree in AI, Software Development, Computer Science, Computer Engineering, or a related field.
  • •Strong programming skills in C++ or Python.
  • •Solid understanding of data structures, algorithms, and computer systems.
  • •Familiarity with PyTorch or TensorFlow and knowledge of Transformer architectures and LLMs.
  • •Passion for building large-scale AI systems with strong problem-solving skills.
Education:
Skills:Problem-solvingPassion for building large-scale AI systems
Tech Stack:C++PythonPyTorchTensorFlowPyTorch or TensorFlowTransformer architecturesLarge Language Models (LLMs)CUDATritonGPU optimizationDistributed trainingKV CacheContinuous BatchingFlashAttentionDPTPPPFSDPZeRO

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