TikTok
AI Infra Engineer Intern (Recommendation & LLM) - 2027 Summer
San Jose, California · Internship · Posted Oct 28, 2026
- Education
- Currently pursuing a Bachelor's degree in Computer Science or a related discipline.
About the job
Build and optimize distributed training and inference infrastructure for TikTok’s recommendation models and LLMs, including GPU performance improvements and production deployment.
What you'll do
- Build and optimize infrastructure for large-scale model training and online inference.
- Develop distributed systems for large recommendation models and LLMs, and improve performance through GPU optimization and efficient communication.
- Collaborate with researchers to develop and deploy LLM training and serving solutions; analyze bottlenecks and optimize performance.
What they ask for
- Currently pursuing a Bachelor's degree in Computer Science or a related discipline.
- Strong programming skills in C++ or Python.
- Good understanding of data structures, algorithms, and computer systems.
- Familiarity with PyTorch or TensorFlow, and knowledge of Transformer architectures and LLMs.
Nice to have
- Hands-on LLM training or inference experience; familiarity with distributed training concepts.
- Experience with GPU programming, LLM serving techniques, or contributions to relevant open-source projects or research.
Tools
C++, Python, PyTorch, TensorFlow, CUDA, Triton, FSDP, ZeRO
AI focus
Develop and optimize large-scale training and online inference infrastructure for recommendation models and LLMs, including distributed systems, GPU computing, and LLM serving.
Model-written summary · the original posting is authoritative