Software Engineer Intern (ML Infra) - 2027 Start (PhD)
Core
Building large-scale ML infrastructure for TikTok's ad system, including data pipelines, training platforms, and GPU resource management.
Role type
ML Infrastructure Engineer (Intern)
Builds
Data pipelines, ML training platforms, GPU scheduling systems, and AI agent workflows for ad operations.
Domain
Advertising technology, Machine Learning Infrastructure, Distributed Systems
Deliverable
infrastructure
Required skills
C++, Python, Go, Java, Data structures, Algorithms, Machine Learning, Computer Systems, Deep learning frameworks, Distributed training, GPU scheduling, Cloud-native technologies
Responsibilities
Develop advertising recommendation models; Build large-scale data pipelines for sample generation and feature processing; Develop ML platforms for distributed training and model serving; Explore MLLM applications in advertising; Apply AI agents to workflow orchestration and automated operations; Build machine learning platforms for GPU allocation and cost accounting.