Tech Lead, Machine Learning Engineer-TikTok Multimedia, Data Platform

$224,000 - $224,000/Yr

Tiktok - San Jose, CA

Full-time - Mid Level

San Jose, CA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

TikTok is the leading destination for short-form mobile video, with a mission to inspire creativity and bring joy. The company has a global presence with offices in major cities around the world. The Multimedia Architecture team is focused on optimizing user experience and cost for video on demand and live streaming systems. This team is dedicated to building intelligent and personalized multimedia systems that power the entire TikTok ecosystem. We are currently seeking a passionate Machine Learning Engineer to join our team. In this role, you will work closely with product and engineering teams to identify opportunities for improvement and enhance overall system performance and efficiency by integrating AI algorithms throughout the multimedia system. Your responsibilities will include designing service-side machine learning systems, implementing machine learning models, and validating their impact on the overall system through A/B testing. As a Machine Learning Engineer, you will develop production machine learning solutions that help understand user preferences and video content, predict video popularity, and manage system resource changes to create a world-class personalized multimedia experience. You will collaborate with engineering and product teams to shape the architecture roadmap by applying machine learning principles. Additionally, you will design and optimize pipelines for server-side machine learning models, which includes building real-time data pipelines, feature engineering, model optimization, and innovation. This role is crucial in ensuring that TikTok continues to provide a seamless and engaging user experience while leveraging the latest advancements in machine learning.

Responsibilities

• Develop production machine learning solutions to understand user preferences and video contents, predict video popularity and the change of system resources to build a world class personalized multimedia experience.
• Collaborate with engineering team and product team to shape the architecture roadmap applying ML.
• Design and optimize the pipelines for server-side machine learning models, including, but not limited to building real-time data pipelines, feature engineering, model optimization and innovation.

Requirements

• 5 years of industry experience developing machine learning models with business impact, and shipping ML solutions to production.
• A master's or PhD degree in computer science, data science, or a ML related field.
• Familiar with applied machine learning, such as classification, deep neural networks, transformers, multi-task learning, etc.
• Proficient in at least one programming language, such as Python and C++.
• Experience in Deep Learning Tools such as tensorflow/pytorch.
• Have a passion for learning and an eagerness to experiment with new techniques and good communication and a cooperative spirit.

Nice-to-haves

• Experience with recommendation systems, online advertising, search, and causal inference is preferred.
• Additional familiarity with LLM is a plus.

Benefits

• 100% premium coverage for employee medical insurance, approximately 75% premium coverage for dependents.
• Health Savings Account (HSA) with a company match.
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life and AD&D insurance plans.
• Flexible Spending Account (FSA) Options like Health Care, Limited Purpose and Dependent Care.
• 10 paid holidays per year plus 17 days of Paid Personal Time Off (PPTO) (prorated upon hire and increased by tenure).
• 10 paid sick days per year.
• 12 weeks of paid Parental leave.
• 8 weeks of paid Supplemental Disability.
• Mental and emotional health benefits through EAP and Lyra.
• 401K company match.
• Gym and cellphone service reimbursements.