AI Infrastructure Engineer Graduate (TikTok Live Recommendation Architecture) - 2027 Start - #1174632

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.​

Job highlights

Yoga and fitness, Stock options, Positive team atmosphere, Career growth opportunity, Paid leave, Flat organization, 100+ mil users, Industry experts, Competitive compensation, Flexible hours

Responsibilities

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 our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.

Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.

Candidates can apply to a maximum of two positions and will be considered for jobs in the order you apply. The application limit is applicable to our Company and its affiliates' jobs globally. Applications will be reviewed on a rolling basis - we encourage you to apply early.

Responsibilities

- Optimize model performance and memory efficiency on GPU-based systems for TikTok Live recommendation scenarios

- Deploy and scale high-throughput training and inference pipelines for live broadcast recommendation models

- Develop tools, libraries, and compiler optimizations to accelerate deep learning workloads for live streaming use cases

- Analyze system performance through GPU profiling, kernel analysis, and throughput tuning for real-time workloads

- Build and maintain high-performance computing frameworks and storage systems for live recommendation model training and inference

Qualifications

- Individuals who are completing or have recently completed a Bachelor's or Master's degree in Artificial Intelligence, Software Development, Computer Science, Computer Engineering or a related discipline.

- Solid programming skills in C++/CUDA/Trition/Python.

- Familiarity with GPU architecture and distributed training is highly desirable.

- Experience programming in at least one of the following programming languages: C, C++, Java or Golang.

- Effective communication skills and a sense of ownership and drive.

- 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, including memory efficiency, latency, and throughput improvements.

- Knowledge of distributed training frameworks (e.g., NCCL, Horovod, DeepSpeed, FSDP) is a plus.

- Familiarity with deep learning compiler frameworks such as TVM or LLVM, and understanding of their underlying principles.

- Experience in livestream or real-time system related areas is a plus.

- Contributions to open-source projects or relevant research publications.

- Agile, quick self learner, highly self-motivated with strong sense of product ownership and creative problem solver.

- Good collaborator and team player, comfortable working in a fast moving, culturally diverse and globally distributed team environment.