Back to jobs

Staff Machine Learning Engineer, CustomerLake (ML/LLM)

New York City, New York

Live, first seen by us 5 days ago
Pay
USD 190,000–253,400/yearAnnual period assumed — pay source
Local Pay Range $190,000—$253,400 USD About Databricks
Read the full posting
Work setting
Not stated
Employment
Not stated

What you’ll work on

Full posting
  • Evaluate ML and LLM approaches for personalization use cases and improve model quality over time

  • Inspect production model traces to understand model reasoning and tune behavior

  • Build platform and evaluation framework tied to business value like purchases and retention

  • Partner with product, engineering, and design to turn ambiguous customer problems into solutions

  • Set technical foundation and best practices for ML/AI personalization work

From the employer’s posting
Evaluate ML and LLM approaches for CustomerLake's personalization use cases
inspect individual traces, understand how the models reason, and tune and improve from there
Build the platform and evaluation framework that let CustomerLake customers optimize for real business value such as purchases, retention, and product usage
Partner closely with product management, engineering, and design to turn ambiguous customer problems into scalable, trustworthy solutions
Set the technical foundation and best practices for our ML/AI personalization work

About Databricks

More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents.

In the employer’s words · Read in context

Apply at Databricks

Posting record

Report an error
Status in our records
Live
First seen by us
Oct 29, 20265 days ago
Recorded sightings
7
Last seen by us
Nov 4, 2026
Reposted
once since Jul 6, 2026

These dates show when we found the posting. Check the employer’s website to confirm it is still accepting applications. We count a repost when the same title and location comes back as a new listing within 60 days of the last one coming down. Some hiring systems expire listings on a schedule, so repost rates vary by system.

How this job compares

Pay

In line with similar jobs

$190K–$253K against $166K–$240K, the middle half of what similar US postings state.

From 1,279 postings in the last year. Stated ranges, not salaries paid.

Time since we first saw it

Day 5

Early. Similar jobs usually come down between day 6 and day 69.

From 4,284 similar postings that came down. Coming down is not always a hire.

Databricks is hiring

19 open jobs

Open jobs by the week we first saw them, last 12 weeks.

  • Hiring at a steady pace
  • Takes jobs down at a usual pace
  • 11% of open jobs were reposted
Hiring record

See how this role fits your experience

Add your resume and we’ll mark which of these tools you already use, and point out anything in the posting that could rule you out.

  • Python
  • Databricks
  • PyTorch
Where the posting names them
- Experience with personalization based on customer behavior (ideal) or transactions (acceptable), such as recommendations, targeting, churn, or lifetime-value modeling - Proficiency in Python and modern ML frameworks (e.g., PyTorch), with hands-on experience in model evaluation and monitoring AI quality in production - Familiarity with LLMs and generative AI, including techniques like retrieval-augmented generation (RAG), prompt design, fine-tuning, and evaluation
P-1607 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best Data Intelligence Platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to tackle technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As one of the first engineers in the NYC Engineering office, you'll join a small, nimble team building new products from the ground up. We're building CustomerLake, the Customer Data Platform on Databricks, to bring enterprise-grade ML and AI personalization to every company whose data already lives on Databricks. The best B2C and B2B brands have historically relied on in-house ML/AI teams to power personalization, recommendations, churn and lifetime-value modeling, and audience targeting. Our goal is to deliver that same capability to companies that don't have an in-house team but already have their data in order on Databricks. This is a true 0-to-1 environment, combining the excitement of a startup with the resources of a tech leader like Databricks.
At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best Data Intelligence Platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to tackle technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started. As one of the first engineers in the NYC Engineering office, you'll join a small, nimble team building new products from the ground up. We're building CustomerLake, the Customer Data Platform on Databricks, to bring enterprise-grade ML and AI personalization to every company whose data already lives on Databricks. The best B2C and B2B brands have historically relied on in-house ML/AI teams to power personalization, recommendations, churn and lifetime-value modeling, and audience targeting. Our goal is to deliver that same capability to companies that don't have an in-house team but already have their data in order on Databricks. This is a true 0-to-1 environment, combining the excitement of a startup with the resources of a tech leader like Databricks. The impact you'll have:
Pay Range Transparency Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here. Local Pay Range
$190,000—$253,400 USD About Databricks Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram.
About Databricks Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram. Benefits
Benefits At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here. Our Commitment to Diversity and Inclusion
Our Commitment to Diversity and Inclusion At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics. Compliance

Your next step

Complete your application on databricks.com. The employer’s form will show what is required.

