AI Infrastructure Engineer Graduate (Algorithm Infrastructure) - 2027 Start (PhD)

TikTok is the leading destination for short-form mobile video, and they are seeking an AI Infrastructure Engineer Graduate to build the inference infrastructure for large-scale AI systems. The role involves developing robust and scalable solutions for online serving and optimizing performance for complex multimodal models.

What you'll do:

• Build and evolve next-generation inference systems for large-scale online traffic, including global scheduling across heterogeneous compute resources, high-concurrency load balancing, and efficient batch formation
• Optimize distributed inference for 200B+ models and complex multimodal models through TP, EP, DP, and related strategies to improve throughput and latency in production
• Develop high-performance kernels for frontier model architectures such as MoE, emerging attention mechanisms, and multimodal fusion layers using CUDA, Triton, and related tools
• Explore AI-driven infrastructure for inference systems, including AI Agents for kernel optimization, performance tuning, consistency validation, deployment pipelines, and intelligent operations

What they're looking for:

• Individuals who are completing or have recently completed a PhD degree in Software Development, Computer Science, Computer Engineering, or a related technical discipline in Computer Science, Software Engineering, Artificial Intelligence, Mathematics, or related fields
• Familiarity with large-model architectures and strong system design skills for complex, high-concurrency environments
• Strong understanding of asynchronous scheduling, resource pooling, and load balancing in distributed microservice systems
• Strong engineering skills in performance optimization and production system development

Benefits:

• Day one access to medical, dental, and vision insurance
• A 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)
• Employees also receive 10 paid holidays per year, 10 paid sick days per year and 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).