Staff AI Software Engineer - Remote Position

About the Role

We are seeking a Staff AI Software Engineer to join our innovative team at Relativity. This remote role offers the opportunity to work on cutting-edge AI technologies that are transforming the way users interact with data. As a Staff AI Software Engineer, you will play a pivotal role in shaping the future of our machine learning platform and ensuring that our AI capabilities are secure, reliable, and scalable.

What You'll Do

• Define and evangelize the multi-year technical roadmap for ML Ops, aligning platform architecture with product and research needs.
• Provide technical direction and mentorship across several engineering squads, evolving the senior expectation to lead a team of engineers.
• Drive the design of training, inference, and monitoring systems that meet demanding timelines and performance goals.
• Partner with product, program, and science leaders to scope and deliver cross-functional roadmaps, minimizing risks and maximizing opportunities.
• Evaluate emerging ML Ops tooling, establish engineering standards, and champion advanced optimization techniques for cost and performance gains.
• Collaborate with security teams to uphold robust data protection and responsible AI practices at scale.
• Foster a culture of learning by leading architecture reviews, publishing best practices, and mentoring senior and lead engineers organization-wide.

Requirements

• 8+ years of professional software engineering experience, including 5+ years in ML/AI or big-data environments.
• 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 track record building and operating CI/CD pipelines for ML workflows and deploying secure, monitored services on AWS, Azure, or GCP.
• Demonstrated ability to mentor senior engineers, influence technical direction, and drive cross-team initiatives to completion.

Nice to Have

• Master’s or PhD in Computer Science, Engineering, Mathematics, or related field.
• Recognized thought leadership through open-source contributions, conference talks, or publications.
• Experience scaling ML platforms that leverage distributed data frameworks such as Spark or Kafka.
• Expertise in implementing advanced model-optimization techniques in production environments.

What We Offer

• Comprehensive health, dental, and vision plans.
• Parental leave for primary and secondary caregivers.
• Flexible work arrangements.
• Two, week-long company breaks per year.
• Additional time off.
• Long-term incentive program.
• Training investment program.

At Relativity, we celebrate diversity and believe that employees are happiest when empowered to be their authentic selves. Join us in our mission to leverage AI for better data management and user experiences.