Software Engineer Intern (ML Infra) - 2027 Start (PhD)

TikTok
San Jose
Workplace: OnsiteInternshipFunction: Software EngineeringEducation: phdSkills: ["Problem-solving","Analytical skills"]

Work with the Ads Infra team to build and optimize large-scale ML systems that power advertising training, serving, and privacy enforcement. Develop advertising recommendation models and ML data pipelines, then build distributed ML platforms for training and online serving. Explore MLLM applications in advertising, apply AI agents for workflow orchestration and automated operations, and design GPU allocation and utilization optimization.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 day ago

Software Engineer Intern (ML Infra) - 2027 Start (PhD)

✓ 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

Work with the Ads Infra team to build and optimize large-scale ML systems that power advertising training, serving, and privacy enforcement. Develop advertising recommendation models and ML data pipelines, then build distributed ML platforms for training and online serving. Explore MLLM applications in advertising, apply AI agents for workflow orchestration and automated operations, and design GPU allocation and utilization optimization.
Location: San Jose
Workplace: Onsite
Employment Type: Internship
Job Function: Software Engineering
Seniority: Intern level

Key Responsibilities

  • •Develop and optimize advertising recommendation models across training, evaluation, inference, and user behavior sequence modeling.
  • •Build large-scale data pipelines for sample generation and joining, feature processing, storage, and data quality assurance.
  • •Develop ML platforms for distributed training, model distribution, online serving, and lifecycle management.
  • •Explore MLLM applications in advertising, including data production, model training, fine-tuning, and inference optimization.
  • •Apply AI agents for workflow orchestration, resource scheduling, failure diagnosis, and automated operations, including GPU allocation and utilization optimization.

Key Requirements

  • •Currently pursuing a PhD in Computer Science, AI, Machine Learning, or a related discipline.
  • •Proficiency in at least one programming language such as C++, Python, Go, or Java.
  • •Solid knowledge of data structures, algorithms, machine learning, and computer systems.
  • •Familiarity with recommendation systems, MLLMs, distributed systems, ML platforms, or large-scale data processing.
  • •Experience with deep learning frameworks, distributed training, GPU scheduling, AI agents, or cloud-native technologies is a plus.
Education:PhD / Doctorate in Computer Science, AI, Machine Learning
Skills:Problem-solvingAnalytical skills
Tech Stack:C++PythonGoJavaMachine LearningDeep LearningRecommendation SystemsLLMsDistributed SystemsML PlatformsLarge-Scale Data ProcessingGPU SchedulingAI AgentsCloud-native technologies

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