Machine Learning Engineer, NLP - TikTok E-commerce Knowledge Graph

$137,750 - $237,500/Yr

Tiktok - Seattle, WA

Full-time - Mid Level

Seattle, WA

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. Our team is dedicated to developing state-of-the-art Natural Language Processing (NLP) and Machine Learning (ML) algorithms that enhance user experiences, improve merchant service quality, and foster a thriving ecosystem on our E-commerce platform. Specifically, we focus on the algorithms that power Product Knowledge Graphs within TikTok's global e-commerce business. As an NLP Algorithm Engineer, you will play a crucial role in constructing massive knowledge graphs of real-world products, which are essential for supporting feed ranking, recommendations, and advertisements. You will collaborate closely with product managers, data scientists, and the product strategy and operations team to define product strategies and features that align with our goals. In this position, you will be responsible for knowledge graph construction, which includes understanding products, content, and feedback, as well as building category, brand, and Stock Keeping Unit (SPU) structures. You will also construct knowledge graphs that represent buyers and products, ensuring that our algorithms are robust and effective in delivering a superior user experience. This role requires a blend of technical expertise, creativity, and collaboration, as you will be working in a dynamic environment that encourages innovation and the pursuit of bold ideas.

Responsibilities

• Participate in the development of massive knowledge graphs of real-world products to support feed ranking, recommendations, and ads.
• Collaborate with product managers, data scientists, and the product strategy & operation team to define product strategies and features.
• Construct knowledge graphs of buyers and products.
• Understand product/content/feedback to aid in category/brand/SPU construction.

Requirements

• Bachelor's degree in Computer Science or related technical field.
• 3+ years of working experience in machine learning, NLP, or computer vision.
• Experience with software development in at least one of the following programming languages: C++, Python, Go, Java.
• Good sense of teamwork and communication skills; practical experience in relevant business scenarios is preferred.

Nice-to-haves

• Proficient in using at least one mainstream deep learning framework such as TensorFlow or PyTorch, with an understanding of distributed training and distillation acceleration.
• Experience in text classification, text matching, sequence labeling, and knowledge graph construction.
• Familiarity with processing methods and optimization experience in domain adaptation, small sample construction, text mining, and unsupervised/semi-supervised learning.
• Knowledge of commonly used machine learning and deep learning algorithms, including basic network model structures (DNN/LSTM/CNN) and text representation methods (LDA/Word2Vec/ELMo/GPT/BERT).
• Experience in large-scale text data processing or cleaning using tools like Hadoop, Spark, Hive, or Flink.

Benefits

• 100% premium coverage for employee medical insurance, approximately 75% for dependents, and a Health Savings Account (HSA) with company match.
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans.
• Flexible Spending Account (FSA) options for 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, and 8 weeks of paid Supplemental Disability.
• Mental and emotional health benefits through EAP and Lyra.
• 401K company match, gym, and cellphone service reimbursements.