Machine Learning Engineer(NLP/CV/Multimodal), 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, and our mission is to inspire creativity and bring joy. Our team is responsible for developing state-of-the-art NLP/ML algorithms and strategies to improve user consumption experience, inspire merchants' service quality and revenue, and build a fair and flourishing ecosystem on our E-commerce Platform. More specifically, our team is responsible for the algorithms of Product Knowledge Graphs under TikTok's global e-commerce business. In this role, you will participate in the development of massive knowledge graphs of real-world products to support feed ranking, recommendations, and ads. You will collaborate with product managers, data scientists, and the product strategy & operation team to define product strategies and features. Your responsibilities will include knowledge graph construction, which involves understanding product/content/feedback and constructing categories, brands, and SPUs. You will also be tasked with constructing knowledge graphs of buyers and products, which is crucial for enhancing the overall user experience on our platform.

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, multimodal, 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.
• Familiarity with processing methods and optimization experience on domain adaptation, small sample construction, text mining, and unsupervised/semi-supervised issues.
• Knowledge of commonly used machine learning and deep learning algorithms, understanding basic network model structures (DNN/LSTM/CNN, etc.) and text representation methods (LDA/Word2Vec/ELMo/GPT/BERT, etc.).
• 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.