Staff ML Engineer - Personalization & Production AI

Databricks Inc.; New York (NY); On-site; USD 190,000 - 253,000; Full time

Job summary

Databricks is seeking an senior engineer in NYC to help build CustomerLake, a data platform enabling enterprise ML personalization at-scale. You will work on production ML/LLM models, analyze model behavior, and drive real business value through personalization features.

Ideal candidates will have a strong product mindset, hands-on production experience, and comfort navigating ambiguous problems from concept to launch within a fast-moving team.

Qualifications

• Hands-on experience building and evaluating ML models for real product use cases.
• Strong Python and ML framework skills (e.g., PyTorch).
• Proficiency with modern ML/AI personalization approaches and production-quality systems.

Responsibilities

• Evaluate ML and LLM approaches for personalization use cases.
• Inspect and improve model behavior in production.
• Build evaluation frameworks to optimize for business value.
• Partner with product management, engineering, and design to deliver scalable solutions.
• Set technical foundations for ML/AI personalization across products.

Skills

Python

PyTorch

ML models

LLM systems

Job description

Databricks is seeking an senior engineer in NYC to help build CustomerLake, a data platform enabling enterprise ML personalization at-scale. You will work on production ML/LLM models, analyze model behavior, and drive real business value through personalization features.

Ideal candidates will have a strong product mindset, hands-on production experience, and comfort navigating ambiguous problems from concept to launch within a fast-moving team.