AI Infra Engineer Graduate (Recommendation & LLM) - 2027 Start

TikTok
San Jose
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningEducation: bachelorsSkills: ["Problem-solving","Collaboration","Performance optimization","Building large-scale AI systems"]

Build and optimize TikTok’s Model Infrastructure for the For You recommendation system and large language models. You’ll develop distributed training and online inference systems, improve performance through GPU optimization and efficient communication, and collaborate with researchers to deploy LLM training and serving solutions. Focus on scalable infrastructure, analyze bottlenecks, and turn AI research into production systems used by hundreds of millions of users.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
5 hours ago

AI Infra Engineer Graduate (Recommendation & LLM) - 2027 Start

✓ Verified Job

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

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

Job Summary

Build and optimize TikTok’s Model Infrastructure for the For You recommendation system and large language models. You’ll develop distributed training and online inference systems, improve performance through GPU optimization and efficient communication, and collaborate with researchers to deploy LLM training and serving solutions. Focus on scalable infrastructure, analyze bottlenecks, and turn AI research into production systems used by hundreds of millions of users.
Location: San Jose
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
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 via 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

  • •Completing or recently completed a Bachelor's degree in Computer Science or related discipline.
  • •Strong programming skills in C++ or Python.
  • •Good understanding of data structures, algorithms, and computer systems.
  • •Familiarity with PyTorch or TensorFlow.
  • •Knowledge of Transformer architectures and large language models (LLMs).
Experience:LLMMachine learning systemsDistributed systemsAI infrastructureOpen source
Education:Bachelor's in Computer Science
Skills:Problem-solvingCollaborationPerformance optimizationBuilding large-scale AI systems
Tech Stack:C++PythonPyTorchTensorFlowTransformerLLMsDistributed trainingDPTPPPFSDPZeROGPU programmingCUDATritonLLM servingKV CacheContinuous BatchingFlashAttention

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