JobsStaff Machine Learning Engineer, CustomerLake (ML/LLM)
Databricks logo

Staff Machine Learning Engineer, CustomerLake (ML/LLM)

Databricks

Location

New York City, NY

Type

Full-time

Posted

8/25/2026

Compensation

Not listed

Undergraduate with 5+ Years of Experience
H-1B FY202699.7% approval+19% YoY
💎 Strong sponsor

Job description

As a senior engineer at Databricks, you will be part of a small team in the NYC Engineering office focused on building CustomerLake, a Customer Data Platform. This role emphasizes developing ML and AI personalization capabilities for companies using Databricks. You will work in a 0-to-1 environment, combining startup excitement with the resources of a tech leader. Your contributions will help optimize customer experiences and drive real business value.

Requirements

  • 10+ years of engineering experience with a strong foundation in shipping and improving ML/AI products.
  • Hands-on experience building and evaluating ML models and/or LLM systems for real product use cases.
  • Experience with personalization based on customer behavior or transactions.
  • Proficiency in Python and modern ML frameworks such as PyTorch.
  • Familiarity with LLMs and generative AI techniques.
  • A demonstrated product mindset with the ability to translate ambiguous customer problems into MVPs.
  • High ownership and bias for action in 0-to-1 environments.

Responsibilities

  • Evaluate ML and LLM approaches for CustomerLake's personalization use cases and improve model quality over time.
  • Inspect individual traces and understand model behavior in production to tune and enhance performance.
  • Build the platform and evaluation framework to optimize for real business value like purchases and retention.
  • Encourage the team to explore new directions and tackle novel methods.
  • Collaborate closely with product management, engineering, and design to create scalable solutions.
  • Set the technical foundation and best practices for ML/AI personalization as the team grows.

H-1B filing history

Public USCIS petition and DOL LCA counts · latest USCIS FY2026, LCA FY2026

Filing entity: Databricks Inc

As of Aug 23, 2026

Initial approvals

83

FY2026

Approval rate

99.7%

FY2026

LCA certified

174

FY2026

Entry-level share

4.0%

FY2026

Initial approvals YoY

+19%

Trend

LCA certified YoY

-54%

Trend

Initial approvals by fiscal year

Approval rate by fiscal year

Continuing vs initial approvals

LCA certified positions by quarter

LCA certified positions by fiscal year

Based on public USCIS and DOL filings; not a sponsorship guarantee.

Is this posting expired or inaccurate?