ML Infra Engineer Intern: Scale ML Pipelines

TikTok; San Jose (CA); On-site; USD 128,943,000 - 171,924,000; Part time

Benefits offered by this job

Health insurance

Housing allowance

Paid holidays

Paid sick time

Job summary

TikTok is seeking an Intern for its Ads Infra team in San Jose. You will help build and optimize ads systems at scale, contributing to data pipelines, ML training, inference, and feature engineering.

The role offers hands-on experience with real-world challenges, exposure to industry practices, and opportunities to collaborate across teams while building a strong foundation for growth in a fast-paced tech 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

Algorithms

Machine learning

Computer systems

Education

Bachelor's/Master's in CS/AI/ML

Job description

TikTok is seeking an Intern for its Ads Infra team in San Jose. You will help build and optimize ads systems at scale, contributing to data pipelines, ML training, inference, and feature engineering.

The role offers hands-on experience with real-world challenges, exposure to industry practices, and opportunities to collaborate across teams while building a strong foundation for growth in a fast-paced tech company.