Graduate ML Infra Engineer for Large-Scale Ads

TikTok; San Jose (CA); On-site; USD 128,000 - 256,000; Full time

Benefits offered by this job

Medical/dental/vision insurance

401(k) with company match

Paid parental leave

Short/long-term disability

Life insurance

Wellbeing benefits

10 holidays per year

10 sick days per year

17 days of Paid Personal Time

Job summary

TikTok is seeking engineers for the Ads Infra team to design and deploy scalable ML-driven advertising systems in San Jose. You will work on training, serving, and optimizing revenue-driving models at massive scale, collaborating across data, ML, and infrastructure groups.

The role emphasizes building robust data pipelines, distributed ML platforms, and applying AI techniques to workflow orchestration and resource management, with a focus on performance and reliability.

Qualifications

• Bachelor’s or Master’s in Computer Science or a related field.
• Proficiency in C++, Python, Go or Java.
• Strong knowledge of data structures, algorithms and ML.
• Familiarity with recommender systems, ML systems or large-scale data processing.

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

Distributed systems

ML platforms

Cloud-native tech

Education

Bachelor's or Master's in CS or related

Tools

TensorFlow

PyTorch

GPU scheduling

Job description

TikTok is seeking engineers for the Ads Infra team to design and deploy scalable ML-driven advertising systems in San Jose. You will work on training, serving, and optimizing revenue-driving models at massive scale, collaborating across data, ML, and infrastructure groups.

The role emphasizes building robust data pipelines, distributed ML platforms, and applying AI techniques to workflow orchestration and resource management, with a focus on performance and reliability.