Tech Lead Manager - Machine Learning Infrastructure

TikTok  logo


View Salaries, Reviews, and more  

Job Summary

S$14,000 - S$28,000 / Monthly

Job Type

Manager Lead

Years of Experience
At least 7 years

Tech Stacks
Hadoop Flink Spark PyTorch TensorFlow

Job Description

About TikTok

TikTok is the leading destination for short-form mobile video. Our mission is to inspire creativity and bring joy. TikTok has global offices including Los Angeles, New York, London, Paris, Berlin, Dubai, Singapore, Jakarta, Seoul and Tokyo.

Why Join Us

Creation is the core of TikTok's purpose. Our platform is built to help imaginations thrive. This is doubly true of the teams that make TikTok possible.

Together, we inspire creativity and bring joy - a mission we all believe in and aim towards achieving every day.

To us, every challenge, no matter how difficult, is an opportunity; to learn, to innovate, and to grow as one team. Status quo? Never. Courage? Always.

At TikTok, we create together and grow together. That's how we drive impact - for ourselves, our company, and the communities we serve.

Join us.

About the Team

The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Tech Lead Manager - Machine Learning Infrastructure to join our team to support and advance that mission.


- Lead the team to design and implement distributed inference/training/scheduling/ochestration/storage/parameter_server infrastructure for feeds, ads and search ranking models.

- Oversee the development of monitoring and management tools to ensure the reliability and scalability of machine learning infra.

- Manage the identification and prioritization of system inefficiencies and bottlenecks, leading efforts to enhance system performance.

- Lead the team in creating tools to analyze bottlenecks and sources of instability, formulating and implementing effective solutions.

- Collaborate with product teams, offering comprehensive solutions tailored to their specific requirements.


- Experience in leading an engineering team.

- Experience in developing and deploying large-scale machine learning systems.

- Strong sense of responsibility and good at communication and teamwork.

- Passionate about solving complex and challenging problems.

Preferred Qualifications

- Experience contributing to an open sourced machine learning framework (tensorflow / jax / pytorch / torchscript / mxnet / tensorrt).

- Experience in big data frameworks (e.g., Spark/Hadoop/Flink), experience in resource management and task scheduling for large scale distributed systems.

- Participated in Parameter Server system optimization, or index structure optimization for search systems.

- Strong background in one of the following fields: Hardware-Software Co-Design, High Performance Computing, ML Hardware Acceleration (GPU/RDMA) or ML for Systems.

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.

Interview Questions of Tech Lead Manager - Machine Learning Infrastructure at TikTok

Interview questions from TikTok that are similar to Tech Lead Manager - Machine Learning Infrastructure
View more interview questions from TikTok โ†’
banner icon
Prepare For Your Interview in 1 Week?
Equip yourself with possible questions that interviewers might ask you, based on your work experience and job description.
Get Started!

Achieve your dream job with our top-notch tools!

Resume Checker Illustration

Resume Checker

Our free resume checker analyzes the job description and identifies important keywords and skills missing from your resume in just a minute!

Check Now
Interview Preparation Illustration

AI InterviewPrep

Utilizing advanced AI, our tool generates tailored interview questions based on your industry, role, and experience. Practice and receive feedback on your answers in real time!

Check Now
Resume Builder Illustration

Resume Builder

Let us show you the differences between a bad, good, and great resume, and guide you in building a resume that helps you stand out to employers, ensuring you land your next position faster!

Check Now