Staff AI Engineer

Posted 3 Days Ago
Hiring Remotely in Illinois, USA
Remote
205K-307K Annually
Senior level
Legal Tech • Software
Organize data. Discover the truth. Act on it.
The Role
Lead the architecture and long-term roadmap for Relativity’s machine learning platform. Design scalable training, inference, monitoring, and ML operations systems; evaluate emerging technologies; improve model performance and cost efficiency through optimization techniques; and ensure secure, reliable AI services. Partner with product, program, and data science leaders, guide cross-functional initiatives, establish engineering standards, and mentor senior engineers across multiple teams.
Summary Generated by Built In

Posting Type

Hybrid/Remote

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, data breach responses, 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 and impactful information in their matters. Beyond our commercial impact, we are committed to expanding access to technology and supporting pro bono legal work.

Job Description and Requirements

WHAT WE DO
The AI team at Relativity is focused on helping users discover the truth faster and act on data with confidence through AI-powered capabilities embedded throughout the discovery process.
  • Build trusted AI solutions that improve user experiences, products, investigations, and legal matters.
  • Develop world-class tools to solve complex challenges through experimentation and innovation.
  • Leverage AI across all stages of the discovery lifecycle to improve outcomes and optimize product operations.
  • Invest in modern data infrastructure, data pipelines, and data lake technologies that enable large-scale AI and analytics.
  • Foster a culture of exploration, experimentation, continuous learning, and innovation.
ABOUT THE ROLE
Staff AI Engineers at Relativity operate as domain architects and strategic accelerators. This role works across multiple engineering and data science teams to define the long-term vision for Relativity’s machine learning platform, guide high-impact initiatives from concept through production, and ensure AI capabilities remain secure, reliable, scalable, and cost-effective.
WHAT YOU’LL DO
  • Define and evangelize the multi-year technical roadmap for ML Ops, aligning platform architecture with product and research objectives.
  • Provide technical direction and mentorship across multiple engineering teams and organizations.
  • Drive the design and architecture of training, inference, and monitoring systems focused on scalability, extensibility, performance, and reliability.
  • Partner with product, program, and data science leaders to scope and deliver cross-functional roadmaps.
  • Evaluate emerging ML Ops technologies and establish engineering standards and best practices.
  • Champion advanced optimization techniques, including sparsity, quantization, and pruning, to improve performance and cost efficiency.
  • Collaborate with security teams to maintain strong data protection practices and responsible AI standards.
  • Lead architecture reviews, publish engineering best practices, and mentor senior and lead engineers across the organization.
WHAT WE’RE LOOKING FOR
Required
  • 8+ years of professional software engineering experience.
  • 5+ years of experience working in ML/AI or big data environments.
  • 4+ years of technical leadership experience across multiple teams.
  • Expert-level proficiency in Python, Java, or Scala for production systems.
  • Deep hands-on experience with Docker, Kubernetes/Helm, and infrastructure-as-code tools such as Terraform or Pulumi.
  • Proven experience building and operating CI/CD pipelines for machine learning workflows using technologies such as Prefect or Airflow.
  • Experience deploying and operating secure, monitored services on AWS, Azure, or GCP.
  • Demonstrated ability to mentor senior engineers, influence technical strategy, and drive cross-team initiatives to successful completion.
Preferred
  • Master’s or PhD in Computer Science, Engineering, Mathematics, or a related field.
  • Recognized technical thought leadership through open-source contributions, conference presentations, or publications.
  • Experience scaling ML platforms utilizing distributed data technologies such as Spark or Kafka.
  • Expertise implementing advanced model optimization techniques, including compression, pruning, and quantization, in production environments.
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:

$205,000 and $307,000

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 Intelligence (AI), Big Data, Cloud Computing, Data Science, Deep Learning, Distributed Systems, Natural Language, Python (Programming Language), Software Engineering

Skills Required

  • 8+ years of professional software engineering experience
  • 5+ years of experience working in ML/AI or big data environments
  • 4+ years of technical leadership experience across multiple teams
  • Expert-level proficiency in Python, Java, or Scala for production systems
  • Hands-on experience with Docker, Kubernetes/Helm, and infrastructure-as-code tools such as Terraform or Pulumi
  • Experience building and operating CI/CD pipelines for machine learning workflows using technologies such as Prefect or Airflow
  • Experience deploying and operating secure, monitored services on AWS, Azure, or GCP
  • Ability to mentor senior engineers, influence technical strategy, and drive cross-team initiatives to completion
  • Master's or PhD in Computer Science, Engineering, Mathematics, or a related field
  • Technical thought leadership through open-source contributions, conference presentations, or publications
  • Experience scaling ML platforms with distributed data technologies such as Spark or Kafka
  • Expertise implementing compression, pruning, and quantization in production environments

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

Clear Street Logo Clear Street

Senior / Staff AI Model Engineer

Fintech • Software • Financial Services
Easy Apply
Remote or Hybrid
USA
581 Employees
200K-350K Annually

Shield AI Logo Shield AI

Staff Engineer

Aerospace • Artificial Intelligence • Machine Learning • Robotics • Software
Remote
USA
200K-300K Annually

Rula Logo Rula

Staff Engineer

Healthtech • Social Impact • Software • Telehealth
Remote
United States
630 Employees
242K-288K Annually

Affirm Logo Affirm

Security Engineer

Big Data • Fintech • Mobile • Payments • Financial Services
Easy Apply
Remote
United States
2200 Employees
204K-290K Annually

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