Tech Expert - Machine Learning Infrastructure

Responsibilities

About Bytedance
Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of more than a dozen products, including TikTok, as well as platforms specific to the China market, including Toutiao, Douyin, and Xigua, ByteDance has made it easier and more fun for people to connect with, consume, and create content.

Why Join Us
Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. To us, every challenge, no matter how ambiguous, is an opportunity; to learn, to innovate, and to grow as one team. Status quo? Never. Courage? Always. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users 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/orchestration/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.

Bytedance 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 Bytedance, 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.