Advanced AI Engineer

LE0001 Relativity ODA LLC; Illinois; Hybrid; USD 103,000 - 155,000; Full time

Benefits offered by this job

Equity program

Parental leave

DTO (PTO)

Job summary

Relativity is seeking an Advanced AI Engineer to bridge core engineering and data science, owning ML lifecycle components from automated training to secure deployment. You will work with senior engineers to advance scalable AI platforms and contribute to technical strategy, implementation, and innovation.

The role emphasizes CI/CD for ML, MLOps tooling, and optimization of production models within a hybrid/remote environment.

Qualifications

• 3+ years of professional software engineering experience
• 1+ year in ML/AI or大 data environments
• Proficiency in Python, Java, or C#
• Production experience with Docker
• Experience deploying cloud-based solutions (AWS/Azure/GCP)
• Experience with IaC tools (Terraform/Pulumi)
• Familiarity with workflow orchestration platforms (Prefect, Airflow)
• Kubernetes and Helm fundamentals
• Experience deploying ML models in production
• Ability to collect/analyze model reliability metrics
• Strong cross-functional collaboration and communication
• Bachelor’s degree in CS/Engineering; Master’s preferred
• Experience with ML lifecycle platforms like MLflow or Kubeflow
• Experience with model optimization techniques
• Exposure to distributed data processing like Spark/EMR/Kafka
• Familiarity with TensorFlow or PyTorch
• Experience in secure data processing environments

Responsibilities

• Design and implement scalable ML/AI platforms
• Collaborate with data scientists and engineers to deliver ML solutions
• Develop CI/CD pipelines for ML models and data workflows
• Build automated model training, deployment, and lifecycle management
• Evaluate new MLOps technologies to improve efficiency and reduce costs
• Deploy, monitor, and troubleshoot production ML models
• Track health, performance, reliability, and cost metrics for AI systems
• Participate in code and design reviews
• Mentor junior engineers and share best practices
• Continuously learn new tools and techniques for AI platform

Skills

Python/Java/C#

Docker

Cloud (AWS/Azure/GCP)

IaC (Terraform/Pulumi)

Kubernetes/Helm

MLOps platforms (MLflow/Kubeflow)

CI/CD for ML

Data/ML metrics

Spark/EMR/Kafka

TensorFlow/PyTorch

Security/compliance data

Collaboration/communication

Education

Bachelor’s degree in CS/Engineering or related field

Master’s degree in a related discipline

Tools

MLflow

Kubeflow

Terraform

Pulumi

Prefect/Airflow

Job description

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.

WHAT WE DO

Our AI team is focused on helping users discover the truth more quickly and act on data with confidence through AI-powered innovation. We are committed to algorithm excellence, building trusted and scalable AI solutions that improve user experiences, products, investigations, and operational efficiency. Relativity’s AI and Data Science teams leverage large-scale datasets, modern data infrastructure, and advanced machine learning technologies to deliver insights at scale. Our engineers and data scientists collaborate to build secure, high-performing platforms that support experimentation, innovation, and continuous improvement across the AI lifecycle.

Job Description and Requirements

ABOUT THE ROLE

As an Advanced AI Engineer, you will bridge the gap between core engineering and data science teams, building and evolving the platforms, pipelines, and practices that transform research prototypes into reliable, scalable production solutions. You will own critical components of the machine learning lifecycle, including automated training pipelines, secure deployments, monitoring, and operational excellence. Working closely with Senior and Lead Engineers, you will help drive continuous improvement across Relativity’s AI platform while contributing to technical strategy, implementation, and innovation.

WHAT YOU’LL DO

• Contribute to the design and implementation of ML/AI platforms with a focus on scalability, reliability, security, and standardized GenAI workflows.
• Partner with data scientists, product managers, security teams, and data engineers to deliver high-impact machine learning solutions.
• Implement and improve CI/CD pipelines for machine learning models and data workflows using containerization, infrastructure-as-code, and orchestration technologies.
• Build and enhance automated model training, deployment, and lifecycle management processes.
• Prototype and evaluate emerging MLOps technologies to improve efficiency, optimize costs, and enable new product capabilities.
• Deploy, monitor, tune, and troubleshoot production machine learning models.
• Establish and track health, performance, reliability, and cost optimization metrics for AI systems.
• Participate in code reviews and design reviews while contributing directly to implementation efforts.
• Mentor junior engineers and share best practices across the engineering and AI organizations.
• Continuously learn and apply new technologies, tools, and techniques to improve the AI platform.

WHAT WE’RE LOOKING FOR

• Required 3+ years of professional software engineering experience, including at least 1 year working in ML/AI or big data environments.
• Proficiency in Python, Java, or C#.
• Production experience using Docker.
• Experience deploying cloud-based solutions on AWS, Azure, or GCP.
• Experience using infrastructure-as-code tools such as Terraform or Pulumi.
• Familiarity with workflow orchestration platforms such as Prefect, Airflow, or similar technologies.
• Understanding of Kubernetes and Helm fundamentals.
• Experience deploying, monitoring, and troubleshooting machine learning models in production environments.
• Ability to collect and analyze metrics related to model reliability and algorithm health.
• Strong collaboration and communication skills with cross-functional stakeholders.
• Preferred Bachelor’s degree in Computer Science, Engineering, Mathematics, or a related field.
• Master’s degree in a relevant discipline.
• Experience with ML lifecycle platforms such as MLflow or Kubeflow.
• Experience with model optimization techniques including quantization, pruning, or compression.
• Exposure to distributed data processing technologies such as Spark, EMR, or Kafka.
• Familiarity with deep learning frameworks such as TensorFlow or PyTorch.
• Experience working in secure and compliant data processing 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: $103,000 and $155,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: Engineering Principle, Hardware Integration, Innovation, Problem Solving, Process Improvements, Quality Assurance (QA), Research and Development, System Designs, Technical Documents, Troubleshooting

We’re solving big data challenges in the legal tech industry, and we’re always looking for more people to join us on the journey. At Relativity, you'll learn cross-functional skills to grow your career and have the chance to make a big impact on our customers, our industry, and our communities. We admire and value our employees, so it’s no surprise that our hiring process is designed to help us really get to know you – and for you to get to know us, too.