Position:
Technology Software Engineer Intern (ML Infra) - 2027 Start (PhD)
Company:
TikTok
Compensation:
The hourly rate range for this position in the selected city is $60- $60.
Location:
San Jose, USA
Employment type:
Intern
Work Arrangement:
Not specified
Short Summary:
The Ads Infra team at TikTok is looking for talented individuals for an internship to develop and optimize advertising recommendation models and build large-scale data pipelines. Interns will gain hands-on experience and industry exposure while participating in social events and learning programs.
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.
Requirement:
- Currently pursuing a PhD 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:
- Interns have day one access to health insurance, life insurance, wellbeing benefits, and more.
- 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).
- Interns who are not working 100% remote may also be eligible for a housing allowance.