Advanced AI Engineer

Posted 13 Days Ago
Be an Early Applicant
7 Locations
In-Office
188K-282K Annually
Mid level
Legal Tech • Software
Organize data. Discover the truth. Act on it.
The Role
Build and extend Relativity’s Python-based agent runtime using LangGraph, LangChain, and Deep Agents. Develop orchestration, tool calling, memory, human-in-the-loop review, streaming, checkpointing, protocols, and multi-agent capabilities. Deliver tested, production-ready cloud-native systems with PostgreSQL, Azure, Docker, and Kubernetes while collaborating with application and applied science teams. Contribute to design reviews and safely move AI capabilities from experimentation into production.
Summary Generated by Built In

Posting Type

Hybrid/Remote Poland

Job Overview

WHO WE ARE
Relativity is a leading legal data intelligence company building technology that helps users organize data, discover the truth, and act on it with confidence. Our AI-powered, cloud platform, RelativityOne, transforms massive volumes of complex information into actionable insights for litigation, investigations, regulatory inquiries, and other high-stakes legal work where accuracy and trust are crucial.
The world’s largest law firms, corporations, and government agencies rely on Relativity’s legal AI software to securely surface and manage the most relevant information in their matters.
WHAT WE DO
At Relativity, engineers do more than write code — they build systems that let users uncover insights from complex data at scale using cloud-native architecture, AI, and modern tools.
The AI Platform group builds the agentic foundation that powers Relativity’s AI products, including aiR Assist and ClaiR. The Agent Harness team owns the runtime that runs our agents: agent orchestration, tool and skill execution, memory, and the protocols that let agents interact with each other and with external systems.
ABOUT THE ROLE
As an Advanced AI Engineer on the Agent Harness team, you will build and extend Relativity’s agent runtime — working hands-on with industry frameworks like LangGraph, LangChain, and Deep Agents and fitting them to Relativity’s needs so every Relativity agent has a solid runtime to build on.
You will help mature the harness with industry-standard capabilities like long-session handling or broader protocol support, so it is ready before ClaiR and aiR Assist multiply agentic traffic, rather than after.
You will contribute to design discussions and turn agreed designs into well-tested, production-ready capabilities that application teams adopt.
You will work in a fast-moving space where agent architecture changes quickly, and grow your depth in agent engineering as you go.

Job Description and Requirements

WHAT YOU’LL DO 

  • Build and extend the agent runtime in Python using LangGraph, LangChain, and Deep Agents (deepagents) — stateful graph agents, harness profiles, and multi-agent / subagent orchestration. 

  • Implement harness capabilities that application teams rely on — for example streaming and structured/generative output, tool calling, human-in-the-loop review, conversation branching and message queues, session memory, and checkpoint/resume (time-travel) over long-running work. 

  • Contribute to the harness’s extensibility and protocol layer: Model Context Protocol (MCP) tool servers and clients, Agent-to-Agent (A2A) interoperability, and a registry of skills, tools, and configurations composed at runtime. 

  • Help agents use the model that fits each use case, working across a range of LLM providers (e.g. OpenAI, Gemini). 

  • Write clean, well-tested code and participate in design and code reviews; take ownership of components and see them through to production. 

  • Collaborate with teammates, Applied Science, and aiR application teams to move new capabilities from experiment into production safely. 

WHAT WE’RE LOOKING FOR 

Required 

  • 3+ years of professional software engineering experience, with strong, recent Python building production systems. 

  • Experience building LLM-powered or agentic systems with frameworks such as LangChain / LangGraph (or equivalent), including tool calling, orchestration, and agent state management. 

  • Good design instincts — writing clean, testable code and contributing to the design of the components you own. 

  • Solid understanding of modern async Python, API and service design, and relational data (PostgreSQL). 

  • Experience delivering cloud-native systems (Azure or similar) with CI/CD and containers (Docker); familiarity with Kubernetes. 

  • Bachelor’s degree in Computer Science, Engineering, or equivalent experience; English proficiency for technical communication. 

Preferred 

  • Experience with Deep Agents, multi-agent / subagent architectures, agent memory, or human-in-the-loop patterns. 

  • Familiarity with the Model Context Protocol (MCP) or agent-to-agent (A2A) interoperability. 

  • Experience with RAG and retrieval systems, citations, and evaluating LLM output quality. 

  • Familiarity with LLM observability and tracing (MLflow, OpenTelemetry) and evals for safe model upgrades — nice to have. 

  • Infrastructure-as-code (Pulumi/Terraform). 

WHY WE COULD BE A GREAT FIT

Impactful Mission

  • Build systems that help customers organize data, discover the truth, and act on it in high-stakes legal matters.

Engineering at Scale

  • Work on distributed, cloud-native systems that process large volumes of data.

