Recommendation Architecture AI/ML Infrastructure Engineer Intern (Data-Arch-TikTok Live) - 2027 Summer

TikTok
San Jose
Workplace: OnsiteInternshipFunction: DevOps, Cloud & InfrastructureEducation: bachelorsSkills: ["Problem-solving","Passion for building large-scale AI systems"]

Build and optimize large-scale AI infrastructure for recommendation training and online inference. Work on distributed systems for recommendation models and LLMs, improving performance through GPU optimization and efficient communication. Collaborate with researchers to deploy LLM training and serving solutions, analyze bottlenecks, and implement optimizations. The internship provides hands-on experience applying AI systems knowledge to production workloads impacting TikTok users.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

Recommendation Architecture AI/ML Infrastructure Engineer Intern (Data-Arch-TikTok Live) - 2027 Summer

✓ 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 large-scale AI infrastructure for recommendation training and online inference. Work on distributed systems for recommendation models and LLMs, improving performance through GPU optimization and efficient communication. Collaborate with researchers to deploy LLM training and serving solutions, analyze bottlenecks, and implement optimizations. The internship provides hands-on experience applying AI systems knowledge to production workloads impacting TikTok users.
Location: San Jose
Workplace: Onsite
Employment Type: Internship
Job Function: DevOps, Cloud & Infrastructure
Seniority: Intern 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

  • •Currently pursuing a Bachelor's or Master's degree in Artificial Intelligence, Software Development, Computer Science, Computer Engineering, or a related technical 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:ResearchOpen-sourceMachine learning systemsDistributed systemsLLM infrastructure
Education:Bachelor's in Artificial Intelligence, Software Development, Computer Science, Computer Engineering, or a related technical discipline
Skills:Problem-solvingPassion for building large-scale AI systems
Tech Stack:C++PythonPyTorchTensorFlowTransformerLLMsDistributed trainingDPTPPPFSDPZeROCUDATritonKV 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