ML Infra Engineer Intern: Ads Platform & Data Pipelines

TikTok; San Jose (CA); On-site; USD 62,000 - 83,000; Full time

Benefits offered by this job

Health insurance

Housing allowance

Paid holidays

Paid sick time

Job summary

TikTok is offering a campus internship focused on ads infrastructure and ML systems. Interns will work on scalable data pipelines, ML platforms, and advertising models while gaining hands-on experience in a fast-paced tech environment.

Applicants should be enrolled in CS/AI/ML programs and proficient in at least one major language. The role emphasizes collaboration, learning, and real-world project experience in a globally connected company.

Qualifications

• Currently pursuing a Bachelor's/ Master's 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.

Responsibilities

• Develop and optimize advertising recommendation models, covering 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 to workflow orchestration, resource scheduling, failure diagnosis and automated operations.
• Build machine learning platforms for GPU allocation, scheduling, usage metering, cost accounting and utilization optimization.

Skills

C++/Python/Go/Java

Data Structures

ML Knowledge

Distributed Systems

GPU Scheduling

Education

Bachelor/Master in CS/AI/ML

Job description

TikTok is offering a campus internship focused on ads infrastructure and ML systems. Interns will work on scalable data pipelines, ML platforms, and advertising models while gaining hands-on experience in a fast-paced tech environment.

Applicants should be enrolled in CS/AI/ML programs and proficient in at least one major language. The role emphasizes collaboration, learning, and real-world project experience in a globally connected company.