Staff ML Engineer, Personalization Platform

Neura Market; New York (NY); On-site; USD 192,000 - 260,000; Full time

Job summary

Databricks in New York City is hiring a senior ML/AI engineer to build CustomerLake, a data platform for personalization. You will evaluate ML/LLM approaches, push models to production, and design evaluation frameworks that drive real business value like purchases and retention.

You'll work closely with product, engineering, and design to turn ambiguous problems into scalable solutions and establish best practices as the team grows. 10+ years of engineering is expected.

Qualifications

• 10+ years of engineering experience in ML/AI products.
• Hands-on experience building and evaluating ML models and LLM systems.
• Experience with personalization such as recommendations or churn.
• Proficiency in Python and ML frameworks, with production monitoring.
• Familiarity with LLMs, retrieval-augmented generation, and evaluation.

Responsibilities

• Evaluate ML/LLM approaches for personalization use cases.
• Inspect model behavior in production; tune and improve from there.
• Build platform and evaluation framework to optimize business value.
• Push team toward new methods and scalable solutions.
• Collaborate with product, engineering, and design to turn ambiguous problems into MVPs.

Skills

ML model development

Model evaluation

Ownership

Product mindset

Data-driven decision making

Tools

Python

PyTorch

LLMs

RAG

Job description

Databricks in New York City is hiring a senior ML/AI engineer to build CustomerLake, a data platform for personalization. You will evaluate ML/LLM approaches, push models to production, and design evaluation frameworks that drive real business value like purchases and retention.

You'll work closely with product, engineering, and design to turn ambiguous problems into scalable solutions and establish best practices as the team grows. 10+ years of engineering is expected.