Model Infra Engineer Intern (TikTok Recommendation Architecture) - 2026 Start (BS/MS)
Singapore
Why you should apply for a job to TikTok:
4.5/5 in overall job satisfaction
4.5/5 in supportive management
100% say women are treated fairly and equally to men
100% would recommend this company to other women
100% say the CEO supports gender diversity
Employee well-being is supported via hybrid work, short-term counseling through our EAP and a premium subscription to Headspace.
We embrace diversity across all dimensions and provide employees with 9 employee resource groups globally, including our WOMEN ERG.
Comprehensive parental leave policy as well as fertility treatment through healthcare providers with a $20,000 lifetime maximum.
#7531736718753646855
Position summary
olling basis - we encourage you to apply early.
Successful candidates must be able to commit to at least 3 months long internship period.
Responsibilities
- Optimize model performance and memory efficiency on GPU-based systems.
- Collaborate with research and infra teams to deploy high-throughput training and inference pipelines.
- Develop tools and libraries to accelerate deep learning workloads at scale.
- Analyze system performance (e.g. GPU profiling, kernel analysis, throughput tuning).
Qualifications
Minimum Qualifications:
- Undergraduate, or Postgraduate who is currently pursuing a degree/master in Computer Science, EE, or related field.
- Familiarity with deep learning and frameworks like PyTorch or TensorFlow. And understanding of GPU basics (memory hierarchy, compute kernels).
- Programming experience in Python/C++/CUDA/Trition is a plus.
Preferred Qualifications:
- Course or project experience with LLMs or recommender systems.
- Exposure to CUDA, NCCL, or deep learning compiler stacks (e.g. XLA, TensorRT).
- Publications or open-source contributions are a bonus.
By submitting an application for this role, you accept and agree to our global applicant privacy policy, which may be accessed here: https://careers.tiktok.com/legal/privacy
If you have any questions, please reach out to us at [email protected]
Why you should apply for a job to TikTok:
4.5/5 in overall job satisfaction
4.5/5 in supportive management
100% say women are treated fairly and equally to men
100% would recommend this company to other women
100% say the CEO supports gender diversity
Employee well-being is supported via hybrid work, short-term counseling through our EAP and a premium subscription to Headspace.
We embrace diversity across all dimensions and provide employees with 9 employee resource groups globally, including our WOMEN ERG.
Comprehensive parental leave policy as well as fertility treatment through healthcare providers with a $20,000 lifetime maximum.