ML Engineer, E-Commerce AI & Multimodal Models

TikTok USDS Joint Venture; Seattle (WA); On-site; USD 130,000 - 246,000; Full time

Job summary

TikTok USDS Joint Venture LLC is seeking a talented AI/ML Engineer to architect and deploy LLM-based systems that process diverse, high-dimensional data for e-commerce applications. You will design fine-grained intent and entity extraction, apply advanced AI techniques, and optimize model serving for production scales.

You should have at least a master's or PhD in a related field and hands-on experience with C++, Python, Go, or Java, along with ML/NLP proficiency.

Qualifications

• MS or PhD in Computer Science, AI, Mathematics, or a related field.
• 1+ years of professional experience in Machine Learning, NLP, or Computer Vision.
• Proficiency in at least one of C++, Python, Go, or Java.

Responsibilities

• Architect and develop LLM-based systems to process high-dimensional unstructured data from diverse sources.
• Define and implement semantic understanding frameworks for fine-grained intent classification and entity extraction.
• Apply state-of-the-art techniques like Instruction Tuning, RLHF/DPO, and Contextual Prompt Engineering to improve models.
• Optimize model serving for production with quantization, distillation, and efficient attention (FlashAttention, vLLM).

Skills

LLM systems

NLP

Programming experience

Education

MS or PhD in Computer Science/AI/Mathematics

Tools

C++

Python

Go

Java

Job description

TikTok USDS Joint Venture LLC is seeking a talented AI/ML Engineer to architect and deploy LLM-based systems that process diverse, high-dimensional data for e-commerce applications. You will design fine-grained intent and entity extraction, apply advanced AI techniques, and optimize model serving for production scales.

You should have at least a master's or PhD in a related field and hands-on experience with C++, Python, Go, or Java, along with ML/NLP proficiency.