CareerPlan

Staff Machine Learning Engineer, CustomerLake (ML/LLM)

San Francisco, CaliforniaFull-time$190k–$253k2026-10-29 → 2026-11-10

Core

Building CustomerLake, a Customer Data Platform on Databricks, to deliver enterprise-grade ML and AI personalization (recommendations, churn, lifetime-value modeling) to companies using Databricks.

Role type

Staff Machine Learning Engineer (0-1 product build)

Builds

CustomerLake platform and evaluation framework for ML/AI personalization

Domain

Data Intelligence / B2C & B2B Personalization

Deliverable

production ML models

Required skills

Full-loop ML/AI product shipping, ML/LLM model building and evaluation, Python, PyTorch, LLM techniques (RAG, prompt design, fine-tuning), production model monitoring, product mindset, 0-1 environment ownership

Preferred skills

Martech experience, academic/research background

Technologies

Python, PyTorch, LLMs, RAG

Responsibilities

Evaluate ML/LLM approaches for personalization, inspect model traces in production, build evaluation frameworks for business value, push team toward novel methods, partner with product/engineering/design, set technical foundation and best practices

Seniority

Staff, hands-on IC with strategic scope