ML Infra Engineer Intern (Ads Infra) - 2027 Summer

About the role

The Ads Infra team at TikTok builds highly efficient and stable infrastructure to collect and process data for machine learning training, serving, and privacy enforcement. This includes feature engineering, feature store, and training data generation. As an intern, you will gain hands-on experience, explore potential career paths, and participate in social events, learning programs, and development workshops alongside industry professionals.

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 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

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

Benefits

• Day one access to health insurance, life insurance, and wellbeing benefits.
• 10 paid holidays per year and paid sick time (56 hours if hired in the first half of the year, 40 if hired in the second half).
• Eligibility for housing allowance if not working 100% remote.

Pay

The hourly rate range for this position is $45–$60.

About TikTok

TikTok is the leading destination for short-form mobile video. Our mission is to inspire creativity and bring joy. With global headquarters in Los Angeles and Singapore, we have offices in major cities worldwide, including New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Why Join Us

We create value for our communities by inspiring creativity and bringing joy. Our culture is built on curiosity, humility, and a desire to make an impact. We embrace challenges, foster innovation, and grow together. Diversity and inclusion are core to our mission, ensuring our workplace reflects the global communities we serve.