AI Context & Data Infrastructure Engineer

Town

San Francisco (CA)

On-site

USD 150,000 - 210,000

Full time

11 days ago
Application generator

Stand out for this role — generate a tailored resume and cover letter in about a minute.

Get past ATS filters

Job summary

Town is building the most personalized AI assistant, starting with a unified search and retrieval layer to surface the right context across messaging, documents, and apps. You’ll own the data model and pipelines that keep context fresh, fast, and cost-effective at scale.

This greenfield role invites you to design the foundations, implement end-to-end retrieval, and drive production-quality data infrastructure for a world-class AI assistant.

Qualifications

  • Experience with lexical and semantic search components (BM25, embeddings, ANN/vector indexes, hybrid retrieval, ranking).
  • Experience building large-scale data infrastructure, including realtime/streaming and batch pipelines.
  • Ability to optimize retrieval for latency, cost, and freshness at scale.
  • Comfort working in greenfield environments where foundations don’t exist yet.
  • Interest in taking systems from prototype to production at scale.
  • Bonus experience with ranking, knowledge graphs, or retrieval for LLM/agentic systems.

Responsibilities

  • Build the search and retrieval layer that puts the right context at every Townie’s fingertips, the moment it’s needed — one shared layer the whole product pulls from instead of refetching context on its own.
  • Combine lexical and semantic search and own the tradeoffs between them: vector vs. lexical, precompute vs. fetch, hybrid retrieval, and ranking.
  • Design the durable data model the assistant's work is built on, so context is relevant, fast, and cost-effective.
  • Build and operate the pipelines behind it — realtime/streaming and batch — that keep the index fresh and correct as the underlying data changes.
  • Stand up the indexing and storage layer and keep it fast and reliable at scale: latency, cost, freshness, and completeness.
  • Lay the groundwork for relating content by meaning across everything the assistant knows — the start of a knowledge graph of people, companies, projects, and connections.

Skills

Lexical search
Semantic search
Data infrastructure

Job description

About Town

Town (town.com) is AI that starts from who you are. We build a persistent model of your identity, your voice, your judgment, your relationships, and your priorities, and use it to do real work on your behalf across every tool where you operate: email, calendar, documents, Slack, and more. Town doesn't wait for you to prompt it. It observes, learns, and acts. The more you use it, the more it becomes an extension of you.

Town was founded by Jean-Denis Greze (CEO), former CTO of Plaid, and Tony Vincent (CPO), former Director of Applied AI Product at Google. We're a small, talent-dense team backed by Andreessen Horowitz, Forerunner Ventures, First Round Capital, and Conviction, with more than $73M raised to date.

About the role

Town is building the most personalized, most capable AI assistant for everyone — and personalization at that level is a retrieval and data problem. The assistant is only as good as the context it can bring into the moment: the right memory, message, document, or relationship, pulled fast and related by meaning across everything a person and their team touch.

You'll build the foundation the whole product reasons over: the search and data infrastructure behind that context. One shared retrieval layer combining lexical and semantic search, the realtime and batch pipelines that keep it fresh and correct, and the durable data model everything else is built on.

This is greenfield and high-leverage: you'll be the first person building this layer.

What you'll do
  • Build the search and retrieval layer that puts the right context at every Townie's fingertips, the moment it's needed — one shared layer the whole product pulls from instead of refetching context on its own.

  • Combine lexical and semantic search and own the tradeoffs between them: vector vs. lexical, precompute vs. fetch, hybrid retrieval, and ranking.

  • Design the durable data model the assistant's work is built on, so context is relevant, fast, and cost-effective.

  • Build and operate the pipelines behind it — realtime/streaming and batch — that keep the index fresh and correct as the underlying data changes.

  • Stand up the indexing and storage layer and keep it fast and reliable at scale: latency, cost, freshness, and completeness.

  • Lay the groundwork for relating content by meaning across everything the assistant knows — the start of a knowledge graph of people, companies, projects, and how they connect.

You might thrive here if you...
  • Have significant, hands-on experience across lexical and semantic search components and approaches (BM25, embeddings, ANN/vector indexes, hybrid retrieval, ranking).

  • Have run large-scale data infrastructure, ideally both realtime/streaming and batch — pipelines, indexing, and storage.

  • Can make retrieval fast and cheap at scale, and reason about the latency, cost, and freshness tradeoffs cold.

  • Are a systems thinker comfortable in greenfield, where the foundation doesn't exist yet.

  • Are excited to take these systems from rapid prototype to production scale.

  • Bonus if you've worked on ranking/relevance, knowledge graphs, or retrieval for LLM or agentic systems.

Location

San Francisco, CA. Five days a week in person at our Financial District office.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

AI Context & Data Infrastructure Engineer
AI Context & Data Infrastructure Engineer

Town.com, Inc. • San Francisco (CA)

On-site
USD 180,000 - 260,000
AI Product Engineer
AI Product Engineer

Town • San Francisco (CA)

On-site
USD 120,000 - 160,000
Staff Backend Engineer
Staff Backend Engineer

Town • San Francisco (CA)

On-site
USD 120,000 - 160,000
Staff Backend Engineer
Staff Backend Engineer

Town • New York (NY)

On-site
USD 140,000 - 210,000
AI Product Engineer San Francisco, California · Full-time · $225K – $300K
AI Product Engineer San Francisco, California · Full-time · $225K – $300K

Town • San Francisco (CA)

On-site
USD 120,000 - 160,000
Staff Backend Engineer San Francisco, California · Full-time · $250K – $300K
Staff Backend Engineer San Francisco, California · Full-time · $250K – $300K

Town • San Francisco (CA)

On-site
USD 120,000 - 180,000
AI Engineer, Evals & Agent Quality
AI Engineer, Evals & Agent Quality

Town • San Francisco (CA)

On-site
USD 180,000 - 240,000
AI Product Engineer
AI Product Engineer

Town.com, Inc. • New York (NY)

On-site
USD 150,000 - 210,000
AI Context & Retrieval Infrastructure Engineer
AI Context & Retrieval Infrastructure Engineer

Town.com, Inc. • San Francisco (CA)

On-site
USD 180,000 - 260,000
Product Support
Product Support

Town • San Francisco (CA)

On-site
USD 70,000 - 110,000
In-office in SF