# Senior Staff Applied AI Engineer - Context Retrieval

**Company:** [Databricks](https://hotfix.jobs/companies/databricks)
**Location:** Mountain View, CA, San Francisco, CA
**Role:** ML Engineering
**Salary:** $229k – $343k/yr
**Experience:** 10+ years
**Skills:** Information Retrieval, RAG, Bm25, Lucene, Elasticsearch, Opensearch, Faiss, Scann, Hnsw, LLMs, Embeddings, Query Rewriting, Re-Ranking, Agentic Systems, Ndcg
**Posted:** 2026-05-07

> Builds retrieval stack and search subagents for Databricks AI agents, handling query understanding, hybrid retrieval across structured/unstructured enterprise data, and evaluation. Requires 10+ years in production IR/RAG systems and agentic workflows.

## Job Description

## What You Will Do

- Build the full retrieval stack from scratch. Own the end-to-end system: query understanding, content understanding and indexing, hybrid retrieval, ranking, and evaluation.
- 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).
- Connect to the SaaS surface area customers actually use. Build connectors and retrieval adapters for the systems where enterprise knowledge lives.
- Optimize for two consumers at once. Retrieval must serve both LLMs (grounded, token-efficient, hallucination-resistant context) and humans (intuitive, explainable discovery).
- Crack query understanding for agents. Build query rewriting, decomposition, intent classification, and entity resolution tuned for multi-turn agentic workflows.
- Crack content understanding at scale. Build the pipelines that extract structure, entities, embeddings, summaries, and metadata from every supported asset type.
- Build search subagents that reason about retrieval. Design the agentic layer that decides what context is needed, which sources to query, how to decompose and route the search, and whether the retrieved content is sufficient.
- Build the evaluation flywheel for both retrieval and subagents. Stand up offline evals (nDCG, MRR, Recall@K, Precision@K), LLM-as-judge harnesses, human-in-the-loop labeling, and online experimentation.
- Set technical direction and grow the team. Set the multi-year roadmap, mentor senior engineers, partner with Research, Product, and Platform leaders.

## What We're Looking For

- 10+ years of software engineering experience, with significant time spent building production retrieval, search, or RAG systems at scale.
- Deep **Information Retrieval (IR)** expertise: lexical retrieval (**BM25**, **Lucene/Elasticsearch/OpenSearch**), dense retrieval (embeddings, ANN indexes — **FAISS**, **ScaNN**, **HNSW**), hybrid retrieval, and learning-to-rank.
- Hands-on experience with modern LLM-era retrieval: **RAG** architectures, query rewriting, re-ranking with cross-encoders, long-context strategies, and grounding techniques.
- Experience designing agentic systems on top of retrieval — search planners, multi-hop / iterative retrieval, self-reflection and sufficiency checks, tool-using agents.
- Strong grasp of relevance evaluation: nDCG, MRR, Precision@K, Recall@K; offline/online experimentation; LLM-as-judge frameworks; building human labeling pipelines.
- Experience working across structured and unstructured data — indexed and ranked over tables, code, and documents in the same system.
- Track record of building 0→1: stood up a retrieval system from an empty repo, made foundational architectural decisions.
- Demonstrated ability to operate as a technical leader: setting direction across teams, mentoring senior engineers.

## Nice to Have

- Experience building retrieval over enterprise SaaS sources (permissions, freshness, multi-tenancy, ACL-aware indexing).
- Background in agentic systems, tool use, or multi-turn retrieval for LLM agents.
- Contributions to open-source IR/search projects, or publications at SIGIR, KDD, WWW, EMNLP.
- Experience training or fine-tuning embedding models, rerankers, or query understanding models.

## Similar roles

- [Sr. Staff AI Engineer (Remote)](https://hotfix.jobs/jobs/5d8a6424-73ee-4190-9f13-402bc73af4cf) - Rula - Remote - $229k – $284k/yr
- [Sr. Staff Machine Learning Engineer, Monetization Engineering](https://hotfix.jobs/jobs/147f2a8a-cc72-414d-ac85-1f9171a29990) - Pinterest - San Francisco, CA - $228k – $469k/yr
- [Sr. Staff Machine Learning Engineer, Content Ecosystem](https://hotfix.jobs/jobs/8fd34968-1a85-471d-b2aa-1ab289b614bb) - Pinterest - Remote - $228k – $469k/yr
- [Staff Machine Learning Engineer](https://hotfix.jobs/jobs/ebc73812-e806-404d-9f4e-435da967dbf5) - Reddit - Remote - $230k – $322k/yr
- [Staff Machine Learning Engineer, Notifications Relevance](https://hotfix.jobs/jobs/5df4bbf6-982b-4563-aaee-d477695b1557) - Reddit - Remote - $230k – $322k/yr

**Apply:** https://hotfix.jobs/jobs/5ed8e84a-a7c1-46c1-9607-b9a2152b8fa0
**Canonical:** https://hotfix.jobs/jobs/5ed8e84a-a7c1-46c1-9607-b9a2152b8fa0