AI Infrastructure Engineer Intern
TikTok
AI Infrastructure Engineer Intern
TikTok
Short-form video platform with ads
Our Live Recommendation Architecture Team is responsible for building up and optimizing the architecture for live broadcast recommendation system to provide the most stable and best experience for our users. The team is responsible for system stability and high availability, online services and offline data flow performance optimization, solving system bottlenecks, reducing cost overhead, building data and service mid-platform, realizing flexible and scalable high-performance storage and computing systems. We work closely with applied machine learning engineers and build scalable systems to support all kinds of innovative algorithms and techniques. We are looking for talented individuals to join us for an internship. Our internship program offers students hands-on experience, industry exposure, and opportunities to apply their knowledge to real-world challenges while building a strong foundation for personal and professional growth. Interns will gain practical experience, explore potential career paths, and participate in social events, learning programs, and development workshops alongside industry professionals. Candidates may apply to a maximum of two positions across Our Company and its affiliates globally. Applications will be considered in the order they are submitted. Applications are reviewed on a rolling basis, so we encourage you to apply early. Please clearly state your availability in your resume, including your start and end dates. Successful candidates must be able to commit to at least 3 months long internship period. Responsibilities - Optimize recommendation model memory usage and inference latency on GPU-based heterogeneous computing systems for live scenarios - Design and implement high-throughput distributed training and inference pipelines for live broadcast recommendation models - Perform GPU kernel profiling and performance tuning to optimize real-time inference throughput for live streaming workloads
Minimum Qualifications: - Currently pursuing a Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Information Systems, or a related technical discipline - Solid programming skills in C++/CUDA/Python, with basic understanding of GPU architecture and distributed training - Familiarity with deep learning frameworks and basic knowledge of recommendation systems Preferred Qualifications: - Experience building production-grade training and inference systems for large-scale models, especially for recommendation or live streaming scenarios - Hands-on experience optimizing recommendation models for memory efficiency, latency, and throughput improvements - Familiarity with distributed training frameworks (NCCL, Horovod, DeepSpeed, FSDP)
About TikTok
TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.
Why Join Us
Inspiring creativity is at the core of TikTok's mission. Our innovative product is built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and bring joy - a mission we work towards every day.
We strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team. We're resilient and embrace challenges as they come. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our company, and our users. When we create and grow together, the possibilities are limitless. Join us.
Diversity & Inclusion
TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.
TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.
Explore more opportunities at TikTok