ML Infra Engineer Graduate (Ads Infra) - 2027 Start

About the role

The Ads Infra team at TikTok builds highly efficient and stable infrastructure to collect and process massive-scale data for machine learning training, serving, and privacy enforcement. This includes feature engineering, feature store, and training-data generation. As a member of the team, you will help develop and optimize advertising recommendation models, data pipelines, and ML platforms that power TikTok’s global ads system.

Responsibilities

• Develop and optimize advertising recommendation models, covering training, evaluation, inference, and user-behavior sequence modeling.
• Build large-scale data pipelines for sample generation, feature processing, storage, and data-quality assurance.
• Develop ML platforms for distributed training, model distribution, online serving, and lifecycle management.
• Explore MLLM (multi-modal large language model) 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.

Requirements

• Completing or recently completed a Bachelor’s or Master’s degree in Computer Science 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.
• Must be able to commit to an onboarding date by the end of the year; state your availability and graduation date in your resume.

Pay

The base salary range for this position is $128,000 – $256,000 annually. Compensation may vary based on qualifications, skills, experience, and location. Base pay is one part of the total package, which may also include discretionary bonuses, incentives, and restricted stock units.

Benefits

• Day-one access to medical, dental, and vision insurance.
• 401(k) savings plan with company match.
• Paid parental leave.
• Short-term and long-term disability coverage.
• Life insurance.
• Wellbeing benefits.
• 10 paid holidays per year.
• 10 paid sick days per year.
• 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).