databricks

Senior Staff Applied AI Engineer - Context Retrieval

Apply Now

At a Glance

Location
Mountain View, California; San Francisco, California
Experience
10+ years
Posted
2026-05-07T17:03:24-04:00

Key Requirements

Domain Knowledge

  • Engineering
  • SaaS

Benefits & Perks

Health Insurance

e strive to provide comprehensive benefits and perks that meet the needs of all of

Requirements

10+ years of software engineering experience, with significant time spent building production retrieval, search, or RAG systems at scale.

: lexical retrieval (BM25, Lucene/Elasticsearch/OpenSearch), dense retrieval (embeddings, ANN indexes — FAISS, ScaNN, HNSW), hybrid retrieval, and learning-to-rank.

Hands-on experience with

: RAG architectures, query rewriting, re-ranking with cross-encoders, long-context strategies, and grounding techniques that reduce hallucination.

Experience designing

agentic systems on top of retrieval

Responsibilities

Own the end-to-end system: query understanding, content understanding and indexing, hybrid retrieval, ranking, and evaluation.

Make the architectural calls that will define how Databricks agents access context for years to come.

Retrieve across heterogeneous data — structured and unstructured.

Index and rank across structured assets (tables, columns, SQL queries, dashboards, code, notebooks, jobs) and unstructured content (docs, wikis, tickets, chat, images, video, audio).

Each modality has its own signals — design retrieval that exploits them rather than flattens them.

Connect to the SaaS surface area customers actually use.

About the Company

Databricks agents are only as good as the context they can retrieve. Whether an agent is answering a question about last quarter's revenue, debugging a failing job, generating SQL against a 10,000-table lakehouse, or summarizing a Wiki page, its quality is bounded by what it can find — and how well it understands what it finds.