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

TikTok
San Jose
Full timeFunction: Research & Scientific (R&D)Education: phdSkills: ["Problem-solving","Collaboration","Performance optimization","Bottleneck analysis","Building large-scale systems"]

Build and optimize TikTok’s model infrastructure for both large-scale recommendation systems and LLM training and online inference. Work on distributed training/inference systems, high-performance GPU computing, and scalable LLM infrastructure. Collaborate with researchers to deploy LLM training/serving solutions, improve performance via GPU optimization and efficient communication, and diagnose system bottlenecks to implement optimizations that impact recommendations for 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 (PhD)

✓ 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 both large-scale recommendation systems and LLM training and online inference. Work on distributed training/inference systems, high-performance GPU computing, and scalable LLM infrastructure. Collaborate with researchers to deploy LLM training/serving solutions, improve performance via GPU optimization and efficient communication, and diagnose system bottlenecks to implement optimizations that impact recommendations for hundreds of millions of users.
Location: San Jose
Employment Type: Full time
Job Function: Research & Scientific (R&D)
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

  • •Completing or recently completed a PhD in Computer Science or a 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:LLM trainingLLM inferenceMachine learning systemsDistributed systemsOpen source
Education:PhD / Doctorate in Computer Science (or related discipline)
Skills:Problem-solvingCollaborationPerformance optimizationBottleneck analysisBuilding large-scale systems
Tech Stack:C++PythonPyTorchTensorFlowTransformer architecturesLarge Language ModelsDPTPPPFSDPZeROCUDATritonKV CacheContinuous BatchingFlashAttentionDistributed trainingDistributed computingGPU programming

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