Recommendation Architecture AI/ML Infrastructure Engineer Graduate (Data-Arch-TikTok Live) - 2027 Start

TikTok
San Jose
Workplace: OnsiteFull timeFunction: DevOps, Cloud & InfrastructureSkills: ["Problem-solving","Collaboration"]

Build and optimize infrastructure for large-scale model training and online inference that powers TikTok’s recommendation systems and LLMs. Work on distributed training and inference systems, high-performance GPU computing, and scalable LLM infrastructure. Collaborate with engineers and researchers to turn cutting-edge AI into production systems, diagnose bottlenecks, and deliver performance improvements through GPU optimization and efficient communication.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

Recommendation Architecture AI/ML Infrastructure Engineer Graduate (Data-Arch-TikTok Live) - 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 infrastructure for large-scale model training and online inference that powers TikTok’s recommendation systems and LLMs. Work on distributed training and inference systems, high-performance GPU computing, and scalable LLM infrastructure. Collaborate with engineers and researchers to turn cutting-edge AI into production systems, diagnose bottlenecks, and deliver performance improvements through GPU optimization and efficient communication.
Location: San Jose
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

  • •Completing or recently completed a Bachelor's or Master's degree in Artificial Intelligence, Software Development, Computer Science, Computer Engineering, or a related discipline.
  • •Strong programming skills in C++ or Python, with solid knowledge of data structures, algorithms, and computer systems.
  • •Familiarity with PyTorch or TensorFlow and knowledge of Transformer architectures and Large Language Models (LLMs).
  • •Ability to analyze system bottlenecks and implement performance optimizations for large-scale AI systems.
  • •Strong problem-solving skills and passion for building large-scale AI systems; can commit to an onboarding date by the end of the year.
Experience:LLM trainingLLM inferenceDistributed trainingGPU programmingOpen source
Education:
Skills:Problem-solvingCollaboration
Tech Stack:C++PythonPyTorchTensorFlowCUDATritonTransformersLLMsDPTPPPFSDPZeROKV 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