Cutting-Edge Technology

  • Build with AI, cloud platforms, and scalable architectures shaping legal tech.

Growth and Ownership

  • Gain experience owning systems end-to-end across cloud and distributed environments.

Collaborative Culture

  • Work in a team focused on knowledge sharing and continuous improvement.

Inclusive Environment

  • Diverse perspectives create stronger teams and better outcomes.

Compensation and Benefits

  • Competitive salary, benefits, DTO, parental leave, and equity program.

Relativity is committed to competitive, fair, and equitable compensation practices.

This position is eligible for total compensation which includes a competitive base salary, an annual performance bonus, and long-term incentives.

The expected salary range for this role is between following values:

188 000 and 282 000PLN

The final offered salary will be based on several factors, including but not limited to the candidate's depth of experience, skill set, qualifications, and internal pay equity. Hiring at the top end of the range would not be typical, to allow for future meaningful salary growth in this position. 

Required Skills:

Algorithms, Artificial Neural Networks (ANNS), Big Data, Computer Vision, Data Science, Deep Learning, Machine Learning (ML), Natural Language, Natural Language Processing (NLP), Software Engineering

Skills Required

  • 3+ years of professional software engineering experience with strong, recent Python experience building production systems
  • Experience building LLM-powered or agentic systems using LangChain, LangGraph, or equivalent frameworks
  • Experience with tool calling, orchestration, and agent state management
  • Ability to write clean, testable code and contribute to component design
  • Understanding of modern async Python, API and service design, and PostgreSQL
  • Experience delivering cloud-native systems using Azure or similar platforms
  • Experience with CI/CD and Docker containers
  • Familiarity with Kubernetes
  • Bachelor’s degree in Computer Science, Engineering, or equivalent experience
  • English proficiency for technical communication
  • Experience with Deep Agents, multi-agent or subagent architectures, agent memory, or human-in-the-loop patterns
  • Familiarity with Model Context Protocol or agent-to-agent interoperability
  • Experience with RAG, retrieval systems, citations, and LLM output evaluation
  • Familiarity with MLflow, OpenTelemetry, LLM observability, tracing, and model evaluation
  • Experience with Pulumi or Terraform infrastructure as code

Relativity Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Relativity and has not been reviewed or approved by Relativity.

  • Leave & Time Off Breadth — Time off options include discretionary time off and two company‑wide breaks each year, providing additional recharge time.
  • Healthcare Strength — Health coverage includes comprehensive medical, dental, and vision plans, telehealth access, and wellness resources such as a Headspace subscription.
  • Parental & Family Support — North America offers up to 12 weeks of fully paid parental leave, with comparable regional programs in EMEA and APAC.

Relativity Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Chicago, IL
1,550 Employees
Year Founded: 2001

What We Do

At Relativity, we build innovative and comprehensive tools for making sense of unstructured data. When more people can find the facts in mountains of documents, emails, and texts, more legal and data-centric matters can be resolved equitably. Join us in our mission to help our customers organize data, discover the truth, and act on it. Relativity makes software to help users organize data, discover the truth and act on it. Its SaaS product, RelativityOne, manages large volumes of data and quickly identifies key issues during litigation and internal investigations. Relativity has more than 300,000 users in approximately 40 countries serving thousands of organizations globally primarily in legal, financial services and government sectors, including the U.S. Department of Justice and 198 of the Am Law 200. Relativity does not tolerate racism or discrimination of any kind. We do not accept unfair treatment of any person or group of people. We’re committed to advocating for change to make our world a more inclusive, just place.

Why Work With Us

We believe in our team members and we want to help you own your career as part of a community of values-driven people who help customers around the world solve complex data challenges. At Relativity, you’ll take on challenging work, but you’ll also partner with talented colleagues and pursue plenty of learning and development opportunities.

Gallery

Gallery

Similar Jobs

Pfizer Logo Pfizer

Senior Manager, HTA, Value and Evidence (HV&E), Genitourinary Cancer

Artificial Intelligence • Healthtech • Machine Learning • Natural Language Processing • Biotech • Pharmaceutical
In-Office or Remote
30 Locations
121990 Employees
139K-232K Annually

Pfizer Logo Pfizer

Staff Software Engineer

Artificial Intelligence • Healthtech • Machine Learning • Natural Language Processing • Biotech • Pharmaceutical
In-Office or Remote
36 Locations
121990 Employees

Capco Logo Capco

Devsecops Engineer

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
Poland
6000 Employees

Capco Logo Capco

Software Engineer

Fintech • Professional Services • Consulting • Energy • Financial Services • Cybersecurity • Generative AI
Remote or Hybrid
Poland
6000 Employees

Similar Companies Hiring

Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees
Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account