Staff Machine Learning Engineer, CustomerLake (ML/LLM)

What you'll do

• Evaluate and improve ML and LLM approaches for personalization, using production traces to diagnose and improve model behavior.
• Build the platform and evaluation framework to optimize for outcomes such as purchases, retention, and product usage.
• Establish ML/AI personalization practices and collaborate across product, engineering, and design to deliver scalable solutions.

What they ask for

• 10+ years of engineering experience, with a strong foundation in shipping and improving ML/AI products.
• Hands-on experience building and evaluating ML models and/or LLM systems for real product or business use cases.
• Proficiency in Python and modern ML frameworks (e.g., PyTorch), with experience evaluating models and monitoring AI quality in production.
• Familiarity with LLMs and generative AI, including RAG, prompt design, fine-tuning, and evaluation.

Nice to have

• Personalization experience based on customer behavior (ideal) or transactions (acceptable), such as recommendations, targeting, churn, or lifetime-value modeling.
• Experience in martech or an academic/research background.

Pay

$190,000 — $253,400 USD (salary range)

Tools

Python, PyTorch, LLMs, RAG

AI focus

Develop and evaluate ML and LLM systems for personalization, improve production model quality, and build evaluation methods focused on business outcomes.