Algorithm AI Infrastructure Engineer Intern

Position:
Algorithm AI Infrastructure Engineer Intern (Algorithm Infrastructure) - 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:
TikTok is an international short video platform available in over 150 countries, dedicated to building inference infrastructure for large-scale AI systems. The internship offers hands-on learning and collaboration with industry experts.

Responsibilities:
- Build and evolve next-generation inference systems for large-scale online traffic.
- Optimize distributed inference for 200B+ models and complex multimodal models.
- Develop high-performance kernels for frontier model architectures using CUDA, Triton, and related tools.
- Explore AI-driven infrastructure for inference systems.

Requirement:
- Currently pursuing a PhD in Software Development, Computer Science, Computer Engineering, or related fields.
- Familiarity with large-model architectures and strong system design skills.
- Strong engineering skills in performance optimization and production system development.

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).
- Housing allowance may be available for non-remote interns.