Machine Learning Engineer (NLP/Multimodal) - E-commerce - San Jose

$136,800 - $280,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. As part of our E-commerce division, we are looking for a Machine Learning Engineer specializing in Natural Language Processing (NLP) and multimodal applications. This role is crucial in developing innovative algorithms and techniques that enhance user engagement and satisfaction on TikTok Shop, which connects customers with quality sellers and products through live-streaming and short videos. In this position, you will be responsible for the design and development of NLP-related systems. This includes building an intent classifier by designing a taxonomy, managing high-quality labeled data, training machine learning models such as BERT and GPT, and monitoring their online performance. You will also create prediction models that map user shopping journeys to chatbot actions, utilizing advanced techniques like Boosting and deep CTR models. Additionally, you will develop generation models that understand business requests and fine-tune large language models with billions of parameters. Collaboration with product and operation teams will be essential to quantitatively assess the impact of our chatbot and align business improvements with model optimizations. We are a team of applied machine learning engineers and data scientists dedicated to solving real-world problems in E-commerce through large-scale machine learning. If you are passionate about using your skills to drive business impact and improve user experiences, we invite you to join us in this exciting journey.

Responsibilities

• Build the intent classifier in a full-stack manner by designing taxonomy, creating and managing high-quality labeled data, training machine learning models, and monitoring their online performance.
• Develop prediction models that map the shopping journey of users to chatbot actions, including suggesting common question-answers and transferring users to human agents.
• Create generation models by understanding business requests, exploring proper instructions, developing high-quality data, and fine-tuning large language models.
• Collaborate closely with product and operation teams to quantitatively assess the impact of chatbots and bridge business improvement actions to model optimization actions.

Requirements

• Bachelor's degree or above in computer science or related field.
• Solid coding skills with proficiency in Python, Go, or C++, and ability to develop in a Linux environment.
• 2+ years of experience with a solid foundation in data structures/algorithms, proficient in machine learning/deep learning theory, and rich practical experience.
• Familiarity with 1-2 areas in natural language processing, computer vision, multimodal, graph algorithms, search algorithms, text/data mining, and large language models (LLM).
• Excellent analytical and problem-solving skills, with a passion for tackling challenging problems.
• Good team spirit and strong communication skills.

Benefits

• 100% premium coverage for employee medical insurance
• Approximately 75% premium coverage for dependents
• 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 healthcare and dependent care
• 10 paid holidays per year
• 17 days of Paid Personal Time Off (PPTO)
• 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