Senior ML Engineer: LLM/NLP for E-commerce AI

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

Job summary

TikTok USDS Joint Venture LLC is seeking an experienced AI/ML engineer to architect and develop sophisticated LLM-based systems that process high-dimensional unstructured data across our ecommerce platforms. You will shape semantic understanding with fine-grained intent classification and entity extraction while collaborating with cross-functional teams in a fast-paced environment.

The role includes exploring Instruction Tuning, RLHF/DPO, and Contextual Prompt Engineering, and optimizing

Responsibilities

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

Skills

C++

Python

Go

Java

Education

MS or PhD in Computer Science, AI, Mathematics, or related field

Job description

TikTok USDS Joint Venture LLC is seeking an experienced AI/ML engineer to architect and develop sophisticated LLM-based systems that process high-dimensional unstructured data across our ecommerce platforms. You will shape semantic understanding with fine-grained intent classification and entity extraction while collaborating with cross-functional teams in a fast-paced environment.

The role includes exploring Instruction Tuning, RLHF/DPO, and Contextual Prompt Engineering, and optimizing