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

TikTok
Singapore
Full timeFunction: Architecture & Urban PlanningSkills: ["Communication","Ownership","Drive","Troubleshooting"]

Join the E-commerce Recommendation Infra team to build and optimize infrastructure for recommendation foundation models. You’ll integrate LLM and recommendation technologies, develop training/inference optimizations, and design next-generation model architectures using open-source stacks like vLLM, SGLang, Megatron-LM, and VeRL. Collaborate with algorithm teams to productionize end-to-end generative recommendation, covering lifelong modeling, KV cache optimization, multimodal tasks, and extreme GPU performance improvements.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

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

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Join the E-commerce Recommendation Infra team to build and optimize infrastructure for recommendation foundation models. You’ll integrate LLM and recommendation technologies, develop training/inference optimizations, and design next-generation model architectures using open-source stacks like vLLM, SGLang, Megatron-LM, and VeRL. Collaborate with algorithm teams to productionize end-to-end generative recommendation, covering lifelong modeling, KV cache optimization, multimodal tasks, and extreme GPU performance improvements.
Location: Singapore
Employment Type: Full time
Job Function: Architecture & Urban Planning
Seniority: Graduate level

Key Responsibilities

  • •Integrate LLM and recommendation technologies by developing, deploying, and optimizing LLM training and inference for recommendation foundation models.
  • •Create automated, agentic, performance-oriented model optimization solutions for frameworks such as PyTorch.
  • •Design and implement next-generation model infra architectures using open-source components (vLLM, SGLang, Megatron, VeRL) for production deployment.
  • •Co-design with algorithm teams to research, innovate, and productionize end-to-end generative recommendation.
  • •Work on LLM-native recommendation and training/inference architectures, including ultra-long-context modeling, KV cache sharing, multimodal foundation models, and inference acceleration topics.

Key Requirements

  • •Completing or 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/Triton/Python.
  • •Familiarity with GPU architecture and distributed training (highly desirable).
  • •Experience programming in C, C++, Java, or Golang.
  • •Effective communication skills with a sense of ownership and drive.
Education:
Skills:CommunicationOwnershipDriveTroubleshooting
Tech Stack:C++CUDATritonPythonCJavaGolangPyTorchVllmSGLangMegatron-LMVeRLDeepSpeedVLLMTensorRTCUTLASSTensorRT-LLMKV cacheGPU kernelsMulti-GPU parallelism

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