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

$145,000 - $250,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, and our mission is to inspire creativity and bring joy. Our team is responsible for developing state-of-the-art Natural Language Processing (NLP) and Machine Learning (ML) algorithms to enhance user experience, improve merchant service quality, and build 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. In this role, you will participate in the development of extensive knowledge graphs that represent 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 mission. Your responsibilities will include constructing knowledge graphs that encompass product understanding, content feedback, and category/brand construction. You will also be tasked with building knowledge graphs that represent both buyers and products, ensuring that our algorithms are robust and effective in delivering a superior user experience.

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 products, content, and feedback understanding.
• Build knowledge graphs of buyers and products.

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, with practical experience in relevant business scenarios 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% 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 for 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.