Staff AI Engineer: End-to-End LLM Architect (Hybrid)

Relativity

Alton (IL)

Hybrid

USD 174,000 - 262,000

Full time

5 days ago
Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Relativity is seeking a hands-on Staff Engineer to own how Discovery systems are built, evaluated, and operated. You will shepherd end-to-end inference paths, from retrieval to outputs, and ensure robust observability in a multi-tenant cloud environment.

You will define evaluation metrics, optimize latency budgets, and collaborate across domains—engineering, product, and design—to deliver defensible, scalable solutions for legal data processing at scale.

Qualifications

  • 7+ years building and operating large-scale distributed cloud systems.
  • 3+ years shipping production model-backed systems.
  • Depth across LLMs, RAG, agentic workflows, and vector/hybrid search.
  • Owned an evaluation practice with eval sets, metrics and gates.
  • Architecture accountability across cloud services in multi-tenant SaaS.
  • Quantitative reasoning for model behavior including accuracy, latency, cost.
  • Strong technical leadership and the ability to present tradeoffs to executives.

Responsibilities

  • Staff engineers set technical direction across teams and build the hardest parts from prototype through production.
  • Own the inference path end to end including retrieval, context construction, routing, and observability.
  • Set the evaluation bar with gold sets, offline suites, CI gates, and online telemetry.
  • Own token economics and latency budgets; decide routing, caching, and model tier based on data.
  • Design for reuse by decoupling capability logic from delivery harness via interface contracts.
  • Architect the data and retrieval layer including indexing, embedding, and hybrid search.
  • Engineer defensibility with citation fidelity, confidentiality controls, and auditability.
  • Lead root-cause analysis, publish learnings, and raise engineering standards.
  • Collaborate across domains with Applied Science, Product, Design, and peers to drive closure.

Skills

Distributed systems
Production ML models
LLMs & RAG
Evaluation practice
Cloud architecture
Quantitative reasoning
Technical leadership

Job description

Relativity is seeking a hands-on Staff Engineer to own how Discovery systems are built, evaluated, and operated. You will shepherd end-to-end inference paths, from retrieval to outputs, and ensure robust observability in a multi-tenant cloud environment.

You will define evaluation metrics, optimize latency budgets, and collaborate across domains—engineering, product, and design—to deliver defensible, scalable solutions for legal data processing at scale.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff AI Platform Architect - Hybrid/Remote
Staff AI Platform Architect - Hybrid/Remote

Relativity • Chicago (IL)

Hybrid
USD 205,000 - 307,000
Equity program
DTO
Parental leave
Senior Staff Engineer, AI-Powered Legal Cloud Platform
Senior Staff Engineer, AI-Powered Legal Cloud Platform

Relativity • Chicago (IL)

Hybrid
USD 174,000 - 262,000
Home office stipend
Equity program
Parental leave
Senior AI Platform Architect (Hybrid/Remote)
Senior AI Platform Architect (Hybrid/Remote)

Relativity • Springfield (IL)

Hybrid
USD 205,000 - 307,000
Equity program
Parental leave
DTO
Principal ML Engineer: Scalable LLM & AI Systems Architect
Principal ML Engineer: Scalable LLM & AI Systems Architect

RELX Inc. Company • North Carolina

On-site
USD 136,000 - 253,000
Staff AI Engineer
Staff AI Engineer

Relativity • Alton (IL)

Hybrid
USD 174,000 - 262,000
Applied Science Lead: AI for Legal Discovery
Applied Science Lead: AI for Legal Discovery

Relativity • Michigan

Hybrid
USD 208,000 - 312,000
AI Engineer (Applied LLM Systems)
AI Engineer (Applied LLM Systems)

Discernis • New York (NY)

On-site
USD 140,000 - 200,000
Remote Staff Software Engineer – AI-Powered Cloud Systems
Remote Staff Software Engineer – AI-Powered Cloud Systems

Relativity • United States

Remote
USD 87,000 - 130,000
Equity program
Parental leave
Annual bonus
Senior AI Engineer: LLM Evaluation, Production & Optimization
Senior AI Engineer: LLM Evaluation, Production & Optimization

LawPro.ai • North Carolina

On-site
USD 140,000 - 190,000
Remote Senior Staff ML Engineer — LLM Systems Lead
Remote Senior Staff ML Engineer — LLM Systems Lead

Undisclosed • United States

Remote
USD 180,000 - 260,000
Remote-friendly culture
Flexible time off
Comprehensive benefits
+1