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.
Responsibilities:
- 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.
Qualifications
- 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.
