Staff Software Engineer - AI/ML
Core
Building CustomerLake, a Customer Data Platform on Databricks, to deliver enterprise-grade ML and AI personalization (recommendations, churn, lifetime-value modeling) to B2C and B2B brands.
Role type
Staff Machine Learning Engineer (0-to-1 product build)
Builds
CustomerLake platform and evaluation framework for personalization use cases
Domain
Data Intelligence / AI / Personalization
Deliverable
production ML models
Required skills
Full-loop ML/AI product shipping, LLM systems (RAG, prompt design, fine-tuning), Python, PyTorch, model evaluation and monitoring, personalization algorithms (recommendations, targeting, churn, LTV), product mindset, 0-to-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 strategy & mentorship