Applying through Greenhouse

  • No account needed: the form is one page, with an “Autofill my application” option that reads your resume.
  • Greenhouse doesn’t read resumes over 2.5 MB, or details placed in a header, footer, text box, table or second column. The file still attaches, but a recruiter has to type you in by hand.

Already applied? Track this application

Has Databricks posted this job before?

Yes. We’ve followed one other Staff Machine Learning Engineer, CustomerLake (ML/LLM) job there since July 2026. It came down after 109 days.

  1. New YorkCame down after 109 days, first seen Jul 6
  2. This jobUp 6 days so far, first seen Oct 29

About the role

We file this job under ML Engineer, by the work the posting describes.

ML engineers put machine learning models into products. They write the code that trains, ships, serves and monitors models so a recommendation or prediction works for real users.

Postings ask for strong Python and software engineering, a framework such as PyTorch, cloud services, and more and more often experience with large language models.

A data scientist often works out which model to build; an ML engineer makes it run reliably in production.

US government figures for computer and information research scientists

Median yearly pay, May 2025
$140,300
Expected change, 2025 to 2035
21.8% more jobsagainst 3.5% for all jobs

The government has no code for ML engineers, so these figures are for the closest group, computer and information research scientists.

Source: US Bureau of Labor Statistics, Occupational Employment and Wage Statistics (May 2025) and Employment Projections (2025 to 2035). Occupation code 15-1221.

Job description

View original posting ↗

P-1607

At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best Data Intelligence Platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to tackle technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started.

 

As one of the first engineers in the NYC Engineering office, you'll join a small, nimble team building new products from the ground up. We're building CustomerLake, the Customer Data Platform on Databricks, to bring enterprise-grade ML and AI personalization to every company whose data already lives on Databricks. The best B2C and B2B brands have historically relied on in-house ML/AI teams to power personalization, recommendations, churn and lifetime-value modeling, and audience targeting. Our goal is to deliver that same capability to companies that don't have an in-house team but already have their data in order on Databricks. This is a true 0-to-1 environment, combining the excitement of a startup with the resources of a tech leader like Databricks.

 

The impact you'll have:

  • Evaluate ML and LLM approaches for CustomerLake's personalization use cases, push the models and algorithms forward, and continuously improve quality over time
  • Go deep on how models behave in production: inspect individual traces, understand how the models reason, and tune and improve from there
  • Build the platform and evaluation framework that let CustomerLake customers optimize for real business value such as purchases, retention, and product usage, not vanity metrics like email opens and clicks
  • Push the team toward new directions and novel methods worth tackling, not just optimizing what already exists
  • Partner closely with product management, engineering, and design to turn ambiguous customer problems into scalable, trustworthy solutions
  • Set the technical foundation and best practices for our ML/AI personalization work as we grow this into several roles across our products over the next 1-2 years

 

What we look for:

  • 10+ years of engineering experience, with a strong foundation across the full loop of 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; your understanding is practical, not purely academic, and you can make models work well inside a product
  • Experience with personalization based on customer behavior (ideal) or transactions (acceptable), such as recommendations, targeting, churn, or lifetime-value modeling
  • Proficiency in Python and modern ML frameworks (e.g., PyTorch), with hands-on experience in model evaluation and monitoring AI quality in production
  • Familiarity with LLMs and generative AI, including techniques like retrieval-augmented generation (RAG), prompt design, fine-tuning, and evaluation
  • A demonstrated product mindset, with the ability to translate ambiguous customer problems into scrappy MVPs and iterate quickly based on data and user feedback
  • High ownership and bias for action in 0-to-1 environments: comfortable making pragmatic trade-offs, operating with incomplete information, and driving projects from idea through launch and adoption

 

Nice to have:

  • Experience in martech, ideally a go-to-market or business use case with an analytical (rather than purely transactional) angle
  • An academic or research background that can help us innovate and develop novel methods

 

Pay Range Transparency

Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles.  Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here.

 

Local Pay Range
$190,000—$253,400 USD

About Databricks

Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram.

Benefits

At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here.

Our Commitment to Diversity and Inclusion

At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics.

Compliance

If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

Source details

View original posting ↗

Source notes

Source excerpts

Selected passages from the saved posting. Check the full description for conditions and exceptions.

Pay
Local Pay Range $190,000—$253,400 USD About Databricks
Location & working pattern

New York City, New York

Working pattern and location restrictions need checking in the full posting.

Work authorization

No clear work-authorization passage found. Eligibility is unconfirmed.

Find answers in the posting

AI
How answers work

AI selects complete passages from this posting. Check them for conditions and exceptions.

Uses this posting and your question. No profile needed.