Staff Software Engineer, AI Retrieval & Grounded Answers

Datasite

New York (NY)

Hybrid

USD 170,000 - 297,000

Full time

3 days ago
Be an early applicant

A complete application in a minute

Tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Health insurance
Retirement plan
Paid time off

Job summary

Datasite is hiring a hands-on Staff Software Engineer to own the retrieval layer used by live-deal AI agents. You will ensure grounded, current context using Python in backend services, APIs, and large-scale data pipelines.

Work with Product and Data Science to balance quality, latency, and cost while scaling document processing and vector search. You will mentor engineers, shape context across teams, and support secure multi-tenant access in a hybrid setup with in-office collaboration on

Qualifications

  • 6+ years of building and operating production software with a track record of leading complex technical work across teams.
  • Strong hands-on Python, with experience in backend services, APIs, and large-scale data pipelines.
  • Experience with production LLM applications that depend on retrieval: tool calling, embeddings, vector search, or context management.
  • Strong foundations in permissions, multi-tenant data, and access control on what an agent can retrieve.
  • Clear product and technical judgment; balancing answer quality, latency, and cost, with ability to explain tradeoffs to engineering/product leaders.

Responsibilities

  • Own retrieval for agents that work on live deals.
  • Ground each answer in the right documents and data, balancing quality, speed, and cost.
  • Scale platform to process large documents and data tables so an agent can reason over content beyond a single pass.
  • Shape the context agents depend on and evolve it as deal workflows grow.
  • Build the search tools agents call and support content indexed and not yet indexed.
  • Return the snippet, citation, and location a dealmaker can trust.
  • Ensure agent context aligns with data rooms and access controls; prevent reading restricted content.
  • Decide when models may answer from retrieved context and when to re-check results.
  • Mentor engineers and set technical direction for retrieval across teams.

Skills

Python
Backend APIs
Data pipelines
LLM retrieval
Permissions & access
Product judgment

Tools

Vector store
Embeddings
RAG
Tool interfaces

Job description

Datasite is hiring a hands-on Staff Software Engineer to own the retrieval layer used by live-deal AI agents. You will ensure grounded, current context using Python in backend services, APIs, and large-scale data pipelines.

Work with Product and Data Science to balance quality, latency, and cost while scaling document processing and vector search. You will mentor engineers, shape context across teams, and support secure multi-tenant access in a hybrid setup with in-office collaboration on

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff Software Engineer - Agentic Search & Retrieval (Hybrid)
Staff Software Engineer - Agentic Search & Retrieval (Hybrid)

Datasite • Minneapolis (MN)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off
+1
Staff AI Retrieval Engineer — Grounded Agent Context
Staff AI Retrieval Engineer — Grounded Agent Context

Perot Jain • Minneapolis (MN)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off to employees
Staff Software Engineer: AI-Driven Backend & Platform Leader
Staff Software Engineer: AI-Driven Backend & Platform Leader

Datasite • Minneapolis (MN)

Hybrid
USD 170,000 - 297,000
Senior/Staff Search & Retrieval Engineer for AI Agents
Senior/Staff Search & Retrieval Engineer for AI Agents

Actively • San Francisco (CA)

On-site
USD 160,000 - 220,000
Health, Dental, Vision
Unlimited PTO
Catered meals
+4
Staff Software Engineer, Agentic Search & Retrieval
Staff Software Engineer, Agentic Search & Retrieval

Perot Jain • Minneapolis (MN)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off to employees
Staff Software Engineer, Agentic Search & Retrieval
Staff Software Engineer, Agentic Search & Retrieval

Datasite • New York (NY)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off
Staff Software Engineer, Agentic Search & Retrieval
Staff Software Engineer, Agentic Search & Retrieval

Datasite • Minneapolis (MN)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off
+1
Senior Staff Applied AI Engineer, Retrieval & Agentic IR
Senior Staff Applied AI Engineer, Retrieval & Agentic IR

Databricks Inc. • San Francisco (CA)

Hybrid
USD 229,000 - 343,000
Staff Software Engineer - AI-Driven, Scalable Backend
Staff Software Engineer - AI-Driven, Scalable Backend

Openkrill • United States

Remote
USD 180,000 - 240,000
Staff Software Engineer: AI Agents & Distributed Systems (Hybrid)
Staff Software Engineer: AI Agents & Distributed Systems (Hybrid)

Datasite • New York (NY)

Hybrid
USD 170,000 - 297,000
Health insurance
Retirement plan
Paid time off
+1