Software Engineer, AI Systems (United States)

Aifund; United States; On-site; USD 180,000 - 300,000; Full time

Job summary

Haven Safety invites an experienced AI/LLM engineer to join the founding team in the US. You will design, ship, instrument, and improve multi-step reasoning pipelines that connect Haven's evidence base and knowledge graph to investigator-facing product experiences.

This role reports to the CTO and demands precision, traceability, and strong production discipline. You will own production LLM systems, manage tool calls and agent orchestration, and work across OpenAI, Anthropic, and Google.

Qualifications

• Build and operate multi-step LLM pipelines coordinating model calls, tool calls, graph queries and retrieval.
• Design context layers across graph traversal and hybrid retrieval to feed models with right evidence.
• Establish datasets, regression suites, and human-review loops for quality attribution.

Responsibilities

• Production reasoning systems: build and operate multi-step LLM pipelines.
• Agent orchestration: extend the coordinated agent team and orchestration layer.
• Grounding and retrieval: ensure right evidence and knowledge are retrieved for each call.
• Evaluation: create scoring, regression tests, and human-label loops.
• Production AI operations: implement tracing, audits, monitoring, and dashboards.
• Technical direction: select models and shape the AI roadmap with product teams.
• Production LLM systems: ship reliable LLM-powered features used by customers.
• Agentic workflows: build and debug multi-step tool-calling workflows.
• Evaluation discipline: establish repeatable LLM evaluation across datasets.
• Retrieval judgment: decide what context to retrieve and how much.
• Production ownership: own deployment through monitoring and incident response.
• Model judgment: explain tradeoffs across quality, latency, cost, and risk.
• Graph reasoning: work with Neo4j/Cypher and live knowledge graphs.
• Security: guard against prompt injection and data leakage.
• Azure and hybrid search: operate on Azure and similar platforms.

Skills

Python

Neo4j

Cypher

FastAPI

LangChain

LangGraph

Azure

OpenAI

Anthropic

Google

Tools

Neo4j

Cypher

Azure AI Search

Pinecone

MongoDB Atlas

Job description

About Haven Safety:

Haven Safety AI is building the enterprise learning intelligence layer for safety. Co-founded with The AES Corporation and AI Fund, the venture studio founded by Andrew Ng, Haven helps high-risk organizations learn faster from what goes wrong so they can prevent what comes next.

Haven works alongside existing EHS enterprise systems to improve how organizations investigate, assess, and learn from incidents. INVESTIGATE guides evidence synthesis, timeline development, multi-threaded causal analysis, and corrective actions. ASSURE continuously reviews completed investigations for evidence quality, causal coverage, guideline adherence, and CAPA strength. LEARN reasons across incident history to surface recurring control failures, repeated corrective-action patterns, CAPA debt, and emerging signals.

The platform combines current incident evidence, company knowledge, historical cases, and an industry knowledge graph. A coordinated team of specialized AI agents examines evidence, controls, engineering factors, procedures, regulations, training, and organizational history, then produces one traceable assessment for human review. Customers have reported an 80% reduction in root cause analysis labor time using Haven.

About the Role:

You will work on the AI and LLM engineering layer that connects Haven's evidence base and knowledge graph to the product experiences investigators and safety leaders use. This is a build-and-operate role reporting to the CTO. You will design the reasoning, ship it, instrument it, and improve it using production evidence.

The work is technically demanding and operationally consequential. Haven serves enterprise customers in regulated, safety-critical industries through multi-tenant and dedicated deployments. A plausible answer and a correct answer can look the same until someone acts on it, so precision, traceability, evaluation, and human oversight are core product requirements.

Haven Safety is a VC-backed pre-seed venture. This role will be a important member of the founding team and will require wearing multiple hats. This role is based in the US and relocation will not be considered.

Responsibilities

• Production reasoning systems. Build and operate multi-step LLM pipelines that coordinate model calls, tool calls, graph queries, retrieval, quality gates, and specialist-agent handoffs.
• Agent orchestration. Extend Haven's coordinated agent team and the orchestration layer that carries an incident from evidence through analysis, review, and enterprise learning.
• Grounding and retrieval. Design the context layer across Neo4j graph traversal, vector search, and hybrid retrieval so every model call receives the right evidence and organizational knowledge.
• Evaluation. Build datasets, scoring, regression suites, model comparisons, human-label loops, and per-stage quality attribution for extraction and reasoning tasks.
• Production AI operations. Implement tracing, tool-call audits, cost and latency monitoring, failure handling, and quality dashboards; catch loops, hallucinations, and silent drift before customers do.
• Technical direction. Select models by task across OpenAI, Anthropic, and Google; partner with product and knowledge engineering; and help shape the AI roadmap.
• Production LLM systems. AI or ML engineering, including shipping LLM systems that real users depend on.
• Agentic workflows. Hands-on experience building and debugging multi-step, tool-calling workflows with LangGraph, LangChain, or an equivalent framework.
• Evaluation discipline. A repeatable approach to LLM evaluation, including representative datasets, regression testing, LLM-as-judge techniques, or human review loops.
• Retrieval judgment. Experience assembling context for LLMs and a clear point of view on what to retrieve, how much, and why.
• Production ownership. A track record of owning systems from deployment through monitoring and incident response, including a strong story about a failure or regression you diagnosed and fixed.
• Model judgment. Comfort working across model providers and explaining tradeoffs in quality, latency, cost, context, and operational risk.
• Graph reasoning. Comfort with Neo4j and Cypher, or a comparable graph store, and the ability to ramp quickly on graph data modeling. Strongly preferred.
• Strong Python. Production habits around FastAPI, asynchronous services, testing, observability, and maintainable interfaces are required.
• Deep graph experience. Cypher fluency, schema evolution, MERGE patterns, embeddings, and operating a live knowledge graph.
• Enterprise AI security. Prompt-injection awareness, context-leak prevention, tenant isolation, role-based access, and policy-layer separation.
• Azure and hybrid search. Experience running production AI services in Azure and using Azure AI Search, Pinecone, MongoDB Atlas, pgvector, Elasticsearch, or a similar platform.
• B2B Enterprise SaaS. Prior experience operating in an enterprise-level environment is strongly preferred.
• Meaningful reasoning problems. Work across evidence, causal pathways, controls, organizational history, and corrective actions in a domain where correctness matters.
• An evaluation-first culture. Make quality measurable, observable, and improvable instead of relying on demos or intuition.
• Visible customer impact. Build for safety teams in energy, utilities, infrastructure, construction, and manufacturing, with direct feedback from the people using the output.
• Small team, high ownership. Work closely with the CTO, product, and knowledge engineering, make consequential technical decisions, and see your work reach customers quickly.

Sponsorship will not be provided for this role.