Machine Learning Engineer - Trust and Safety, NLP

$145,000 - $355,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 Trust and Safety team is rapidly expanding and is responsible for developing machine learning models and systems that protect users from negative content. This team plays a crucial role in safeguarding billions of users and publishers globally, utilizing state-of-the-art machine learning technologies to manage the vast amounts of data generated on the platform. The continuous efforts of this team ensure that TikTok provides an exceptional user experience while promoting a safe environment for all users. As a Machine Learning Engineer in the Trust and Safety team, you will be tasked with leading the development of content understanding models for content moderation. This includes working with advanced techniques such as unimodal and multimodal content classifiers, graph models, representation learning models, and large language models (LLM). You will collaborate closely with product teams to define business objectives and enhance the trust and safety strategy, as well as with engineering teams to deploy machine learning models and design online model workflows at scale. Additionally, you will work alongside data scientists to identify key challenges and propose data-driven strategies to address them. The ideal candidate will possess a strong foundation in machine learning and software engineering, with a passion for tackling complex safety and integrity issues on an internet scale. You will have the opportunity to take initiative, design, and develop advanced machine learning solutions that will be deployed directly to TikTok's global platform, making a significant impact on user safety and content moderation.

Responsibilities

• Lead and build content understanding models for content moderation.
• Work on state-of-the-art content understanding techniques including unimodal/multimodal content classifiers, graph models, representation learning models, and LLM.
• Collaborate with product teams to define business objectives and improve trust and safety strategy.
• Collaborate with engineering teams to deploy machine learning models, design online model workflow, and implement serving pipeline at scale.
• Collaborate with data scientist team to understand the key challenges and propose data-driven strategies.

Requirements

• BS/MS degree in Computer Science, Statistics, Operation Research, Applied Mathematics, Physics or similar quantitative fields, with 2+ years of related experience.
• Hands-on experience in one or more of the following areas: machine learning (CV/NLP/Multimodal/Graph/LLM), statistical models and applied mathematical methods.
• Strong coding skills, especially in Python and similar languages.
• Curiosity towards new technologies and entrepreneurship, good communication and teamwork skills.

Nice-to-haves

• Experience in LLM, multimodal modeling, representation learning and/or publications in top AI/ML conferences are a plus.
• Understanding of the business value of content understanding and content moderation.
• Experience in developing online ML systems including ranking, retrieval.

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) and 10 paid sick days per year.
• 12 weeks of paid Parental leave and 8 weeks of paid Supplemental Disability.
• Mental and emotional health benefits through EAP and Lyra.
• 401K company match, gym and cellphone service reimbursements.