Senior Staff Applied AI Engineer - Context Retrieval
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.
About the job
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.
Skills
Information Retrieval, RAG, Bm25, Lucene, Elasticsearch, Opensearch, Faiss, Scann, Hnsw, LLMs, Embeddings, Query Rewriting, Re-Ranking, Agentic Systems, Ndcg
Similar jobs
ML Engineering jobsLeads the design and operation of reliable, scalable model infrastructure powering AI inference across multiple providers. Requires 7+ years of distributed-systems engineering experience, strong programming skills, and expertise in production reliability and cloud infrastructure.
Leads the design, implementation, integration, and field validation of tactical autonomy and multi-agent coordination capabilities for unmanned platforms. Requires 7+ years of relevant experience, production C++, technical leadership, and eligibility for a U.S. Secret clearance.
Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.
Leads development and integration of advanced maritime autonomy for USVs, UUVs, and cooperating UAVs, including motion planning, localization, safety, and multi-agent coordination. Requires staff-level technical leadership, substantial robotics experience, C++ and Python proficiency, and eligibility for a SECRET clearance.
Leads architecture and technical direction for agentic search systems combining LLMs, retrieval, and content-understanding pipelines for contract intelligence. The role requires 10+ years building production systems, deep search or LLM expertise, and strong cross-team technical leadership.