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.
229k – 343k/yr
Hybrid10+ YOEML Engineering
About the role
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 RetrievalRAGBm25LuceneElasticsearchOpensearchFaissScannHnswLLMsEmbeddingsQuery RewritingRe-RankingAgentic SystemsNdcg
Builds production-grade generative AI systems powering patient engagement, provider workflows, and clinical operations in mental healthcare. Requires 10+ years software engineering, deep Python/cloud expertise, and 2+ years scaling AI products with foundation models and LLM patterns.
Sr. Staff ML Engineer building backend systems, statistical models, and experiments to optimize Pinterest's ads marketplace and balance long/short-term objectives. Requires MS/PhD (or equiv), 7+ years experience, strong software engineering and math skills.
Technical lead defining ML strategy and systems to improve Pinterest's content ecosystem health, marketplace dynamics, and long-term outcomes. Requires strong ML fundamentals, marketplace thinking, and experience leading high-scale, multi-objective optimization projects.
228k – 469k/yr
Remote8+ YOEML Engineering
Staff Machine Learning Engineer
RedditSan Francisco, CA
Staff Machine Learning Engineer designing, training, and deploying advanced ML models (DNNs, transformers, RL) for Reddit's large-scale online advertising ranking, optimization, and marketplace systems. Requires Master's, 3+ years experience, technical leadership, and expertise in production ML systems.
Technical leader for Reddit's Notifications Relevance ML systems, driving large-scale recommendation systems spanning retrieval, ranking, budget optimization, and LLM-powered experiences.